Changes to HttpNetworkTransaction for WebSocket Handshake

net/http/http_transaction_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/http/http_transaction_unittest.h"
 
 #include <algorithm>
 
 #include "base/bind.h"
 #include "base/message_loop/message_loop.h"
@@ skipping 210 matching lines @@
       NOTREACHED();
   }
 }
 
 MockNetworkTransaction::MockNetworkTransaction(
     net::RequestPriority priority,
     MockNetworkLayer* factory)
     : weak_factory_(this),
       data_cursor_(0),
       priority_(priority),
+      websocket_handshake_stream_factory_(NULL),
       transaction_factory_(factory->AsWeakPtr()),
       socket_log_id_(net::NetLog::Source::kInvalidId) {
 }
 
 MockNetworkTransaction::~MockNetworkTransaction() {}
 
 int MockNetworkTransaction::Start(const net::HttpRequestInfo* request,
                                   const net::CompletionCallback& callback,
                                   const net::BoundNetLog& net_log) {
   const MockTransaction* t = FindMockTransaction(request->url);
@@ skipping 127 matching lines @@
     load_timing_info->send_start = base::TimeTicks::Now();
     load_timing_info->send_end = base::TimeTicks::Now();
   }
   return true;
 }
 
 void MockNetworkTransaction::SetPriority(net::RequestPriority priority) {
   priority_ = priority;
 }
 
+void MockNetworkTransaction::SetWebSocketHandshakeStreamFactory(
+    net::WebSocketHandshakeStreamBase::Factory* factory) {
+  websocket_handshake_stream_factory_ = factory;
+}
+
 void MockNetworkTransaction::CallbackLater(
     const net::CompletionCallback& callback, int result) {
   base::MessageLoop::current()->PostTask(
       FROM_HERE, base::Bind(&MockNetworkTransaction::RunCallback,
                             weak_factory_.GetWeakPtr(), callback, result));
 }
 
 void MockNetworkTransaction::RunCallback(
     const net::CompletionCallback& callback, int result) {
   callback.Run(result);
@@ skipping 48 matching lines @@
 
     if (rv > 0)
       content.append(buf->data(), rv);
     else if (rv < 0)
       return rv;
   } while (rv > 0);
 
   result->swap(content);
   return net::OK;
 }