URLRequestJob: change ReadRawData contract

net/http/http_transaction_test_util.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 "net/http/http_transaction_test_util.h"
 
 #include <algorithm>
 
 #include "base/bind.h"
 #include "base/location.h"
@@ skipping 226 matching lines @@
 MockNetworkTransaction::MockNetworkTransaction(RequestPriority priority,
                                                MockNetworkLayer* factory)
     : request_(NULL),
       data_cursor_(0),
       priority_(priority),
       websocket_handshake_stream_create_helper_(NULL),
       transaction_factory_(factory->AsWeakPtr()),
       received_bytes_(0),
       sent_bytes_(0),
       socket_log_id_(NetLog::Source::kInvalidId),
+      done_reading_called_(false),
       weak_factory_(this) {}
 
 MockNetworkTransaction::~MockNetworkTransaction() {}
 
 int MockNetworkTransaction::Start(const HttpRequestInfo* request,
                                   const CompletionCallback& callback,
                                   const BoundNetLog& net_log) {
   if (request_)
     return ERR_FAILED;
 
@@ skipping 37 matching lines @@
 
   // Allow the mock server to decide whether authentication is required or not.
   std::string status_line = response_.headers->GetStatusLine();
   return status_line.find(" 401 ") != std::string::npos ||
       status_line.find(" 407 ") != std::string::npos;
 }
 
 int MockNetworkTransaction::Read(IOBuffer* buf,
                                  int buf_len,
                                  const CompletionCallback& callback) {
+  CHECK(!done_reading_called_);
   int data_len = static_cast<int>(data_.size());
   int num = std::min(buf_len, data_len - data_cursor_);
   if (test_mode_ & TEST_MODE_SLOW_READ)
     num = std::min(num, 1);
   if (num) {
     memcpy(buf->data(), data_.data() + data_cursor_, num);
     data_cursor_ += num;
   }
   if (test_mode_ & TEST_MODE_SYNC_NET_READ)
     return num;
@@ skipping 14 matching lines @@
 
 int64_t MockNetworkTransaction::GetTotalReceivedBytes() const {
   return received_bytes_;
 }
 
 int64_t MockNetworkTransaction::GetTotalSentBytes() const {
   return sent_bytes_;
 }
 
 void MockNetworkTransaction::DoneReading() {
+  CHECK(!done_reading_called_);
+  done_reading_called_ = true;
   if (transaction_factory_.get())
     transaction_factory_->TransactionDoneReading();
 }
 
 const HttpResponseInfo* MockNetworkTransaction::GetResponseInfo() const {
   return &response_;
 }
 
 LoadState MockNetworkTransaction::GetLoadState() const {
   if (data_cursor_)
@@ skipping 151 matching lines @@
     : transaction_count_(0),
       done_reading_called_(false),
       stop_caching_called_(false),
       last_create_transaction_priority_(DEFAULT_PRIORITY),
       clock_(nullptr) {
 }
 
 MockNetworkLayer::~MockNetworkLayer() {}
 
 void MockNetworkLayer::TransactionDoneReading() {
+  CHECK(!done_reading_called_);
   done_reading_called_ = true;
 }
 
 void MockNetworkLayer::TransactionStopCaching() {
   stop_caching_called_ = true;
 }
 
 int MockNetworkLayer::CreateTransaction(RequestPriority priority,
                                         scoped_ptr<HttpTransaction>* trans) {
   transaction_count_++;
@@ skipping 43 matching lines @@
       content.append(buf->data(), rv);
     else if (rv < 0)
       return rv;
   } while (rv > 0);
 
   result->swap(content);
   return OK;
 }
 
 }  // namespace net