Changes to HttpNetworkTransaction for WebSocket Handshake

net/http/http_transaction_unittest.h

Index: net/http/http_transaction_unittest.h
diff --git a/net/http/http_transaction_unittest.h b/net/http/http_transaction_unittest.h
index 407fc9fff9a1587558f5c48165461345db5b6c33..3e8adaf7b9a8591e6a553e212aa9e248300f6cae 100644
--- a/net/http/http_transaction_unittest.h
+++ b/net/http/http_transaction_unittest.h
@@ -162,6 +162,7 @@ class MockNetworkLayer;
 class MockNetworkTransaction
     : public net::HttpTransaction,
       public base::SupportsWeakPtr<MockNetworkTransaction> {
+  typedef net::WebSocketHandshakeStreamBase::Factory WebSocketFactory;
  public:
   MockNetworkTransaction(net::RequestPriority priority,
                          MockNetworkLayer* factory);
@@ -205,6 +206,12 @@ class MockNetworkTransaction
 
   virtual void SetPriority(net::RequestPriority priority) OVERRIDE;
 
+  virtual void SetWebSocketHandshakeStreamFactory(
+      WebSocketFactory* factory) OVERRIDE;
+
+  WebSocketFactory* websocket_handshake_stream_factory() {
+    return websocket_handshake_stream_factory_;
+  }
   net::RequestPriority priority() const { return priority_; }
 
  private:
@@ -217,6 +224,7 @@ class MockNetworkTransaction
   int data_cursor_;
   int test_mode_;
   net::RequestPriority priority_;
+  WebSocketFactory* websocket_handshake_stream_factory_;
   base::WeakPtr<MockNetworkLayer> transaction_factory_;
 
   // NetLog ID of the fake / non-existent underlying socket used by the