Implement WebDataConsumerHandle::Reader.

content/child/web_data_consumer_handle_impl_unittest.cc

 // Copyright 2014 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 "content/child/web_data_consumer_handle_impl.h"
 
 #include <algorithm>
 #include <string>
 #include "base/bind.h"
 #include "base/location.h"
@@ skipping 51 matching lines @@
         main_message_loop_(main_message_loop),
         on_done_(on_done) {}
 
   const std::string& result() const { return result_; }
 
   void ReadMore() override { ReadData(); }
 
   void ReadData() {
     if (!client_) {
       client_.reset(new ClientImpl(this));
-      handle_->registerClient(client_.get());
+      reader_ = handle_->ObtainReader(client_.get());
     }
 
     Result rv = kOk;
     size_t readSize = 0;
 
     while (true) {
       char buffer[16];
-      rv = handle_->read(&buffer, sizeof(buffer), kNone, &readSize);
+      rv = reader_->read(&buffer, sizeof(buffer), kNone, &readSize);
       if (rv != kOk)
         break;
       result_.insert(result_.size(), &buffer[0], readSize);
     }
 
     if (rv == kShouldWait) {
       // Wait a while...
       return;
     }
 
     if (rv != kDone) {
       // Something is wrong.
       result_ = "error";
     }
 
     // The operation is done.
+    reader_.reset();
     main_message_loop_->task_runner()->PostTask(FROM_HERE, on_done_);
   }
 
  private:
-  scoped_ptr<WebDataConsumerHandle> handle_;
+  scoped_ptr<WebDataConsumerHandleImpl> handle_;
+  scoped_ptr<WebDataConsumerHandle::Reader> reader_;
   scoped_ptr<WebDataConsumerHandle::Client> client_;
   base::MessageLoop* main_message_loop_;
   base::Closure on_done_;
   std::string result_;
 };
 
 class TwoPhaseReadDataOperation : public ReadDataOperationBase {
  public:
   typedef WebDataConsumerHandle::Result Result;
   TwoPhaseReadDataOperation(mojo::ScopedDataPipeConsumerHandle handle,
                             base::MessageLoop* main_message_loop,
                             const base::Closure& on_done)
       : handle_(new WebDataConsumerHandleImpl(handle.Pass())),
         main_message_loop_(main_message_loop), on_done_(on_done) {}
 
   const std::string& result() const { return result_; }
 
   void ReadMore() override {
     ReadData();
   }
 
   void ReadData() {
     if (!client_) {
       client_.reset(new ClientImpl(this));
-      handle_->registerClient(client_.get());
+      reader_ = handle_->ObtainReader(client_.get());
     }
 
     Result rv;
     while (true) {
       const void* buffer = nullptr;
       size_t size;
-      rv = handle_->beginRead(&buffer, kNone, &size);
+      rv = reader_->beginRead(&buffer, kNone, &size);
       if (rv != kOk)
         break;
       // In order to verify endRead, we read at most one byte for each time.
       size_t read_size = std::max(static_cast<size_t>(1), size);
       result_.insert(result_.size(), static_cast<const char*>(buffer),
                      read_size);
-      rv = handle_->endRead(read_size);
+      rv = reader_->endRead(read_size);
       if (rv != kOk) {
         // Something is wrong.
         result_ = "error";
         main_message_loop_->task_runner()->PostTask(FROM_HERE, on_done_);
         return;
       }
     }
 
     if (rv == kShouldWait) {
       // Wait a while...
       return;
     }
 
     if (rv != kDone) {
       // Something is wrong.
       result_ = "error";
     }
 
     // The operation is done.
+    reader_.reset();
     main_message_loop_->task_runner()->PostTask(FROM_HERE, on_done_);
   }
 
  private:
-  scoped_ptr<WebDataConsumerHandle> handle_;
+  scoped_ptr<WebDataConsumerHandleImpl> handle_;
+  scoped_ptr<WebDataConsumerHandle::Reader> reader_;
   scoped_ptr<WebDataConsumerHandle::Client> client_;
   base::MessageLoop* main_message_loop_;
   base::Closure on_done_;
   std::string result_;
 };
 
 class WebDataConsumerHandleImplTest : public ::testing::Test {
  public:
   typedef WebDataConsumerHandle::Result Result;
 
@@ skipping 85 matching lines @@
 
   run_loop.Run();
   t.Stop();
 
   EXPECT_EQ(expected, operation->result());
 }
 
 }  // namespace
 
 }  // namespace content