net: Make UploadDataStream::Read() asynchronous

net/base/upload_data_stream_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "net/base/upload_data_stream.h"
 
+#include <algorithm>
 #include <vector>
 
 #include "base/basictypes.h"
 #include "base/bind.h"
 #include "base/file_path.h"
 #include "base/file_util.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/message_loop.h"
 #include "base/time.h"
 #include "net/base/io_buffer.h"
@@ skipping 15 matching lines @@
 
 const char kTestData[] = "0123456789";
 const size_t kTestDataSize = arraysize(kTestData) - 1;
 const size_t kTestBufferSize = 1 << 14;  // 16KB.
 
 // Reads data from the upload data stream, and returns the data as string.
 std::string ReadFromUploadDataStream(UploadDataStream* stream) {
   std::string data_read;
   scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize);
   while (!stream->IsEOF()) {
-    const int bytes_read = stream->Read(buf, kTestBufferSize);
+    const int bytes_read = stream->ReadSync(buf, kTestBufferSize);
     data_read.append(buf->data(), bytes_read);
   }
   return data_read;
 }
 
 // A mock class of UploadElementReader.
 class MockUploadElementReader : public UploadElementReader {
  public:
-  MockUploadElementReader() : init_result_(OK) {}
+  MockUploadElementReader(int content_length, bool is_in_memory)
+      : content_length_(content_length),
+        bytes_remaining_(content_length),
+        is_in_memory_(is_in_memory),
+        init_result_(OK),
+        read_result_(OK) {}
+
   virtual ~MockUploadElementReader() {}
 
   MOCK_METHOD1(Init, int(const CompletionCallback& callback));
-  MOCK_CONST_METHOD0(GetContentLength, uint64());
-  MOCK_CONST_METHOD0(BytesRemaining, uint64());
-  MOCK_CONST_METHOD0(IsInMemory, bool());
-  MOCK_METHOD2(ReadSync, int(char* buf, int buf_length));
+  uint64 GetContentLength() const OVERRIDE { return content_length_; }
+  uint64 BytesRemaining() const OVERRIDE { return bytes_remaining_; }
+  bool IsInMemory() const OVERRIDE { return is_in_memory_; }
+  MOCK_METHOD3(Read, int(IOBuffer* buf,
+                         int buf_length,
+                         const CompletionCallback& callback));
 
   // Sets expectation to return the specified result from Init() asynchronously.
   void SetAsyncInitExpectation(int result) {
     init_result_ = result;
     EXPECT_CALL(*this, Init(_))
         .WillOnce(DoAll(Invoke(this, &MockUploadElementReader::OnInit),
                         Return(ERR_IO_PENDING)));
   }
 
+  // Sets expectation to return the specified result from Read().
+  void SetReadExpectation(int result) {
+    read_result_ = result;
+    EXPECT_CALL(*this, Read(_, _, _))
+        .WillOnce(Invoke(this, &MockUploadElementReader::OnRead));
+  }
+
  private:
   void OnInit(const CompletionCallback& callback) {
     MessageLoop::current()->PostTask(FROM_HERE,
                                      base::Bind(callback, init_result_));
   }
 
+  int OnRead(IOBuffer* buf,
+             int buf_length,
+             const CompletionCallback& callback) {
+    bytes_remaining_ = std::max(0, bytes_remaining_ - read_result_);
+    if (IsInMemory()) {
+      return read_result_;
+    } else {
+      MessageLoop::current()->PostTask(FROM_HERE,
+                                       base::Bind(callback, read_result_));
+      return ERR_IO_PENDING;
+    }
+  }
+
+  int content_length_;
+  int bytes_remaining_;
+  bool is_in_memory_;
+
   // Result value returned from Init().
   int init_result_;
+
+  // Result value returned from Read().
+  int read_result_;
 };
 
 // A mock CompletionCallback.
 class MockCompletionCallback {
  public:
   MOCK_METHOD1(Run, void(int result));
 
   CompletionCallback CreateCallback() {
     return base::Bind(&MockCompletionCallback::Run, base::Unretained(this));
   }
@@ skipping 27 matching lines @@
   upload_data_->AppendBytes(kTestData, kTestDataSize);
   scoped_ptr<UploadDataStream> stream(new UploadDataStream(upload_data_));
   ASSERT_EQ(OK, stream->InitSync());
   EXPECT_TRUE(stream->IsInMemory());
   ASSERT_TRUE(stream.get());
   EXPECT_EQ(kTestDataSize, stream->size());
   EXPECT_EQ(0U, stream->position());
   EXPECT_FALSE(stream->IsEOF());
   scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize);
   while (!stream->IsEOF()) {
-    int bytes_read = stream->Read(buf, kTestBufferSize);
+    int bytes_read = stream->ReadSync(buf, kTestBufferSize);
     ASSERT_LE(0, bytes_read);  // Not an error.
   }
   EXPECT_EQ(kTestDataSize, stream->position());
   ASSERT_TRUE(stream->IsEOF());
 }
 
 TEST_F(UploadDataStreamTest, File) {
   FilePath temp_file_path;
   ASSERT_TRUE(file_util::CreateTemporaryFile(&temp_file_path));
   ASSERT_EQ(static_cast<int>(kTestDataSize),
             file_util::WriteFile(temp_file_path, kTestData, kTestDataSize));
 
   std::vector<UploadElement> elements;
   UploadElement element;
   element.SetToFilePath(temp_file_path);
   elements.push_back(element);
   upload_data_->SetElements(elements);
 
   scoped_ptr<UploadDataStream> stream(new UploadDataStream(upload_data_));
   ASSERT_EQ(OK, stream->InitSync());
   EXPECT_FALSE(stream->IsInMemory());
   ASSERT_TRUE(stream.get());
   EXPECT_EQ(kTestDataSize, stream->size());
   EXPECT_EQ(0U, stream->position());
   EXPECT_FALSE(stream->IsEOF());
   scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize);
   while (!stream->IsEOF()) {
-    int bytes_read = stream->Read(buf, kTestBufferSize);
+    int bytes_read = stream->ReadSync(buf, kTestBufferSize);
     ASSERT_LE(0, bytes_read);  // Not an error.
   }
   EXPECT_EQ(kTestDataSize, stream->position());
   ASSERT_TRUE(stream->IsEOF());
   file_util::Delete(temp_file_path, false);
 }
 
 TEST_F(UploadDataStreamTest, FileSmallerThanLength) {
   FilePath temp_file_path;
   ASSERT_TRUE(file_util::CreateTemporaryFile(&temp_file_path));
@@ skipping 13 matching lines @@
   scoped_ptr<UploadDataStream> stream(new UploadDataStream(upload_data_));
   ASSERT_EQ(OK, stream->InitSync());
   EXPECT_FALSE(stream->IsInMemory());
   ASSERT_TRUE(stream.get());
   EXPECT_EQ(kFakeSize, stream->size());
   EXPECT_EQ(0U, stream->position());
   EXPECT_FALSE(stream->IsEOF());
   uint64 read_counter = 0;
   scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize);
   while (!stream->IsEOF()) {
-    int bytes_read = stream->Read(buf, kTestBufferSize);
+    int bytes_read = stream->ReadSync(buf, kTestBufferSize);
     ASSERT_LE(0, bytes_read);  // Not an error.
     read_counter += bytes_read;
     EXPECT_EQ(read_counter, stream->position());
   }
   // UpdateDataStream will pad out the file with 0 bytes so that the HTTP
   // transaction doesn't hang.  Therefore we expected the full size.
   EXPECT_EQ(kFakeSize, read_counter);
   EXPECT_EQ(read_counter, stream->position());
 
   file_util::Delete(temp_file_path, false);
