Index: net/websockets/websocket_test_util.h |
diff --git a/net/websockets/websocket_test_util.h b/net/websockets/websocket_test_util.h |
index 5dba6cde691450d7ef825f39516404a21f613f22..9a776b6a314d5c0ab26555fe317dd1175c9580b9 100644 |
--- a/net/websockets/websocket_test_util.h |
+++ b/net/websockets/websocket_test_util.h |
@@ -21,10 +21,11 @@ class Origin; |
namespace net { |
class BoundNetLog; |
+class DeterministicMockClientSocketFactory; |
class DeterministicSocketData; |
class URLRequestContext; |
class WebSocketHandshakeStreamCreateHelper; |
-class DeterministicMockClientSocketFactory; |
+struct SSLSocketDataProvider; |
class LinearCongruentialGenerator { |
public: |
@@ -72,8 +73,12 @@ class WebSocketDeterministicMockClientSocketFactoryMaker { |
void SetExpectations(const std::string& expect_written, |
const std::string& return_to_read); |
- // A low-level interface to permit arbitrary expectations to be set. |
- void SetRawExpectations(scoped_ptr<DeterministicSocketData> socket_data); |
+ // A low-level interface to permit arbitrary expectations to be added. |
+ void AddRawExpectations(scoped_ptr<DeterministicSocketData> socket_data); |
+ |
+ // Allow an SSL socket data provider to be added. |
+ void AddSSLSocketDataProvider( |
+ scoped_ptr<SSLSocketDataProvider> ssl_socket_data); |
// Call to get a pointer to the factory, which remains owned by this object. |
DeterministicMockClientSocketFactory* factory(); |
@@ -98,7 +103,11 @@ struct WebSocketTestURLRequestContextHost { |
maker_.SetExpectations(expect_written, return_to_read); |
} |
- void SetRawExpectations(scoped_ptr<DeterministicSocketData> socket_data); |
+ void AddRawExpectations(scoped_ptr<DeterministicSocketData> socket_data); |
+ |
+ // Allow an SSL socket data provider to be added. |
+ void AddSSLSocketDataProvider( |
+ scoped_ptr<SSLSocketDataProvider> ssl_socket_data); |
// Call after calling one of SetExpections() or SetRawExpectations(). The |
// returned pointer remains owned by this object. This should only be called |