| OLD | NEW |
|---|
| (Empty) |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include <algorithm> | |
| 6 #include <memory> | |
| 7 #include <vector> | |
| 8 | |
| 9 #include "content/browser/websockets/websocket_manager.h" | |
| 10 #include "content/public/test/test_browser_thread_bundle.h" | |
| 11 #include "ipc/ipc_message.h" | |
| 12 #include "testing/gtest/include/gtest/gtest.h" | |
| 13 #include "url/gurl.h" | |
| 14 #include "url/origin.h" | |
| 15 | |
| 16 namespace content { | |
| 17 namespace { | |
| 18 | |
| 19 // This number is unlikely to occur by chance. | |
| 20 static const int kMagicRenderProcessId = 506116062; | |
| 21 | |
| 22 class TestWebSocketImpl : public WebSocketImpl { | |
| 23 public: | |
| 24 TestWebSocketImpl(Delegate* delegate, | |
| 25 mojom::WebSocketRequest request, | |
| 26 int frame_id, | |
| 27 base::TimeDelta delay) | |
| 28 : WebSocketImpl(delegate, std::move(request), frame_id, delay) {} | |
| 29 | |
| 30 base::TimeDelta delay() const { return delay_; } | |
| 31 | |
| 32 void SimulateConnectionError() { | |
| 33 OnConnectionError(); | |
| 34 } | |
| 35 }; | |
| 36 | |
| 37 class TestWebSocketManager : public WebSocketManager { | |
| 38 public: | |
| 39 TestWebSocketManager() | |
| 40 : WebSocketManager(kMagicRenderProcessId, nullptr) {} | |
| 41 | |
| 42 const std::vector<TestWebSocketImpl*>& sockets() const { | |
| 43 return sockets_; | |
| 44 } | |
| 45 | |
| 46 int num_pending_connections() const { | |
| 47 return num_pending_connections_; | |
| 48 } | |
| 49 int64_t num_failed_connections() const { | |
| 50 return num_current_failed_connections_ + num_previous_failed_connections_; | |
| 51 } | |
| 52 int64_t num_succeeded_connections() const { | |
| 53 return num_current_succeeded_connections_ + | |
| 54 num_previous_succeeded_connections_; | |
| 55 } | |
| 56 | |
| 57 void DoCreateWebSocket(mojom::WebSocketRequest request) { | |
| 58 WebSocketManager::DoCreateWebSocket(MSG_ROUTING_NONE, std::move(request)); | |
| 59 } | |
| 60 | |
| 61 private: | |
| 62 WebSocketImpl* CreateWebSocketImpl(WebSocketImpl::Delegate* delegate, | |
| 63 mojom::WebSocketRequest request, | |
| 64 int frame_id, | |
| 65 base::TimeDelta delay) override { | |
| 66 TestWebSocketImpl* impl = | |
| 67 new TestWebSocketImpl(delegate, std::move(request), frame_id, delay); | |
| 68 // We keep a vector of sockets here to track their creation order. | |
| 69 sockets_.push_back(impl); | |
| 70 return impl; | |
| 71 } | |
| 72 | |
| 73 void OnLostConnectionToClient(WebSocketImpl* impl) override { | |
| 74 auto it = std::find(sockets_.begin(), sockets_.end(), | |
| 75 static_cast<TestWebSocketImpl*>(impl)); | |
| 76 ASSERT_TRUE(it != sockets_.end()); | |
| 77 sockets_.erase(it); | |
| 78 | |
| 79 WebSocketManager::OnLostConnectionToClient(impl); | |
| 80 } | |
| 81 | |
| 82 std::vector<TestWebSocketImpl*> sockets_; | |
| 83 }; | |
| 84 | |
| 85 class WebSocketManagerTest : public ::testing::Test { | |
| 86 public: | |
| 87 WebSocketManagerTest() | |
| 88 : thread_bundle_(TestBrowserThreadBundle::IO_MAINLOOP) { | |
| 89 websocket_manager_.reset(new TestWebSocketManager()); | |
| 90 } | |
| 91 | |
| 92 void AddMultipleChannels(int number_of_channels) { | |
| 93 for (int i = 0; i < number_of_channels; ++i) { | |
| 94 mojom::WebSocketPtr websocket; | |
| 95 websocket_manager_->DoCreateWebSocket(mojo::GetProxy(&websocket)); | |
| 96 } | |
| 97 } | |
| 98 | |
| 99 void AddAndCancelMultipleChannels(int number_of_channels) { | |
| 100 for (int i = 0; i < number_of_channels; ++i) { | |
| 101 mojom::WebSocketPtr websocket; | |
| 102 websocket_manager_->DoCreateWebSocket(mojo::GetProxy(&websocket)); | |
| 103 websocket_manager_->sockets().back()->SimulateConnectionError(); | |
| 104 } | |
| 105 } | |
| 106 | |
| 107 TestWebSocketManager* websocket_manager() { return websocket_manager_.get(); } | |
| 108 | |
| 109 private: | |
| 110 TestBrowserThreadBundle thread_bundle_; | |
| 111 std::unique_ptr<TestWebSocketManager> websocket_manager_; | |
| 112 }; | |
| 113 | |
| 114 TEST_F(WebSocketManagerTest, Construct) { | |
| 115 // Do nothing. | |
| 116 } | |
| 117 | |
| 118 TEST_F(WebSocketManagerTest, CreateWebSocket) { | |
| 119 mojom::WebSocketPtr websocket; | |
| 120 | |
| 121 websocket_manager()->DoCreateWebSocket(mojo::GetProxy(&websocket)); | |
| 122 | |
| 123 EXPECT_EQ(1U, websocket_manager()->sockets().size()); | |
| 124 } | |
| 125 | |
| 126 TEST_F(WebSocketManagerTest, SendFrameButNotConnectedYet) { | |
| 127 mojom::WebSocketPtr websocket; | |
| 128 | |
| 129 websocket_manager()->DoCreateWebSocket(mojo::GetProxy(&websocket)); | |
| 130 | |
| 131 // This should not crash. | |
| 132 mojo::Array<uint8_t> data; | |
| 133 websocket->SendFrame( | |
| 134 true, mojom::WebSocketMessageType::TEXT, std::move(data)); | |
| 135 } | |
| 136 | |
| 137 TEST_F(WebSocketManagerTest, DelayFor4thPendingConnectionIsZero) { | |
| 138 AddMultipleChannels(4); | |
| 139 | |
| 140 EXPECT_EQ(4, websocket_manager()->num_pending_connections()); | |
| 141 EXPECT_EQ(0, websocket_manager()->num_failed_connections()); | |
| 142 EXPECT_EQ(0, websocket_manager()->num_succeeded_connections()); | |
| 143 | |
| 144 ASSERT_EQ(4U, websocket_manager()->sockets().size()); | |
| 145 EXPECT_EQ(base::TimeDelta(), websocket_manager()->sockets()[3]->delay()); | |
| 146 } | |
| 147 | |
| 148 TEST_F(WebSocketManagerTest, DelayFor8thPendingConnectionIsNonZero) { | |
| 149 AddMultipleChannels(8); | |
| 150 | |
| 151 EXPECT_EQ(8, websocket_manager()->num_pending_connections()); | |
| 152 EXPECT_EQ(0, websocket_manager()->num_failed_connections()); | |
| 153 EXPECT_EQ(0, websocket_manager()->num_succeeded_connections()); | |
| 154 | |
| 155 ASSERT_EQ(8U, websocket_manager()->sockets().size()); | |
| 156 EXPECT_LT(base::TimeDelta(), websocket_manager()->sockets()[7]->delay()); | |
| 157 } | |
| 158 | |
| 159 TEST_F(WebSocketManagerTest, DelayFor17thPendingConnection) { | |
| 160 AddMultipleChannels(17); | |
| 161 | |
| 162 EXPECT_EQ(17, websocket_manager()->num_pending_connections()); | |
| 163 EXPECT_EQ(0, websocket_manager()->num_failed_connections()); | |
| 164 EXPECT_EQ(0, websocket_manager()->num_succeeded_connections()); | |
| 165 | |
| 166 ASSERT_EQ(17U, websocket_manager()->sockets().size()); | |
| 167 EXPECT_LE(base::TimeDelta::FromMilliseconds(1000), | |
| 168 websocket_manager()->sockets()[16]->delay()); | |
| 169 EXPECT_GE(base::TimeDelta::FromMilliseconds(5000), | |
| 170 websocket_manager()->sockets()[16]->delay()); | |
| 171 } | |
| 172 | |
| 173 // The 256th connection is rejected by per-renderer WebSocket throttling. | |
| 174 // This is not counted as a failure. | |
| 175 TEST_F(WebSocketManagerTest, Rejects256thPendingConnection) { | |
| 176 AddMultipleChannels(256); | |
| 177 | |
| 178 EXPECT_EQ(255, websocket_manager()->num_pending_connections()); | |
| 179 EXPECT_EQ(0, websocket_manager()->num_failed_connections()); | |
| 180 EXPECT_EQ(0, websocket_manager()->num_succeeded_connections()); | |
| 181 | |
| 182 ASSERT_EQ(255U, websocket_manager()->sockets().size()); | |
| 183 } | |
| 184 | |
| 185 TEST_F(WebSocketManagerTest, DelayIsZeroAfter3FailedConnections) { | |
| 186 AddAndCancelMultipleChannels(3); | |
| 187 | |
| 188 EXPECT_EQ(0, websocket_manager()->num_pending_connections()); | |
| 189 EXPECT_EQ(3, websocket_manager()->num_failed_connections()); | |
| 190 EXPECT_EQ(0, websocket_manager()->num_succeeded_connections()); | |
| 191 | |
| 192 AddMultipleChannels(1); | |
| 193 | |
| 194 ASSERT_EQ(1U, websocket_manager()->sockets().size()); | |
| 195 EXPECT_EQ(base::TimeDelta(), websocket_manager()->sockets()[0]->delay()); | |
| 196 } | |
| 197 | |
| 198 TEST_F(WebSocketManagerTest, DelayIsNonZeroAfter7FailedConnections) { | |
| 199 AddAndCancelMultipleChannels(7); | |
| 200 | |
| 201 EXPECT_EQ(0, websocket_manager()->num_pending_connections()); | |
| 202 EXPECT_EQ(7, websocket_manager()->num_failed_connections()); | |
| 203 EXPECT_EQ(0, websocket_manager()->num_succeeded_connections()); | |
| 204 | |
| 205 AddMultipleChannels(1); | |
| 206 | |
| 207 ASSERT_EQ(1U, websocket_manager()->sockets().size()); | |
| 208 EXPECT_LT(base::TimeDelta(), websocket_manager()->sockets()[0]->delay()); | |
| 209 } | |
| 210 | |
| 211 TEST_F(WebSocketManagerTest, DelayAfter16FailedConnections) { | |
| 212 AddAndCancelMultipleChannels(16); | |
| 213 | |
| 214 EXPECT_EQ(0, websocket_manager()->num_pending_connections()); | |
| 215 EXPECT_EQ(16, websocket_manager()->num_failed_connections()); | |
| 216 EXPECT_EQ(0, websocket_manager()->num_succeeded_connections()); | |
| 217 | |
| 218 AddMultipleChannels(1); | |
| 219 | |
| 220 ASSERT_EQ(1U, websocket_manager()->sockets().size()); | |
| 221 EXPECT_LE(base::TimeDelta::FromMilliseconds(1000), | |
| 222 websocket_manager()->sockets()[0]->delay()); | |
| 223 EXPECT_GE(base::TimeDelta::FromMilliseconds(5000), | |
| 224 websocket_manager()->sockets()[0]->delay()); | |
| 225 } | |
| 226 | |
| 227 TEST_F(WebSocketManagerTest, NotRejectedAfter255FailedConnections) { | |
| 228 AddAndCancelMultipleChannels(255); | |
| 229 | |
| 230 EXPECT_EQ(0, websocket_manager()->num_pending_connections()); | |
| 231 EXPECT_EQ(255, websocket_manager()->num_failed_connections()); | |
| 232 EXPECT_EQ(0, websocket_manager()->num_succeeded_connections()); | |
| 233 | |
| 234 AddMultipleChannels(1); | |
| 235 | |
| 236 EXPECT_EQ(1, websocket_manager()->num_pending_connections()); | |
| 237 EXPECT_EQ(255, websocket_manager()->num_failed_connections()); | |
| 238 EXPECT_EQ(0, websocket_manager()->num_succeeded_connections()); | |
| 239 } | |
| 240 | |
| 241 } // namespace | |
| 242 } // namespace content | |
| OLD | NEW |
|---|
Chromium Code Reviews
Issue 2225133002:
Revert "Port WebSockets to Mojo IPC" (Closed)
Base URL: https://chromium.googlesource.com/chromium/src.git@master