@@ skipping 15 matching lines @@
   const uint64 kStreamSize = kTestDataSize + kFileRangeLength;
   scoped_ptr<UploadDataStream> stream(new UploadDataStream(upload_data_));
   ASSERT_EQ(OK, stream->InitSync());
   EXPECT_FALSE(stream->IsInMemory());
   ASSERT_TRUE(stream.get());
   EXPECT_EQ(kStreamSize, stream->size());
   EXPECT_EQ(0U, stream->position());
   EXPECT_FALSE(stream->IsEOF());
   scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize);
   while (!stream->IsEOF()) {
-    int bytes_read = stream->Read(buf, kTestBufferSize);
+    int bytes_read = stream->ReadSync(buf, kTestBufferSize);
     ASSERT_LE(0, bytes_read);  // Not an error.
   }
   EXPECT_EQ(kStreamSize, stream->position());
   ASSERT_TRUE(stream->IsEOF());
 
   file_util::Delete(temp_file_path, false);
 }
 
 TEST_F(UploadDataStreamTest, Chunk) {
   upload_data_->set_is_chunked(true);
   upload_data_->AppendChunk(kTestData, kTestDataSize, false);
   upload_data_->AppendChunk(kTestData, kTestDataSize, true);
 
   const uint64 kStreamSize = kTestDataSize*2;
   scoped_ptr<UploadDataStream> stream(new UploadDataStream(upload_data_));
   ASSERT_EQ(OK, stream->InitSync());
   EXPECT_FALSE(stream->IsInMemory());
   ASSERT_TRUE(stream.get());
   EXPECT_EQ(0U, stream->size());  // Content-Length is 0 for chunked data.
   EXPECT_EQ(0U, stream->position());
   EXPECT_FALSE(stream->IsEOF());
   scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize);
   while (!stream->IsEOF()) {
-    int bytes_read = stream->Read(buf, kTestBufferSize);
+    int bytes_read = stream->ReadSync(buf, kTestBufferSize);
     ASSERT_LE(0, bytes_read);  // Not an error.
   }
   EXPECT_EQ(kStreamSize, stream->position());
   ASSERT_TRUE(stream->IsEOF());
 }
 
 // Init() with on-memory and not-on-memory readers.
 TEST_F(UploadDataStreamTest, InitAsync) {
   // Create stream without element readers.
   UploadDataStream stream(upload_data_);
 
   // Set mock readers to the stream.
   MockUploadElementReader* reader = NULL;
 
-  reader = new MockUploadElementReader;
+  reader = new MockUploadElementReader(kTestDataSize, true);
   EXPECT_CALL(*reader, Init(_)).WillOnce(Return(OK));
-  EXPECT_CALL(*reader, GetContentLength()).WillRepeatedly(Return(0));
-  EXPECT_CALL(*reader, IsInMemory()).WillRepeatedly(Return(true));
   stream.element_readers_.push_back(reader);
 
-  reader = new MockUploadElementReader;
+  reader = new MockUploadElementReader(kTestDataSize, true);
   EXPECT_CALL(*reader, Init(_)).WillOnce(Return(OK));
-  EXPECT_CALL(*reader, GetContentLength()).WillRepeatedly(Return(0));
-  EXPECT_CALL(*reader, IsInMemory()).WillRepeatedly(Return(true));
   stream.element_readers_.push_back(reader);
 
-  reader = new MockUploadElementReader;
+  reader = new MockUploadElementReader(kTestDataSize, false);
   reader->SetAsyncInitExpectation(OK);
-  EXPECT_CALL(*reader, GetContentLength()).WillRepeatedly(Return(0));
-  EXPECT_CALL(*reader, IsInMemory()).WillRepeatedly(Return(false));
   stream.element_readers_.push_back(reader);
 
-  reader = new MockUploadElementReader;
+  reader = new MockUploadElementReader(kTestDataSize, false);
   reader->SetAsyncInitExpectation(OK);
-  EXPECT_CALL(*reader, GetContentLength()).WillRepeatedly(Return(0));
-  EXPECT_CALL(*reader, IsInMemory()).WillRepeatedly(Return(false));
   stream.element_readers_.push_back(reader);
 
-  reader = new MockUploadElementReader;
+  reader = new MockUploadElementReader(kTestDataSize, true);
   EXPECT_CALL(*reader, Init(_)).WillOnce(Return(OK));
-  EXPECT_CALL(*reader, GetContentLength()).WillRepeatedly(Return(0));
-  EXPECT_CALL(*reader, IsInMemory()).WillRepeatedly(Return(true));
   stream.element_readers_.push_back(reader);
 
   // Run Init().
   MockCompletionCallback mock_callback;
   EXPECT_CALL(mock_callback, Run(OK)).Times(1);
   EXPECT_EQ(stream.Init(mock_callback.CreateCallback()), ERR_IO_PENDING);
   MessageLoop::current()->RunAllPending();
 }
 
 // Init() of a reader fails asynchronously.
 TEST_F(UploadDataStreamTest, InitAsyncFailureAsync) {
   // Create stream without element readers.
   UploadDataStream stream(upload_data_);
 
   // Set a mock reader to the stream.
   MockUploadElementReader* reader = NULL;
 
-  reader = new MockUploadElementReader;
+  reader = new MockUploadElementReader(kTestDataSize, false);
   reader->SetAsyncInitExpectation(ERR_FAILED);
-  EXPECT_CALL(*reader, IsInMemory()).WillRepeatedly(Return(false));
   stream.element_readers_.push_back(reader);
 
   // Run Init().
   MockCompletionCallback mock_callback;
   EXPECT_CALL(mock_callback, Run(ERR_FAILED)).Times(1);
   EXPECT_EQ(stream.Init(mock_callback.CreateCallback()), ERR_IO_PENDING);
   MessageLoop::current()->RunAllPending();
 }
 
 // Init() of a reader fails synchronously.
 TEST_F(UploadDataStreamTest, InitAsyncFailureSync) {
   // Create stream without element readers.
   UploadDataStream stream(upload_data_);
 
   // Set mock readers to the stream.
   MockUploadElementReader* reader = NULL;
 
-  reader = new MockUploadElementReader;
+  reader = new MockUploadElementReader(kTestDataSize, false);
   reader->SetAsyncInitExpectation(OK);
-  EXPECT_CALL(*reader, IsInMemory()).WillRepeatedly(Return(false));
   stream.element_readers_.push_back(reader);
 
-  reader = new MockUploadElementReader;
+  reader = new MockUploadElementReader(kTestDataSize, true);
   EXPECT_CALL(*reader, Init(_)).WillOnce(Return(ERR_FAILED));
-  EXPECT_CALL(*reader, IsInMemory()).WillRepeatedly(Return(true));
   stream.element_readers_.push_back(reader);
 
   // Run Init().
   MockCompletionCallback mock_callback;
   EXPECT_CALL(mock_callback, Run(ERR_FAILED)).Times(1);
   EXPECT_EQ(stream.Init(mock_callback.CreateCallback()), ERR_IO_PENDING);
   MessageLoop::current()->RunAllPending();
 }
 
