Support recovery from SSL errors for new WebSocket implementation

net/websockets/websocket_test_util.cc

 // Copyright 2013 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "net/websockets/websocket_test_util.h"
 
 #include <algorithm>
 #include <vector>
 
 #include "base/basictypes.h"
+#include "base/memory/scoped_vector.h"
 #include "base/stl_util.h"
 #include "base/strings/stringprintf.h"
 #include "net/socket/socket_test_util.h"
 
 namespace net {
 
 namespace {
 const uint64 kA =
     (static_cast<uint64>(0x5851f42d) << 32) + static_cast<uint64>(0x4c957f2d);
 const uint64 kC = 12345;
@@ skipping 44 matching lines @@
       "Sec-WebSocket-Accept: s3pPLMBiTxaQ9kYGzzhZRbK+xOo=\r\n"
       "%s\r\n",
       extra_headers.c_str());
 }
 
 struct WebSocketDeterministicMockClientSocketFactoryMaker::Detail {
   std::string expect_written;
   std::string return_to_read;
   std::vector<MockRead> reads;
   MockWrite write;
-  scoped_ptr<DeterministicSocketData> data;
+  ScopedVector<DeterministicSocketData> socket_data_vector;
+  ScopedVector<SSLSocketDataProvider> ssl_socket_data_vector;
   DeterministicMockClientSocketFactory factory;
 };
 
 WebSocketDeterministicMockClientSocketFactoryMaker::
     WebSocketDeterministicMockClientSocketFactoryMaker()
     : detail_(new Detail) {}
 
 WebSocketDeterministicMockClientSocketFactoryMaker::
     ~WebSocketDeterministicMockClientSocketFactoryMaker() {}
 
@@ skipping 24 matching lines @@
                           kHttpStreamParserBufferSize),
                  sequence++));
   }
   scoped_ptr<DeterministicSocketData> socket_data(
       new DeterministicSocketData(vector_as_array(&detail_->reads),
                                   detail_->reads.size(),
                                   &detail_->write,
                                   1));
   socket_data->set_connect_data(MockConnect(SYNCHRONOUS, OK));
   socket_data->SetStop(sequence);
-  SetRawExpectations(socket_data.Pass());
+  AddRawExpectations(socket_data.Pass());
 }
 
-void WebSocketDeterministicMockClientSocketFactoryMaker::SetRawExpectations(
+void WebSocketDeterministicMockClientSocketFactoryMaker::AddRawExpectations(
     scoped_ptr<DeterministicSocketData> socket_data) {
-  detail_->data = socket_data.Pass();
-  detail_->factory.AddSocketDataProvider(detail_->data.get());
+  detail_->factory.AddSocketDataProvider(socket_data.get());
+  detail_->socket_data_vector.push_back(socket_data.release());
+}
+
+void
+WebSocketDeterministicMockClientSocketFactoryMaker::AddSSLSocketDataProvider(
+    scoped_ptr<SSLSocketDataProvider> ssl_socket_data) {
+  detail_->factory.AddSSLSocketDataProvider(ssl_socket_data.get());
+  detail_->ssl_socket_data_vector.push_back(ssl_socket_data.release());
 }
 
 WebSocketTestURLRequestContextHost::WebSocketTestURLRequestContextHost()
     : url_request_context_(true) {
   url_request_context_.set_client_socket_factory(maker_.factory());
 }
 
 WebSocketTestURLRequestContextHost::~WebSocketTestURLRequestContextHost() {}
 
-void WebSocketTestURLRequestContextHost::SetRawExpectations(
+void WebSocketTestURLRequestContextHost::AddRawExpectations(
     scoped_ptr<DeterministicSocketData> socket_data) {
-  maker_.SetRawExpectations(socket_data.Pass());
+  maker_.AddRawExpectations(socket_data.Pass());
+}
+
+void WebSocketTestURLRequestContextHost::AddSSLSocketDataProvider(
+    scoped_ptr<SSLSocketDataProvider> ssl_socket_data) {
+  maker_.AddSSLSocketDataProvider(ssl_socket_data.Pass());
 }
 
 TestURLRequestContext*
 WebSocketTestURLRequestContextHost::GetURLRequestContext() {
   url_request_context_.Init();
   // A Network Delegate is required to make the URLRequest::Delegate work.
   url_request_context_.set_network_delegate(&network_delegate_);
   return &url_request_context_;
 }
 
 }  // namespace net