15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/socket_stream/socket_stream.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <string> 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <vector> 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/bind.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/bind_helpers.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/callback.h" 13868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/utf_string_conversions.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/auth.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/net_log.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/net_log_unittest.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/test_completion_callback.h" 182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "net/dns/mock_host_resolver.h" 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/http/http_network_session.h" 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/proxy/proxy_service.h" 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/socket/socket_test_util.h" 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/url_request/url_request_test_util.h" 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/platform_test.h" 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)namespace net { 272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)struct SocketStreamEvent { 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) enum EventType { 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EVENT_START_OPEN_CONNECTION, EVENT_CONNECTED, EVENT_SENT_DATA, 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EVENT_RECEIVED_DATA, EVENT_CLOSE, EVENT_AUTH_REQUIRED, EVENT_ERROR, 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketStreamEvent(EventType type, 372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SocketStream* socket_stream, 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int num, 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::string& str, 402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) AuthChallengeInfo* auth_challenge_info, 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int error) 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : event_type(type), socket(socket_stream), number(num), data(str), 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) auth_info(auth_challenge_info), error_code(error) {} 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EventType event_type; 462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SocketStream* socket; 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int number; 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string data; 492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_refptr<AuthChallengeInfo> auth_info; 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int error_code; 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)class SocketStreamEventRecorder : public SocketStream::Delegate { 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 557d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) // |callback| will be run when the OnClose() or OnError() method is called. 567d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) // For OnClose(), |callback| is called with OK. For OnError(), it's called 577d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) // with the error code. 582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) explicit SocketStreamEventRecorder(const CompletionCallback& callback) 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : callback_(callback) {} 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~SocketStreamEventRecorder() {} 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SetOnStartOpenConnection( 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const base::Callback<int(SocketStreamEvent*)>& callback) { 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) on_start_open_connection_ = callback; 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SetOnConnected( 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const base::Callback<void(SocketStreamEvent*)>& callback) { 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) on_connected_ = callback; 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SetOnSentData( 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const base::Callback<void(SocketStreamEvent*)>& callback) { 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) on_sent_data_ = callback; 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SetOnReceivedData( 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const base::Callback<void(SocketStreamEvent*)>& callback) { 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) on_received_data_ = callback; 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SetOnClose(const base::Callback<void(SocketStreamEvent*)>& callback) { 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) on_close_ = callback; 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SetOnAuthRequired( 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const base::Callback<void(SocketStreamEvent*)>& callback) { 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) on_auth_required_ = callback; 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SetOnError(const base::Callback<void(SocketStreamEvent*)>& callback) { 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) on_error_ = callback; 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual int OnStartOpenConnection( 902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SocketStream* socket, 912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const CompletionCallback& callback) OVERRIDE { 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) connection_callback_ = callback; 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) events_.push_back( 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketStreamEvent(SocketStreamEvent::EVENT_START_OPEN_CONNECTION, 952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) socket, 0, std::string(), NULL, OK)); 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!on_start_open_connection_.is_null()) 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return on_start_open_connection_.Run(&events_.back()); 982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return OK; 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void OnConnected(SocketStream* socket, 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int num_pending_send_allowed) OVERRIDE { 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) events_.push_back( 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketStreamEvent(SocketStreamEvent::EVENT_CONNECTED, 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket, num_pending_send_allowed, std::string(), 1052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NULL, OK)); 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!on_connected_.is_null()) 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) on_connected_.Run(&events_.back()); 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void OnSentData(SocketStream* socket, 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int amount_sent) OVERRIDE { 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) events_.push_back( 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketStreamEvent(SocketStreamEvent::EVENT_SENT_DATA, socket, 1132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) amount_sent, std::string(), NULL, OK)); 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!on_sent_data_.is_null()) 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) on_sent_data_.Run(&events_.back()); 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void OnReceivedData(SocketStream* socket, 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const char* data, int len) OVERRIDE { 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) events_.push_back( 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketStreamEvent(SocketStreamEvent::EVENT_RECEIVED_DATA, socket, len, 1212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::string(data, len), NULL, OK)); 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!on_received_data_.is_null()) 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) on_received_data_.Run(&events_.back()); 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void OnClose(SocketStream* socket) OVERRIDE { 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) events_.push_back( 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketStreamEvent(SocketStreamEvent::EVENT_CLOSE, socket, 0, 1282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::string(), NULL, OK)); 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!on_close_.is_null()) 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) on_close_.Run(&events_.back()); 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!callback_.is_null()) 1322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) callback_.Run(OK); 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void OnAuthRequired(SocketStream* socket, 1352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) AuthChallengeInfo* auth_info) OVERRIDE { 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) events_.push_back( 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketStreamEvent(SocketStreamEvent::EVENT_AUTH_REQUIRED, socket, 0, 1382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::string(), auth_info, OK)); 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!on_auth_required_.is_null()) 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) on_auth_required_.Run(&events_.back()); 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void OnError(const SocketStream* socket, int error) OVERRIDE { 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) events_.push_back( 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SocketStreamEvent(SocketStreamEvent::EVENT_ERROR, NULL, 0, 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string(), NULL, error)); 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!on_error_.is_null()) 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) on_error_.Run(&events_.back()); 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!callback_.is_null()) 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) callback_.Run(error); 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void DoClose(SocketStreamEvent* event) { 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) event->socket->Close(); 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void DoRestartWithAuth(SocketStreamEvent* event) { 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) VLOG(1) << "RestartWithAuth username=" << credentials_.username() 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) << " password=" << credentials_.password(); 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) event->socket->RestartWithAuth(credentials_); 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) void SetAuthInfo(const AuthCredentials& credentials) { 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) credentials_ = credentials; 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void CompleteConnection(int result) { 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) connection_callback_.Run(result); 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::vector<SocketStreamEvent>& GetSeenEvents() const { 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return events_; 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<SocketStreamEvent> events_; 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Callback<int(SocketStreamEvent*)> on_start_open_connection_; 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Callback<void(SocketStreamEvent*)> on_connected_; 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Callback<void(SocketStreamEvent*)> on_sent_data_; 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Callback<void(SocketStreamEvent*)> on_received_data_; 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Callback<void(SocketStreamEvent*)> on_close_; 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Callback<void(SocketStreamEvent*)> on_auth_required_; 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Callback<void(SocketStreamEvent*)> on_error_; 1802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const CompletionCallback callback_; 1812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CompletionCallback connection_callback_; 1822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) AuthCredentials credentials_; 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(SocketStreamEventRecorder); 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 18790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)// This is used for the test OnErrorDetachDelegate. 18890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)class SelfDeletingDelegate : public SocketStream::Delegate { 18990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) public: 19090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // |callback| must cause the test message loop to exit when called. 19190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) explicit SelfDeletingDelegate(const CompletionCallback& callback) 19290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) : socket_stream_(), callback_(callback) {} 19390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 19490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) virtual ~SelfDeletingDelegate() {} 19590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 19690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // Call DetachDelegate(), delete |this|, then run the callback. 19790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) virtual void OnError(const SocketStream* socket, int error) OVERRIDE { 19890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // callback_ will be deleted when we delete |this|, so copy it to call it 19990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // afterwards. 20090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) CompletionCallback callback = callback_; 20190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) socket_stream_->DetachDelegate(); 20290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) delete this; 20390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) callback.Run(OK); 20490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) } 20590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 20690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // This can't be passed in the constructor because this object needs to be 20790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // created before SocketStream. 20890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) void set_socket_stream(const scoped_refptr<SocketStream>& socket_stream) { 20990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) socket_stream_ = socket_stream; 21090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) EXPECT_EQ(socket_stream_->delegate(), this); 21190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) } 21290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 21390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) virtual void OnConnected(SocketStream* socket, int max_pending_send_allowed) 21490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) OVERRIDE { 21590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) ADD_FAILURE() << "OnConnected() should not be called"; 21690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) } 21790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) virtual void OnSentData(SocketStream* socket, int amount_sent) OVERRIDE { 21890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) ADD_FAILURE() << "OnSentData() should not be called"; 21990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) } 22090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) virtual void OnReceivedData(SocketStream* socket, const char* data, int len) 22190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) OVERRIDE { 22290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) ADD_FAILURE() << "OnReceivedData() should not be called"; 22390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) } 22490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) virtual void OnClose(SocketStream* socket) OVERRIDE { 22590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) ADD_FAILURE() << "OnClose() should not be called"; 22690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) } 22790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 22890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) private: 22990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) scoped_refptr<SocketStream> socket_stream_; 23090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) const CompletionCallback callback_; 23190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 23290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(SelfDeletingDelegate); 23390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)}; 23490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class TestURLRequestContextWithProxy : public TestURLRequestContext { 2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) explicit TestURLRequestContextWithProxy(const std::string& proxy) 2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : TestURLRequestContext(true) { 2392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) context_storage_.set_proxy_service(ProxyService::CreateFixed(proxy)); 2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Init(); 2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~TestURLRequestContextWithProxy() {} 2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class TestSocketStreamNetworkDelegate : public TestNetworkDelegate { 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestSocketStreamNetworkDelegate() 2482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) : before_connect_result_(OK) {} 2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~TestSocketStreamNetworkDelegate() {} 2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual int OnBeforeSocketStreamConnect( 2522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SocketStream* stream, 2532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const CompletionCallback& callback) OVERRIDE { 2545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return before_connect_result_; 2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SetBeforeConnectResult(int result) { 2585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) before_connect_result_ = result; 2595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int before_connect_result_; 2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class SocketStreamTest : public PlatformTest { 2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~SocketStreamTest() {} 2705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void SetUp() { 2715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) mock_socket_factory_.reset(); 2725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) handshake_request_ = kWebSocketHandshakeRequest; 2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) handshake_response_ = kWebSocketHandshakeResponse; 2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void TearDown() { 2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) mock_socket_factory_.reset(); 2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void SetWebSocketHandshakeMessage( 2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const char* request, const char* response) { 2815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) handshake_request_ = request; 2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) handshake_response_ = response; 2835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void AddWebSocketMessage(const std::string& message) { 2855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) messages_.push_back(message); 2865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual MockClientSocketFactory* GetMockClientSocketFactory() { 2895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) mock_socket_factory_.reset(new MockClientSocketFactory); 2905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return mock_socket_factory_.get(); 2915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void DoSendWebSocketHandshake(SocketStreamEvent* event) { 2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) event->socket->SendData( 2955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) handshake_request_.data(), handshake_request_.size()); 2965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void DoCloseFlushPendingWriteTest(SocketStreamEvent* event) { 2995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // handshake response received. 3005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) for (size_t i = 0; i < messages_.size(); i++) { 3015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<char> frame; 3025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) frame.push_back('\0'); 3035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) frame.insert(frame.end(), messages_[i].begin(), messages_[i].end()); 3045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) frame.push_back('\xff'); 3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(event->socket->SendData(&frame[0], frame.size())); 3065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Actual StreamSocket close must happen after all frames queued by 3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // SendData above are sent out. 3095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) event->socket->Close(); 3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 312c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) virtual void DoFailByTooBigDataAndClose(SocketStreamEvent* event) { 313c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) std::string frame(event->number + 1, 0x00); 314c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) VLOG(1) << event->number; 315c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(event->socket->SendData(&frame[0], frame.size())); 316c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) event->socket->Close(); 317c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 318c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual int DoSwitchToSpdyTest(SocketStreamEvent* event) { 3202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return ERR_PROTOCOL_SWITCHED; 3215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual int DoIOPending(SocketStreamEvent* event) { 3242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) io_test_callback_.callback().Run(OK); 3252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return ERR_IO_PENDING; 3265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static const char kWebSocketHandshakeRequest[]; 3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static const char kWebSocketHandshakeResponse[]; 3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) protected: 3325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestCompletionCallback io_test_callback_; 3335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 3355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string handshake_request_; 3365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::string handshake_response_; 3375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<std::string> messages_; 3385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<MockClientSocketFactory> mock_socket_factory_; 3405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 3415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char SocketStreamTest::kWebSocketHandshakeRequest[] = 3435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "GET /demo HTTP/1.1\r\n" 3445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Host: example.com\r\n" 3455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Connection: Upgrade\r\n" 3465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Sec-WebSocket-Key2: 12998 5 Y3 1 .P00\r\n" 3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Sec-WebSocket-Protocol: sample\r\n" 3485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Upgrade: WebSocket\r\n" 3495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Sec-WebSocket-Key1: 4 @1 46546xW%0l 1 5\r\n" 3505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Origin: http://example.com\r\n" 3515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "\r\n" 3525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "^n:ds[4U"; 3535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char SocketStreamTest::kWebSocketHandshakeResponse[] = 3555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "HTTP/1.1 101 WebSocket Protocol Handshake\r\n" 3565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Upgrade: WebSocket\r\n" 3575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Connection: Upgrade\r\n" 3585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Sec-WebSocket-Origin: http://example.com\r\n" 3595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Sec-WebSocket-Location: ws://example.com/demo\r\n" 3605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Sec-WebSocket-Protocol: sample\r\n" 3615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "\r\n" 3625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "8jKS'y:G*Co,Wxa-"; 3635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(SocketStreamTest, CloseFlushPendingWrite) { 3655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestCompletionCallback test_callback; 3665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<SocketStreamEventRecorder> delegate( 3685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new SocketStreamEventRecorder(test_callback.callback())); 3695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate->SetOnConnected(base::Bind( 3705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &SocketStreamTest::DoSendWebSocketHandshake, base::Unretained(this))); 3715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate->SetOnReceivedData(base::Bind( 3725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &SocketStreamTest::DoCloseFlushPendingWriteTest, 3735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Unretained(this))); 3745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestURLRequestContext context; 3765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<SocketStream> socket_stream( 3785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new SocketStream(GURL("ws://example.com/demo"), delegate.get())); 3795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->set_context(&context); 3815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockWrite data_writes[] = { 3835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockWrite(SocketStreamTest::kWebSocketHandshakeRequest), 3845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockWrite(ASYNC, "\0message1\xff", 10), 3855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockWrite(ASYNC, "\0message2\xff", 10) 3865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 3875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead data_reads[] = { 3885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead(SocketStreamTest::kWebSocketHandshakeResponse), 3895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Server doesn't close the connection after handshake. 3905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead(ASYNC, ERR_IO_PENDING) 3915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 3925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AddWebSocketMessage("message1"); 3935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AddWebSocketMessage("message2"); 3945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DelayedSocketData data_provider( 3965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1, data_reads, arraysize(data_reads), 3975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) data_writes, arraysize(data_writes)); 3985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockClientSocketFactory* mock_socket_factory = 4005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) GetMockClientSocketFactory(); 4015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) mock_socket_factory->AddSocketDataProvider(&data_provider); 4025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->SetClientSocketFactory(mock_socket_factory); 4045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->Connect(); 4065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) test_callback.WaitForResult(); 4085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::vector<SocketStreamEvent>& events = delegate->GetSeenEvents(); 410c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_EQ(7U, events.size()); 4115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_START_OPEN_CONNECTION, 4135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) events[0].event_type); 4145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_CONNECTED, events[1].event_type); 4155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_SENT_DATA, events[2].event_type); 4165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_RECEIVED_DATA, events[3].event_type); 4175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_SENT_DATA, events[4].event_type); 4185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_SENT_DATA, events[5].event_type); 419c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_CLOSE, events[6].event_type); 420c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 421c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 422868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)TEST_F(SocketStreamTest, ResolveFailure) { 423868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) TestCompletionCallback test_callback; 424868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 425868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) scoped_ptr<SocketStreamEventRecorder> delegate( 426868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) new SocketStreamEventRecorder(test_callback.callback())); 427868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 428868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) scoped_refptr<SocketStream> socket_stream( 429868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) new SocketStream(GURL("ws://example.com/demo"), delegate.get())); 430868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 431868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) // Make resolver fail. 432868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) TestURLRequestContext context; 433868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) scoped_ptr<MockHostResolver> mock_host_resolver( 434868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) new MockHostResolver()); 435868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) mock_host_resolver->rules()->AddSimulatedFailure("example.com"); 436868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) context.set_host_resolver(mock_host_resolver.get()); 437868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) socket_stream->set_context(&context); 438868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 439868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) // No read/write on socket is expected. 440868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) StaticSocketDataProvider data_provider(NULL, 0, NULL, 0); 441868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) MockClientSocketFactory* mock_socket_factory = 442868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) GetMockClientSocketFactory(); 443868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) mock_socket_factory->AddSocketDataProvider(&data_provider); 444868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) socket_stream->SetClientSocketFactory(mock_socket_factory); 445868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 446868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) socket_stream->Connect(); 447868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 448868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) test_callback.WaitForResult(); 449868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 450868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) const std::vector<SocketStreamEvent>& events = delegate->GetSeenEvents(); 451868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) ASSERT_EQ(2U, events.size()); 452868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 453868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_ERROR, events[0].event_type); 454868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_CLOSE, events[1].event_type); 455868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)} 456868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) 457c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)TEST_F(SocketStreamTest, ExceedMaxPendingSendAllowed) { 458c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) TestCompletionCallback test_callback; 459c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 460c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) scoped_ptr<SocketStreamEventRecorder> delegate( 461c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) new SocketStreamEventRecorder(test_callback.callback())); 462c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) delegate->SetOnConnected(base::Bind( 463c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) &SocketStreamTest::DoFailByTooBigDataAndClose, base::Unretained(this))); 464c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 465c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) TestURLRequestContext context; 466c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 467c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) scoped_refptr<SocketStream> socket_stream( 468c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) new SocketStream(GURL("ws://example.com/demo"), delegate.get())); 469c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 470c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) socket_stream->set_context(&context); 471c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 472c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) DelayedSocketData data_provider(1, NULL, 0, NULL, 0); 473c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 474c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) MockClientSocketFactory* mock_socket_factory = 475c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) GetMockClientSocketFactory(); 476c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) mock_socket_factory->AddSocketDataProvider(&data_provider); 477c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 478c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) socket_stream->SetClientSocketFactory(mock_socket_factory); 479c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 480c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) socket_stream->Connect(); 481c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 482c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) test_callback.WaitForResult(); 483c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 484c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const std::vector<SocketStreamEvent>& events = delegate->GetSeenEvents(); 485c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_EQ(4U, events.size()); 486c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 487c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_START_OPEN_CONNECTION, 488c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) events[0].event_type); 489c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_CONNECTED, events[1].event_type); 490c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_ERROR, events[2].event_type); 491c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_CLOSE, events[3].event_type); 4925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(SocketStreamTest, BasicAuthProxy) { 4955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockClientSocketFactory mock_socket_factory; 4965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockWrite data_writes1[] = { 4975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockWrite("CONNECT example.com:80 HTTP/1.1\r\n" 4985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Host: example.com\r\n" 4995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Proxy-Connection: keep-alive\r\n\r\n"), 5005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 5015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead data_reads1[] = { 5025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead("HTTP/1.1 407 Proxy Authentication Required\r\n"), 5035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead("Proxy-Authenticate: Basic realm=\"MyRealm1\"\r\n"), 5045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead("\r\n"), 5055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 5065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) StaticSocketDataProvider data1(data_reads1, arraysize(data_reads1), 5075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) data_writes1, arraysize(data_writes1)); 5085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) mock_socket_factory.AddSocketDataProvider(&data1); 5095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockWrite data_writes2[] = { 5115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockWrite("CONNECT example.com:80 HTTP/1.1\r\n" 5125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Host: example.com\r\n" 5135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Proxy-Connection: keep-alive\r\n" 5145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Proxy-Authorization: Basic Zm9vOmJhcg==\r\n\r\n"), 5155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 5165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead data_reads2[] = { 5175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead("HTTP/1.1 200 Connection Established\r\n"), 5185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead("Proxy-agent: Apache/2.2.8\r\n"), 5195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead("\r\n"), 5205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // SocketStream::DoClose is run asynchronously. Socket can be read after 5215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // "\r\n". We have to give ERR_IO_PENDING to SocketStream then to indicate 5225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // server doesn't close the connection. 5235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead(ASYNC, ERR_IO_PENDING) 5245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 5255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) StaticSocketDataProvider data2(data_reads2, arraysize(data_reads2), 5265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) data_writes2, arraysize(data_writes2)); 5275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) mock_socket_factory.AddSocketDataProvider(&data2); 5285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestCompletionCallback test_callback; 5305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<SocketStreamEventRecorder> delegate( 5325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new SocketStreamEventRecorder(test_callback.callback())); 5335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate->SetOnConnected(base::Bind(&SocketStreamEventRecorder::DoClose, 5345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Unretained(delegate.get()))); 5352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) delegate->SetAuthInfo(AuthCredentials(ASCIIToUTF16("foo"), 5362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASCIIToUTF16("bar"))); 5375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate->SetOnAuthRequired(base::Bind( 5385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &SocketStreamEventRecorder::DoRestartWithAuth, 5395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Unretained(delegate.get()))); 5405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<SocketStream> socket_stream( 5425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new SocketStream(GURL("ws://example.com/demo"), delegate.get())); 5435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestURLRequestContextWithProxy context("myproxy:70"); 5455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->set_context(&context); 5475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->SetClientSocketFactory(&mock_socket_factory); 5485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->Connect(); 5505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) test_callback.WaitForResult(); 5525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::vector<SocketStreamEvent>& events = delegate->GetSeenEvents(); 5545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(5U, events.size()); 5555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_START_OPEN_CONNECTION, 5575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) events[0].event_type); 5585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_AUTH_REQUIRED, events[1].event_type); 5595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_CONNECTED, events[2].event_type); 5605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_ERROR, events[3].event_type); 5612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(ERR_ABORTED, events[3].error_code); 5625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_CLOSE, events[4].event_type); 5635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(eroman): Add back NetLogTest here... 5655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(SocketStreamTest, BasicAuthProxyWithAuthCache) { 5685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockClientSocketFactory mock_socket_factory; 5695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockWrite data_writes[] = { 5705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // WebSocket(SocketStream) always uses CONNECT when it is configured to use 5715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // proxy so the port may not be 443. 5725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockWrite("CONNECT example.com:80 HTTP/1.1\r\n" 5735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Host: example.com\r\n" 5745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Proxy-Connection: keep-alive\r\n" 5755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Proxy-Authorization: Basic Zm9vOmJhcg==\r\n\r\n"), 5765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 5775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead data_reads[] = { 5785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead("HTTP/1.1 200 Connection Established\r\n"), 5795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead("Proxy-agent: Apache/2.2.8\r\n"), 5805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead("\r\n"), 5815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead(ASYNC, ERR_IO_PENDING) 5825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 5835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) StaticSocketDataProvider data(data_reads, arraysize(data_reads), 5845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) data_writes, arraysize(data_writes)); 5855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) mock_socket_factory.AddSocketDataProvider(&data); 5865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestCompletionCallback test_callback; 5885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<SocketStreamEventRecorder> delegate( 5895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new SocketStreamEventRecorder(test_callback.callback())); 5905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate->SetOnConnected(base::Bind(&SocketStreamEventRecorder::DoClose, 5915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Unretained(delegate.get()))); 5925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<SocketStream> socket_stream( 5945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new SocketStream(GURL("ws://example.com/demo"), delegate.get())); 5955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestURLRequestContextWithProxy context("myproxy:70"); 5975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HttpAuthCache* auth_cache = 5985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) context.http_transaction_factory()->GetSession()->http_auth_cache(); 5995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) auth_cache->Add(GURL("http://myproxy:70"), 6005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "MyRealm1", 6015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HttpAuth::AUTH_SCHEME_BASIC, 6025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Basic realm=MyRealm1", 6035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AuthCredentials(ASCIIToUTF16("foo"), 6045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASCIIToUTF16("bar")), 6055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "/"); 6065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->set_context(&context); 6085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->SetClientSocketFactory(&mock_socket_factory); 6095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->Connect(); 6115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) test_callback.WaitForResult(); 6135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::vector<SocketStreamEvent>& events = delegate->GetSeenEvents(); 6155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(4U, events.size()); 6165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_START_OPEN_CONNECTION, 6175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) events[0].event_type); 6185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_CONNECTED, events[1].event_type); 6192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(ERR_ABORTED, events[2].error_code); 6205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_CLOSE, events[3].event_type); 6215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(SocketStreamTest, WSSBasicAuthProxyWithAuthCache) { 6245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockClientSocketFactory mock_socket_factory; 6255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockWrite data_writes1[] = { 6265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockWrite("CONNECT example.com:443 HTTP/1.1\r\n" 6275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Host: example.com\r\n" 6285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Proxy-Connection: keep-alive\r\n" 6295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Proxy-Authorization: Basic Zm9vOmJhcg==\r\n\r\n"), 6305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 6315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead data_reads1[] = { 6325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead("HTTP/1.1 200 Connection Established\r\n"), 6335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead("Proxy-agent: Apache/2.2.8\r\n"), 6345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead("\r\n"), 6355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead(ASYNC, ERR_IO_PENDING) 6365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 6375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) StaticSocketDataProvider data1(data_reads1, arraysize(data_reads1), 6385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) data_writes1, arraysize(data_writes1)); 6395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) mock_socket_factory.AddSocketDataProvider(&data1); 6405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SSLSocketDataProvider data2(ASYNC, OK); 6425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) mock_socket_factory.AddSSLSocketDataProvider(&data2); 6435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestCompletionCallback test_callback; 6455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<SocketStreamEventRecorder> delegate( 6465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new SocketStreamEventRecorder(test_callback.callback())); 6475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate->SetOnConnected(base::Bind(&SocketStreamEventRecorder::DoClose, 6485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Unretained(delegate.get()))); 6495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<SocketStream> socket_stream( 6515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new SocketStream(GURL("wss://example.com/demo"), delegate.get())); 6525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestURLRequestContextWithProxy context("myproxy:70"); 6545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HttpAuthCache* auth_cache = 6555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) context.http_transaction_factory()->GetSession()->http_auth_cache(); 6565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) auth_cache->Add(GURL("http://myproxy:70"), 6575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "MyRealm1", 6585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HttpAuth::AUTH_SCHEME_BASIC, 6595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Basic realm=MyRealm1", 6605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AuthCredentials(ASCIIToUTF16("foo"), 6615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASCIIToUTF16("bar")), 6625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "/"); 6635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->set_context(&context); 6655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->SetClientSocketFactory(&mock_socket_factory); 6665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->Connect(); 6685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) test_callback.WaitForResult(); 6705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::vector<SocketStreamEvent>& events = delegate->GetSeenEvents(); 6725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(4U, events.size()); 6735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_START_OPEN_CONNECTION, 6745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) events[0].event_type); 6755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_CONNECTED, events[1].event_type); 6762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(ERR_ABORTED, events[2].error_code); 6775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_CLOSE, events[3].event_type); 6785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(SocketStreamTest, IOPending) { 6815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestCompletionCallback test_callback; 6825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<SocketStreamEventRecorder> delegate( 6845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new SocketStreamEventRecorder(test_callback.callback())); 6855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate->SetOnConnected(base::Bind( 6865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &SocketStreamTest::DoSendWebSocketHandshake, base::Unretained(this))); 6875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate->SetOnReceivedData(base::Bind( 6885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &SocketStreamTest::DoCloseFlushPendingWriteTest, 6895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Unretained(this))); 6905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate->SetOnStartOpenConnection(base::Bind( 6915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &SocketStreamTest::DoIOPending, base::Unretained(this))); 6925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestURLRequestContext context; 6945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<SocketStream> socket_stream( 6965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new SocketStream(GURL("ws://example.com/demo"), delegate.get())); 6975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->set_context(&context); 6995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockWrite data_writes[] = { 7015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockWrite(SocketStreamTest::kWebSocketHandshakeRequest), 7025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockWrite(ASYNC, "\0message1\xff", 10), 7035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockWrite(ASYNC, "\0message2\xff", 10) 7045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 7055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead data_reads[] = { 7065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead(SocketStreamTest::kWebSocketHandshakeResponse), 7075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Server doesn't close the connection after handshake. 7085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead(ASYNC, ERR_IO_PENDING) 7095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 7105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AddWebSocketMessage("message1"); 7115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) AddWebSocketMessage("message2"); 7125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DelayedSocketData data_provider( 7145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1, data_reads, arraysize(data_reads), 7155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) data_writes, arraysize(data_writes)); 7165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockClientSocketFactory* mock_socket_factory = 7185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) GetMockClientSocketFactory(); 7195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) mock_socket_factory->AddSocketDataProvider(&data_provider); 7205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->SetClientSocketFactory(mock_socket_factory); 7225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->Connect(); 7245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) io_test_callback_.WaitForResult(); 7252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(SocketStream::STATE_RESOLVE_PROTOCOL_COMPLETE, 7265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->next_state_); 7272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) delegate->CompleteConnection(OK); 7285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(OK, test_callback.WaitForResult()); 7305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::vector<SocketStreamEvent>& events = delegate->GetSeenEvents(); 732c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_EQ(7U, events.size()); 7335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_START_OPEN_CONNECTION, 7355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) events[0].event_type); 7365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_CONNECTED, events[1].event_type); 7375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_SENT_DATA, events[2].event_type); 7385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_RECEIVED_DATA, events[3].event_type); 7395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_SENT_DATA, events[4].event_type); 7405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_SENT_DATA, events[5].event_type); 741c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_CLOSE, events[6].event_type); 7425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 7435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(SocketStreamTest, SwitchToSpdy) { 7455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestCompletionCallback test_callback; 7465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<SocketStreamEventRecorder> delegate( 7485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new SocketStreamEventRecorder(test_callback.callback())); 7495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate->SetOnStartOpenConnection(base::Bind( 7505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &SocketStreamTest::DoSwitchToSpdyTest, base::Unretained(this))); 7515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestURLRequestContext context; 7535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<SocketStream> socket_stream( 7555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new SocketStream(GURL("ws://example.com/demo"), delegate.get())); 7565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->set_context(&context); 7585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->Connect(); 7605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(ERR_PROTOCOL_SWITCHED, test_callback.WaitForResult()); 7625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::vector<SocketStreamEvent>& events = delegate->GetSeenEvents(); 7645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(2U, events.size()); 7655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_START_OPEN_CONNECTION, 7675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) events[0].event_type); 7685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_ERROR, events[1].event_type); 7692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(ERR_PROTOCOL_SWITCHED, events[1].error_code); 7705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 7715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(SocketStreamTest, SwitchAfterPending) { 7735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestCompletionCallback test_callback; 7745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<SocketStreamEventRecorder> delegate( 7765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new SocketStreamEventRecorder(test_callback.callback())); 7775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate->SetOnStartOpenConnection(base::Bind( 7785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &SocketStreamTest::DoIOPending, base::Unretained(this))); 7795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestURLRequestContext context; 7815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<SocketStream> socket_stream( 7835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new SocketStream(GURL("ws://example.com/demo"), delegate.get())); 7845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->set_context(&context); 7865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->Connect(); 7885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) io_test_callback_.WaitForResult(); 7892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(SocketStream::STATE_RESOLVE_PROTOCOL_COMPLETE, 7905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->next_state_); 7912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) delegate->CompleteConnection(ERR_PROTOCOL_SWITCHED); 7925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(ERR_PROTOCOL_SWITCHED, test_callback.WaitForResult()); 7945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::vector<SocketStreamEvent>& events = delegate->GetSeenEvents(); 7965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(2U, events.size()); 7975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_START_OPEN_CONNECTION, 7995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) events[0].event_type); 8005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_ERROR, events[1].event_type); 8012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(ERR_PROTOCOL_SWITCHED, events[1].error_code); 8025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 8035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Test a connection though a secure proxy. 8055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(SocketStreamTest, SecureProxyConnectError) { 8065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockClientSocketFactory mock_socket_factory; 8075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockWrite data_writes[] = { 8085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockWrite("CONNECT example.com:80 HTTP/1.1\r\n" 8095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Host: example.com\r\n" 8105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Proxy-Connection: keep-alive\r\n\r\n") 8115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 8125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead data_reads[] = { 8135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead("HTTP/1.1 200 Connection Established\r\n"), 8145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead("Proxy-agent: Apache/2.2.8\r\n"), 8155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead("\r\n"), 8165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // SocketStream::DoClose is run asynchronously. Socket can be read after 8175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // "\r\n". We have to give ERR_IO_PENDING to SocketStream then to indicate 8185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // server doesn't close the connection. 8195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead(ASYNC, ERR_IO_PENDING) 8205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 8215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) StaticSocketDataProvider data(data_reads, arraysize(data_reads), 8225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) data_writes, arraysize(data_writes)); 8235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) mock_socket_factory.AddSocketDataProvider(&data); 8245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SSLSocketDataProvider ssl(SYNCHRONOUS, ERR_SSL_PROTOCOL_ERROR); 8255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) mock_socket_factory.AddSSLSocketDataProvider(&ssl); 8265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestCompletionCallback test_callback; 8285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestURLRequestContextWithProxy context("https://myproxy:70"); 8295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<SocketStreamEventRecorder> delegate( 8315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new SocketStreamEventRecorder(test_callback.callback())); 8325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate->SetOnConnected(base::Bind(&SocketStreamEventRecorder::DoClose, 8335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Unretained(delegate.get()))); 8345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<SocketStream> socket_stream( 8365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new SocketStream(GURL("ws://example.com/demo"), delegate.get())); 8375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->set_context(&context); 8395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->SetClientSocketFactory(&mock_socket_factory); 8405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->Connect(); 8425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) test_callback.WaitForResult(); 8445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::vector<SocketStreamEvent>& events = delegate->GetSeenEvents(); 8465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(3U, events.size()); 8475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_START_OPEN_CONNECTION, 8495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) events[0].event_type); 8505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_ERROR, events[1].event_type); 8512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(ERR_SSL_PROTOCOL_ERROR, events[1].error_code); 8525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_CLOSE, events[2].event_type); 8535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 8545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Test a connection though a secure proxy. 8565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(SocketStreamTest, SecureProxyConnect) { 8575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockClientSocketFactory mock_socket_factory; 8585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockWrite data_writes[] = { 8595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockWrite("CONNECT example.com:80 HTTP/1.1\r\n" 8605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Host: example.com\r\n" 8615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "Proxy-Connection: keep-alive\r\n\r\n") 8625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 8635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead data_reads[] = { 8645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead("HTTP/1.1 200 Connection Established\r\n"), 8655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead("Proxy-agent: Apache/2.2.8\r\n"), 8665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead("\r\n"), 8675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // SocketStream::DoClose is run asynchronously. Socket can be read after 8685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // "\r\n". We have to give ERR_IO_PENDING to SocketStream then to indicate 8695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // server doesn't close the connection. 8705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockRead(ASYNC, ERR_IO_PENDING) 8715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) }; 8725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) StaticSocketDataProvider data(data_reads, arraysize(data_reads), 8735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) data_writes, arraysize(data_writes)); 8745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) mock_socket_factory.AddSocketDataProvider(&data); 8755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SSLSocketDataProvider ssl(SYNCHRONOUS, OK); 8765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) mock_socket_factory.AddSSLSocketDataProvider(&ssl); 8775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestCompletionCallback test_callback; 8795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestURLRequestContextWithProxy context("https://myproxy:70"); 8805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<SocketStreamEventRecorder> delegate( 8825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new SocketStreamEventRecorder(test_callback.callback())); 8835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) delegate->SetOnConnected(base::Bind(&SocketStreamEventRecorder::DoClose, 8845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Unretained(delegate.get()))); 8855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<SocketStream> socket_stream( 8875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new SocketStream(GURL("ws://example.com/demo"), delegate.get())); 8885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->set_context(&context); 8905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->SetClientSocketFactory(&mock_socket_factory); 8915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->Connect(); 8935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) test_callback.WaitForResult(); 8955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::vector<SocketStreamEvent>& events = delegate->GetSeenEvents(); 8975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(4U, events.size()); 8985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_START_OPEN_CONNECTION, 9005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) events[0].event_type); 9015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_CONNECTED, events[1].event_type); 9025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_ERROR, events[2].event_type); 9032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(ERR_ABORTED, events[2].error_code); 9045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_CLOSE, events[3].event_type); 9055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 9065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST_F(SocketStreamTest, BeforeConnectFailed) { 9085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestCompletionCallback test_callback; 9095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<SocketStreamEventRecorder> delegate( 9115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new SocketStreamEventRecorder(test_callback.callback())); 9125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestURLRequestContext context; 9145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestSocketStreamNetworkDelegate network_delegate; 9155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) network_delegate.SetBeforeConnectResult(ERR_ACCESS_DENIED); 9165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) context.set_network_delegate(&network_delegate); 9175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<SocketStream> socket_stream( 9195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new SocketStream(GURL("ws://example.com/demo"), delegate.get())); 9205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->set_context(&context); 9225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) socket_stream->Connect(); 9245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) test_callback.WaitForResult(); 9265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const std::vector<SocketStreamEvent>& events = delegate->GetSeenEvents(); 9285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_EQ(2U, events.size()); 9295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 9305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_ERROR, events[0].event_type); 9312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(ERR_ACCESS_DENIED, events[0].error_code); 9325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(SocketStreamEvent::EVENT_CLOSE, events[1].event_type); 9335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 9345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 93590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)// Check that a connect failure, followed by the delegate calling DetachDelegate 93690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)// and deleting itself in the OnError callback, is handled correctly. 93790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)TEST_F(SocketStreamTest, OnErrorDetachDelegate) { 93890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) MockClientSocketFactory mock_socket_factory; 93990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) TestCompletionCallback test_callback; 94090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 94190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // SelfDeletingDelegate is self-owning; we just need a pointer to it to 94290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // connect it and the SocketStream. 94390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) SelfDeletingDelegate* delegate = 94490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) new SelfDeletingDelegate(test_callback.callback()); 94590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) MockConnect mock_connect(ASYNC, ERR_CONNECTION_REFUSED); 94690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) StaticSocketDataProvider data; 94790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) data.set_connect_data(mock_connect); 94890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) mock_socket_factory.AddSocketDataProvider(&data); 94990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 95090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) TestURLRequestContext context; 95190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) scoped_refptr<SocketStream> socket_stream( 95290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) new SocketStream(GURL("ws://localhost:9998/echo"), delegate)); 95390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) socket_stream->set_context(&context); 95490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) socket_stream->SetClientSocketFactory(&mock_socket_factory); 95590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) delegate->set_socket_stream(socket_stream); 95690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // The delegate pointer will become invalid during the test. Set it to NULL to 95790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // avoid holding a dangling pointer. 95890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) delegate = NULL; 95990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 96090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) socket_stream->Connect(); 96190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 96290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) EXPECT_EQ(OK, test_callback.WaitForResult()); 96390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)} 96490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 9655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace net 966