Cancel URLRequest with an appropriate code when WebSocket handshake succeeds

net/websockets/websocket_test_util.h

 // Copyright 2013 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.
 
 #ifndef NET_WEBSOCKETS_WEBSOCKET_TEST_UTIL_H_
 #define NET_WEBSOCKETS_WEBSOCKET_TEST_UTIL_H_
 
 #include <stdint.h>
 
 #include <memory>
@@ skipping 108 matching lines @@
   // Allow a proxy to be set. Usage:
   //   SetProxyConfig("proxy1:8000");
   // Any syntax accepted by net::ProxyConfig::ParseFromString() will work.
   // Do not call after GetURLRequestContext() has been called.
   void SetProxyConfig(const std::string& proxy_rules);
 
   // Call after calling one of SetExpections() or AddRawExpectations(). The
   // returned pointer remains owned by this object.
   TestURLRequestContext* GetURLRequestContext();
 
+  const TestNetworkDelegate& network_delegate() const {
+    return network_delegate_;
+  }
+
  private:
   WebSocketMockClientSocketFactoryMaker maker_;
   TestURLRequestContext url_request_context_;
   TestNetworkDelegate network_delegate_;
   std::unique_ptr<ProxyService> proxy_service_;
   bool url_request_context_initialized_;
 
   DISALLOW_COPY_AND_ASSIGN(WebSocketTestURLRequestContextHost);
 };
 
 }  // namespace net
 
 #endif  // NET_WEBSOCKETS_WEBSOCKET_TEST_UTIL_H_