+// Read() with on-memory and not-on-memory readers.
+TEST_F(UploadDataStreamTest, ReadAsync) {
+  // Create stream without element readers.
+  UploadDataStream stream(upload_data_);
+
+  // Set mock readers to the stream.
+  MockUploadElementReader* reader = NULL;
+
+  reader = new MockUploadElementReader(kTestDataSize, true);
+  EXPECT_CALL(*reader, Init(_)).WillOnce(Return(OK));
+  reader->SetReadExpectation(kTestDataSize);
+  stream.element_readers_.push_back(reader);
+
+  reader = new MockUploadElementReader(kTestDataSize, false);
+  reader->SetAsyncInitExpectation(OK);
+  reader->SetReadExpectation(kTestDataSize);
+  stream.element_readers_.push_back(reader);
+
+  reader = new MockUploadElementReader(kTestDataSize, true);
+  EXPECT_CALL(*reader, Init(_)).WillOnce(Return(OK));
+  reader->SetReadExpectation(kTestDataSize);
+  stream.element_readers_.push_back(reader);
+
+  reader = new MockUploadElementReader(kTestDataSize, true);
+  reader->SetAsyncInitExpectation(OK);
+  reader->SetReadExpectation(kTestDataSize);
+  stream.element_readers_.push_back(reader);
+
+  // Run Init().
+  MockCompletionCallback mock_callback;
+  EXPECT_CALL(mock_callback, Run(OK)).Times(1);
+  EXPECT_EQ(ERR_IO_PENDING, stream.Init(mock_callback.CreateCallback()));
+  MessageLoop::current()->RunAllPending();
+
+  scoped_refptr<IOBuffer> buf = new IOBuffer(kTestBufferSize);
+
+  // Consume the first element.
+  EXPECT_CALL(mock_callback, Run(kTestDataSize)).Times(1);
+  EXPECT_EQ(ERR_IO_PENDING,
+            stream.Read(buf, kTestDataSize, mock_callback.CreateCallback()));
+  MessageLoop::current()->RunAllPending();
+
+  // Consume the second element.
+  EXPECT_CALL(mock_callback, Run(kTestDataSize)).Times(1);
+  EXPECT_EQ(ERR_IO_PENDING,
+            stream.Read(buf, kTestDataSize, mock_callback.CreateCallback()));
+  MessageLoop::current()->RunAllPending();
+
+  // Consume the third and the fourth elements.
+  EXPECT_CALL(mock_callback, Run(kTestDataSize*2)).Times(1);
+  EXPECT_EQ(ERR_IO_PENDING,
+            stream.Read(buf, kTestDataSize*2, mock_callback.CreateCallback()));
+  MessageLoop::current()->RunAllPending();
+}
+
 void UploadDataStreamTest::FileChangedHelper(const FilePath& file_path,
                                              const base::Time& time,
                                              bool error_expected) {
   std::vector<UploadElement> elements;
   UploadElement element;
   element.SetToFilePathRange(file_path, 1, 2, time);
   elements.push_back(element);
   // Don't use upload_data_ here, as this function is called twice, and
   // reusing upload_data_ is wrong.
   scoped_refptr<UploadData> upload_data(new UploadData);
@@ skipping 54 matching lines @@
     UploadDataStream stream(upload_data_);
     ASSERT_EQ(OK, stream.InitSync());
     EXPECT_EQ(kTestDataSize, stream.size());
     EXPECT_EQ(kTestData, ReadFromUploadDataStream(&stream));
   }
 
   file_util::Delete(temp_file_path, false);
 }
 
 }  // namespace net