mock_source_stream.cc 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102
  1. // Copyright 2015 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "net/filter/mock_source_stream.h"
  5. #include <utility>
  6. #include "base/check_op.h"
  7. #include "net/base/io_buffer.h"
  8. #include "testing/gtest/include/gtest/gtest.h"
  9. namespace net {
  10. MockSourceStream::MockSourceStream() : SourceStream(SourceStream::TYPE_NONE) {}
  11. MockSourceStream::~MockSourceStream() {
  12. DCHECK(!awaiting_completion_);
  13. // All data should have been consumed.
  14. EXPECT_TRUE(results_.empty());
  15. }
  16. int MockSourceStream::Read(IOBuffer* dest_buffer,
  17. int buffer_size,
  18. CompletionOnceCallback callback) {
  19. DCHECK(!awaiting_completion_);
  20. DCHECK(!results_.empty());
  21. if (results_.empty())
  22. return ERR_UNEXPECTED;
  23. QueuedResult r = results_.front();
  24. DCHECK_GE(buffer_size, r.len);
  25. if (r.mode == ASYNC) {
  26. awaiting_completion_ = true;
  27. dest_buffer_ = dest_buffer;
  28. dest_buffer_size_ = buffer_size;
  29. callback_ = std::move(callback);
  30. return ERR_IO_PENDING;
  31. }
  32. results_.pop();
  33. memcpy(dest_buffer->data(), r.data, r.len);
  34. return r.error == OK ? r.len : r.error;
  35. }
  36. std::string MockSourceStream::Description() const {
  37. return "";
  38. }
  39. bool MockSourceStream::MayHaveMoreBytes() const {
  40. if (always_report_has_more_bytes_)
  41. return true;
  42. return !results_.empty();
  43. }
  44. MockSourceStream::QueuedResult::QueuedResult(const char* data,
  45. int len,
  46. Error error,
  47. Mode mode)
  48. : data(data), len(len), error(error), mode(mode) {}
  49. void MockSourceStream::AddReadResult(const char* data,
  50. int len,
  51. Error error,
  52. Mode mode) {
  53. if (error != OK) {
  54. // Doesn't make any sense to have both an error and data.
  55. DCHECK_EQ(len, 0);
  56. } else {
  57. // The read result must be between 0 and 32k (inclusive) because the read
  58. // buffer used in FilterSourceStream is 32k.
  59. DCHECK_GE(32 * 1024, len);
  60. DCHECK_LE(0, len);
  61. }
  62. if (len > 0 && read_one_byte_at_a_time_) {
  63. for (int i = 0; i < len; ++i) {
  64. QueuedResult result(data + i, 1, OK, mode);
  65. results_.push(result);
  66. }
  67. return;
  68. }
  69. QueuedResult result(data, len, error, mode);
  70. results_.push(result);
  71. }
  72. void MockSourceStream::CompleteNextRead() {
  73. DCHECK(awaiting_completion_);
  74. awaiting_completion_ = false;
  75. QueuedResult r = results_.front();
  76. DCHECK_EQ(ASYNC, r.mode);
  77. results_.pop();
  78. DCHECK_GE(dest_buffer_size_, r.len);
  79. memcpy(dest_buffer_->data(), r.data, r.len);
  80. dest_buffer_ = nullptr;
  81. std::move(callback_).Run(r.error == OK ? r.len : r.error);
  82. }
  83. } // namespace net