Fix WebSocket to set first party for cookies

BUG=618962
R=mkwst,yhirano

Review-Url: https://codereview.chromium.org/2102993002
Cr-Commit-Position: refs/heads/master@{#405057}
diff --git a/net/websockets/websocket_stream_create_test_base.h b/net/websockets/websocket_stream_create_test_base.h
index 1cdc4b40..3d6a7cf 100644
--- a/net/websockets/websocket_stream_create_test_base.h
+++ b/net/websockets/websocket_stream_create_test_base.h
@@ -19,6 +19,8 @@
 #include "net/websockets/websocket_event_interface.h"
 #include "net/websockets/websocket_test_util.h"
 
+class GURL;
+
 namespace net {
 
 class HttpRequestHeaders;
@@ -37,9 +39,10 @@
 
   // A wrapper for CreateAndConnectStreamForTesting that knows about our default
   // parameters.
-  void CreateAndConnectStream(const std::string& socket_url,
+  void CreateAndConnectStream(const GURL& socket_url,
                               const std::vector<std::string>& sub_protocols,
                               const url::Origin& origin,
+                              const GURL& first_party_for_cookies,
                               const std::string& additional_headers,
                               std::unique_ptr<base::Timer> timer);