| Index: net/http/http_transaction_unittest.cc
|
| ===================================================================
|
| --- net/http/http_transaction_unittest.cc (revision 95711)
|
| +++ net/http/http_transaction_unittest.cc (working copy)
|
| @@ -212,8 +212,10 @@
|
| }
|
|
|
|
|
| -MockNetworkTransaction::MockNetworkTransaction() :
|
| - ALLOW_THIS_IN_INITIALIZER_LIST(task_factory_(this)), data_cursor_(0) {
|
| +MockNetworkTransaction::MockNetworkTransaction(MockNetworkLayer* factory)
|
| + : ALLOW_THIS_IN_INITIALIZER_LIST(task_factory_(this)),
|
| + data_cursor_(0),
|
| + transaction_factory_(factory->AsWeakPtr()) {
|
| }
|
|
|
| MockNetworkTransaction::~MockNetworkTransaction() {}
|
| @@ -295,6 +297,11 @@
|
|
|
| void MockNetworkTransaction::StopCaching() {}
|
|
|
| +void MockNetworkTransaction::DoneReading() {
|
| + if (transaction_factory_)
|
| + transaction_factory_->TransactionDoneReading();
|
| +}
|
| +
|
| const net::HttpResponseInfo* MockNetworkTransaction::GetResponseInfo() const {
|
| return &response_;
|
| }
|
| @@ -320,14 +327,19 @@
|
| callback->Run(result);
|
| }
|
|
|
| -MockNetworkLayer::MockNetworkLayer() : transaction_count_(0) {}
|
| +MockNetworkLayer::MockNetworkLayer()
|
| + : transaction_count_(0), done_reading_called_(false) {}
|
|
|
| MockNetworkLayer::~MockNetworkLayer() {}
|
|
|
| +void MockNetworkLayer::TransactionDoneReading() {
|
| + done_reading_called_ = true;
|
| +}
|
| +
|
| int MockNetworkLayer::CreateTransaction(
|
| scoped_ptr<net::HttpTransaction>* trans) {
|
| transaction_count_++;
|
| - trans->reset(new MockNetworkTransaction());
|
| + trans->reset(new MockNetworkTransaction(this));
|
| return net::OK;
|
| }
|
|
|
|
|