| Index: content/browser/renderer_host/websocket_dispatcher_host_unittest.cc
|
| diff --git a/content/browser/renderer_host/websocket_dispatcher_host_unittest.cc b/content/browser/renderer_host/websocket_dispatcher_host_unittest.cc
|
| index 0d2e7057bbcef5c0d90dadebc580843114e208bd..939092efa508187b430a70758d365c1b5390bc43 100644
|
| --- a/content/browser/renderer_host/websocket_dispatcher_host_unittest.cc
|
| +++ b/content/browser/renderer_host/websocket_dispatcher_host_unittest.cc
|
| @@ -11,6 +11,7 @@
|
| #include "base/bind_helpers.h"
|
| #include "base/memory/ref_counted.h"
|
| #include "base/memory/weak_ptr.h"
|
| +#include "base/message_loop/message_loop.h"
|
| #include "content/browser/renderer_host/websocket_host.h"
|
| #include "content/common/websocket.h"
|
| #include "content/common/websocket_messages.h"
|
| @@ -33,6 +34,7 @@ class MockWebSocketHost : public WebSocketHost {
|
| MockWebSocketHost(int routing_id,
|
| WebSocketDispatcherHost* dispatcher,
|
| net::URLRequestContext* url_request_context,
|
| + base::TimeDelta delay,
|
| WebSocketDispatcherHostTest* owner);
|
|
|
| ~MockWebSocketHost() override {}
|
| @@ -82,6 +84,8 @@ class WebSocketDispatcherHostTest : public ::testing::Test {
|
| std::vector<MockWebSocketHost*> mock_hosts_;
|
| std::vector<int> gone_hosts_;
|
|
|
| + base::MessageLoop message_loop_;
|
| +
|
| base::WeakPtrFactory<WebSocketDispatcherHostTest> weak_ptr_factory_;
|
|
|
| private:
|
| @@ -89,9 +93,9 @@ class WebSocketDispatcherHostTest : public ::testing::Test {
|
| return NULL;
|
| }
|
|
|
| - WebSocketHost* CreateWebSocketHost(int routing_id) {
|
| - MockWebSocketHost* host =
|
| - new MockWebSocketHost(routing_id, dispatcher_host_.get(), NULL, this);
|
| + WebSocketHost* CreateWebSocketHost(int routing_id, base::TimeDelta delay) {
|
| + MockWebSocketHost* host = new MockWebSocketHost(
|
| + routing_id, dispatcher_host_.get(), NULL, delay, this);
|
| mock_hosts_.push_back(host);
|
| return host;
|
| }
|
| @@ -101,8 +105,9 @@ MockWebSocketHost::MockWebSocketHost(
|
| int routing_id,
|
| WebSocketDispatcherHost* dispatcher,
|
| net::URLRequestContext* url_request_context,
|
| + base::TimeDelta delay,
|
| WebSocketDispatcherHostTest* owner)
|
| - : WebSocketHost(routing_id, dispatcher, url_request_context),
|
| + : WebSocketHost(routing_id, dispatcher, url_request_context, delay),
|
| owner_(owner->GetWeakPtr()) {}
|
|
|
| void MockWebSocketHost::GoAway() {
|
|
|