diff options
Diffstat (limited to 'content/browser/renderer_host/websocket_dispatcher_host_unittest.cc')
-rw-r--r-- | content/browser/renderer_host/websocket_dispatcher_host_unittest.cc | 49 |
1 files changed, 4 insertions, 45 deletions
diff --git a/content/browser/renderer_host/websocket_dispatcher_host_unittest.cc b/content/browser/renderer_host/websocket_dispatcher_host_unittest.cc index 8504d8f..e1506d9 100644 --- a/content/browser/renderer_host/websocket_dispatcher_host_unittest.cc +++ b/content/browser/renderer_host/websocket_dispatcher_host_unittest.cc @@ -4,7 +4,6 @@ #include "content/browser/renderer_host/websocket_dispatcher_host.h" -#include <algorithm> #include <vector> #include "base/bind.h" @@ -24,30 +23,23 @@ namespace { // This number is unlikely to occur by chance. static const int kMagicRenderProcessId = 506116062; -class WebSocketDispatcherHostTest; - // A mock of WebsocketHost which records received messages. class MockWebSocketHost : public WebSocketHost { public: MockWebSocketHost(int routing_id, WebSocketDispatcherHost* dispatcher, - net::URLRequestContext* url_request_context, - WebSocketDispatcherHostTest* owner) - : WebSocketHost(routing_id, dispatcher, url_request_context), - owner_(owner) { + net::URLRequestContext* url_request_context) + : WebSocketHost(routing_id, dispatcher, url_request_context) { } virtual ~MockWebSocketHost() {} - virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE { + virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE{ received_messages_.push_back(message); return true; } - virtual void GoAway() OVERRIDE; - std::vector<IPC::Message> received_messages_; - WebSocketDispatcherHostTest* owner_; }; class WebSocketDispatcherHostTest : public ::testing::Test { @@ -63,17 +55,12 @@ class WebSocketDispatcherHostTest : public ::testing::Test { virtual ~WebSocketDispatcherHostTest() {} - void GoAway(int routing_id) { - gone_hosts_.push_back(routing_id); - } - protected: scoped_refptr<WebSocketDispatcherHost> dispatcher_host_; // Stores allocated MockWebSocketHost instances. Doesn't take ownership of // them. std::vector<MockWebSocketHost*> mock_hosts_; - std::vector<int> gone_hosts_; private: net::URLRequestContext* OnGetRequestContext() { @@ -82,16 +69,12 @@ class WebSocketDispatcherHostTest : public ::testing::Test { WebSocketHost* CreateWebSocketHost(int routing_id) { MockWebSocketHost* host = - new MockWebSocketHost(routing_id, dispatcher_host_.get(), NULL, this); + new MockWebSocketHost(routing_id, dispatcher_host_.get(), NULL); mock_hosts_.push_back(host); return host; } }; -void MockWebSocketHost::GoAway() { - owner_->GoAway(routing_id()); -} - TEST_F(WebSocketDispatcherHostTest, Construct) { // Do nothing. } @@ -173,29 +156,5 @@ TEST_F(WebSocketDispatcherHostTest, SendFrame) { } } -TEST_F(WebSocketDispatcherHostTest, Destruct) { - WebSocketHostMsg_AddChannelRequest message1( - 123, GURL("ws://example.com/test"), std::vector<std::string>(), - url::Origin("http://example.com"), -1); - WebSocketHostMsg_AddChannelRequest message2( - 456, GURL("ws://example.com/test2"), std::vector<std::string>(), - url::Origin("http://example.com"), -1); - - ASSERT_TRUE(dispatcher_host_->OnMessageReceived(message1)); - ASSERT_TRUE(dispatcher_host_->OnMessageReceived(message2)); - - ASSERT_EQ(2u, mock_hosts_.size()); - - mock_hosts_.clear(); - dispatcher_host_ = NULL; - - ASSERT_EQ(2u, gone_hosts_.size()); - // The gone_hosts_ ordering is not predictable because it depends on the - // hash_map ordering. - std::sort(gone_hosts_.begin(), gone_hosts_.end()); - EXPECT_EQ(123, gone_hosts_[0]); - EXPECT_EQ(456, gone_hosts_[1]); -} - } // namespace } // namespace content |