Index: net/socket/socket_test_util.cc |
diff --git a/net/socket/socket_test_util.cc b/net/socket/socket_test_util.cc |
index 658182b736b4d63283f31daf1d9028affb4560cf..791c9f684804dd7fd74ac8adee3b2a5e7d8cc164 100644 |
--- a/net/socket/socket_test_util.cc |
+++ b/net/socket/socket_test_util.cc |
@@ -24,6 +24,7 @@ |
#include "net/http/http_response_headers.h" |
#include "net/socket/client_socket_pool_histograms.h" |
#include "net/socket/socket.h" |
+#include "net/socket/websocket_endpoint_lock_manager.h" |
#include "net/ssl/ssl_cert_request_info.h" |
#include "net/ssl/ssl_connection_status_flags.h" |
#include "net/ssl/ssl_info.h" |
@@ -1989,6 +1990,21 @@ void MockSOCKSClientSocketPool::ReleaseSocket(const std::string& group_name, |
return transport_pool_->ReleaseSocket(group_name, socket.Pass(), id); |
} |
+ScopedWebSocketEndpointZeroUnlockDelay:: |
+ ScopedWebSocketEndpointZeroUnlockDelay() { |
+ old_delay_ = |
+ WebSocketEndpointLockManager::GetInstance()->SetUnlockDelayForTesting( |
+ base::TimeDelta()); |
+} |
+ |
+ScopedWebSocketEndpointZeroUnlockDelay:: |
+ ~ScopedWebSocketEndpointZeroUnlockDelay() { |
+ base::TimeDelta active_delay = |
+ WebSocketEndpointLockManager::GetInstance()->SetUnlockDelayForTesting( |
+ old_delay_); |
+ EXPECT_EQ(active_delay, base::TimeDelta()); |
+} |
+ |
const char kSOCKS5GreetRequest[] = { 0x05, 0x01, 0x00 }; |
const int kSOCKS5GreetRequestLength = arraysize(kSOCKS5GreetRequest); |