Fix WebSocket to set first party for cookies

net/websockets/websocket_stream_create_test_base.cc

 // Copyright 2015 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/websockets/websocket_stream_create_test_base.h"
 
 #include <utility>
 
 #include "base/callback.h"
 #include "base/macros.h"
@@ skipping 80 matching lines @@
 };
 
 WebSocketStreamCreateTestBase::WebSocketStreamCreateTestBase()
     : has_failed_(false), ssl_fatal_(false) {
 }
 
 WebSocketStreamCreateTestBase::~WebSocketStreamCreateTestBase() {
 }
 
 void WebSocketStreamCreateTestBase::CreateAndConnectStream(
-    const std::string& socket_url,
+    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) {
   for (size_t i = 0; i < ssl_data_.size(); ++i) {
     url_request_context_host_.AddSSLSocketDataProvider(std::move(ssl_data_[i]));
   }
   ssl_data_.clear();
   std::unique_ptr<WebSocketStream::ConnectDelegate> connect_delegate(
       new TestConnectDelegate(this, connect_run_loop_.QuitClosure()));
   WebSocketStream::ConnectDelegate* delegate = connect_delegate.get();
   std::unique_ptr<WebSocketHandshakeStreamCreateHelper> create_helper(
       new DeterministicKeyWebSocketHandshakeStreamCreateHelper(delegate,
                                                                sub_protocols));
   stream_request_ = CreateAndConnectStreamForTesting(
-      GURL(socket_url), std::move(create_helper), origin, additional_headers,
-      url_request_context_host_.GetURLRequestContext(), BoundNetLog(),
-      std::move(connect_delegate),
+      socket_url, std::move(create_helper), origin, first_party_for_cookies,
+      additional_headers, url_request_context_host_.GetURLRequestContext(),
+      BoundNetLog(), std::move(connect_delegate),
       timer ? std::move(timer)
             : std::unique_ptr<base::Timer>(new base::Timer(false, false)));
 }
 
 std::vector<HeaderKeyValuePair>
 WebSocketStreamCreateTestBase::RequestHeadersToVector(
     const HttpRequestHeaders& headers) {
   HttpRequestHeaders::Iterator it(headers);
   std::vector<HeaderKeyValuePair> result;
   while (it.GetNext())
@@ skipping 14 matching lines @@
 
 void WebSocketStreamCreateTestBase::WaitUntilConnectDone() {
   connect_run_loop_.Run();
 }
 
 std::vector<std::string> WebSocketStreamCreateTestBase::NoSubProtocols() {
   return std::vector<std::string>();
 }
 
 }  // namespace net