| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include <algorithm> | 5 #include <algorithm> |
| 6 #include <memory> | 6 #include <memory> |
| 7 #include <vector> | 7 #include <vector> |
| 8 | 8 |
| 9 #include "content/browser/websockets/websocket_manager.h" | 9 #include "content/browser/websockets/websocket_manager.h" |
| 10 #include "content/public/test/test_browser_thread_bundle.h" | 10 #include "content/public/test/test_browser_thread_bundle.h" |
| (...skipping 80 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 91 class WebSocketManagerTest : public ::testing::Test { | 91 class WebSocketManagerTest : public ::testing::Test { |
| 92 public: | 92 public: |
| 93 WebSocketManagerTest() | 93 WebSocketManagerTest() |
| 94 : thread_bundle_(TestBrowserThreadBundle::IO_MAINLOOP) { | 94 : thread_bundle_(TestBrowserThreadBundle::IO_MAINLOOP) { |
| 95 websocket_manager_.reset(new TestWebSocketManager()); | 95 websocket_manager_.reset(new TestWebSocketManager()); |
| 96 } | 96 } |
| 97 | 97 |
| 98 void AddMultipleChannels(int number_of_channels) { | 98 void AddMultipleChannels(int number_of_channels) { |
| 99 for (int i = 0; i < number_of_channels; ++i) { | 99 for (int i = 0; i < number_of_channels; ++i) { |
| 100 blink::mojom::WebSocketPtr websocket; | 100 blink::mojom::WebSocketPtr websocket; |
| 101 websocket_manager_->DoCreateWebSocket(mojo::GetProxy(&websocket)); | 101 websocket_manager_->DoCreateWebSocket(mojo::MakeRequest(&websocket)); |
| 102 } | 102 } |
| 103 } | 103 } |
| 104 | 104 |
| 105 void AddAndCancelMultipleChannels(int number_of_channels) { | 105 void AddAndCancelMultipleChannels(int number_of_channels) { |
| 106 for (int i = 0; i < number_of_channels; ++i) { | 106 for (int i = 0; i < number_of_channels; ++i) { |
| 107 blink::mojom::WebSocketPtr websocket; | 107 blink::mojom::WebSocketPtr websocket; |
| 108 websocket_manager_->DoCreateWebSocket(mojo::GetProxy(&websocket)); | 108 websocket_manager_->DoCreateWebSocket(mojo::MakeRequest(&websocket)); |
| 109 websocket_manager_->sockets().back()->SimulateConnectionError(); | 109 websocket_manager_->sockets().back()->SimulateConnectionError(); |
| 110 } | 110 } |
| 111 } | 111 } |
| 112 | 112 |
| 113 TestWebSocketManager* websocket_manager() { return websocket_manager_.get(); } | 113 TestWebSocketManager* websocket_manager() { return websocket_manager_.get(); } |
| 114 | 114 |
| 115 private: | 115 private: |
| 116 TestBrowserThreadBundle thread_bundle_; | 116 TestBrowserThreadBundle thread_bundle_; |
| 117 std::unique_ptr<TestWebSocketManager> websocket_manager_; | 117 std::unique_ptr<TestWebSocketManager> websocket_manager_; |
| 118 }; | 118 }; |
| 119 | 119 |
| 120 TEST_F(WebSocketManagerTest, Construct) { | 120 TEST_F(WebSocketManagerTest, Construct) { |
| 121 // Do nothing. | 121 // Do nothing. |
| 122 } | 122 } |
| 123 | 123 |
| 124 TEST_F(WebSocketManagerTest, CreateWebSocket) { | 124 TEST_F(WebSocketManagerTest, CreateWebSocket) { |
| 125 blink::mojom::WebSocketPtr websocket; | 125 blink::mojom::WebSocketPtr websocket; |
| 126 | 126 |
| 127 websocket_manager()->DoCreateWebSocket(mojo::GetProxy(&websocket)); | 127 websocket_manager()->DoCreateWebSocket(mojo::MakeRequest(&websocket)); |
| 128 | 128 |
| 129 EXPECT_EQ(1U, websocket_manager()->sockets().size()); | 129 EXPECT_EQ(1U, websocket_manager()->sockets().size()); |
| 130 } | 130 } |
| 131 | 131 |
| 132 TEST_F(WebSocketManagerTest, SendFrameButNotConnectedYet) { | 132 TEST_F(WebSocketManagerTest, SendFrameButNotConnectedYet) { |
| 133 blink::mojom::WebSocketPtr websocket; | 133 blink::mojom::WebSocketPtr websocket; |
| 134 | 134 |
| 135 websocket_manager()->DoCreateWebSocket(mojo::GetProxy(&websocket)); | 135 websocket_manager()->DoCreateWebSocket(mojo::MakeRequest(&websocket)); |
| 136 | 136 |
| 137 // This should not crash. | 137 // This should not crash. |
| 138 mojo::Array<uint8_t> data; | 138 mojo::Array<uint8_t> data; |
| 139 websocket->SendFrame( | 139 websocket->SendFrame( |
| 140 true, blink::mojom::WebSocketMessageType::TEXT, std::move(data)); | 140 true, blink::mojom::WebSocketMessageType::TEXT, std::move(data)); |
| 141 } | 141 } |
| 142 | 142 |
| 143 TEST_F(WebSocketManagerTest, DelayFor4thPendingConnectionIsZero) { | 143 TEST_F(WebSocketManagerTest, DelayFor4thPendingConnectionIsZero) { |
| 144 AddMultipleChannels(4); | 144 AddMultipleChannels(4); |
| 145 | 145 |
| (...skipping 93 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 239 | 239 |
| 240 AddMultipleChannels(1); | 240 AddMultipleChannels(1); |
| 241 | 241 |
| 242 EXPECT_EQ(1, websocket_manager()->num_pending_connections()); | 242 EXPECT_EQ(1, websocket_manager()->num_pending_connections()); |
| 243 EXPECT_EQ(255, websocket_manager()->num_failed_connections()); | 243 EXPECT_EQ(255, websocket_manager()->num_failed_connections()); |
| 244 EXPECT_EQ(0, websocket_manager()->num_succeeded_connections()); | 244 EXPECT_EQ(0, websocket_manager()->num_succeeded_connections()); |
| 245 } | 245 } |
| 246 | 246 |
| 247 } // namespace | 247 } // namespace |
| 248 } // namespace content | 248 } // namespace content |
| OLD | NEW |