| Index: content/browser/websockets/websocket_manager_unittest.cc
|
| diff --git a/content/browser/websockets/websocket_manager_unittest.cc b/content/browser/websockets/websocket_manager_unittest.cc
|
| deleted file mode 100644
|
| index fc5f94f42afdafc285055b6c0988188335f1dbeb..0000000000000000000000000000000000000000
|
| --- a/content/browser/websockets/websocket_manager_unittest.cc
|
| +++ /dev/null
|
| @@ -1,242 +0,0 @@
|
| -// Copyright 2013 The Chromium Authors. All rights reserved.
|
| -// Use of this source code is governed by a BSD-style license that can be
|
| -// found in the LICENSE file.
|
| -
|
| -#include <algorithm>
|
| -#include <memory>
|
| -#include <vector>
|
| -
|
| -#include "content/browser/websockets/websocket_manager.h"
|
| -#include "content/public/test/test_browser_thread_bundle.h"
|
| -#include "ipc/ipc_message.h"
|
| -#include "testing/gtest/include/gtest/gtest.h"
|
| -#include "url/gurl.h"
|
| -#include "url/origin.h"
|
| -
|
| -namespace content {
|
| -namespace {
|
| -
|
| -// This number is unlikely to occur by chance.
|
| -static const int kMagicRenderProcessId = 506116062;
|
| -
|
| -class TestWebSocketImpl : public WebSocketImpl {
|
| - public:
|
| - TestWebSocketImpl(Delegate* delegate,
|
| - mojom::WebSocketRequest request,
|
| - int frame_id,
|
| - base::TimeDelta delay)
|
| - : WebSocketImpl(delegate, std::move(request), frame_id, delay) {}
|
| -
|
| - base::TimeDelta delay() const { return delay_; }
|
| -
|
| - void SimulateConnectionError() {
|
| - OnConnectionError();
|
| - }
|
| -};
|
| -
|
| -class TestWebSocketManager : public WebSocketManager {
|
| - public:
|
| - TestWebSocketManager()
|
| - : WebSocketManager(kMagicRenderProcessId, nullptr) {}
|
| -
|
| - const std::vector<TestWebSocketImpl*>& sockets() const {
|
| - return sockets_;
|
| - }
|
| -
|
| - int num_pending_connections() const {
|
| - return num_pending_connections_;
|
| - }
|
| - int64_t num_failed_connections() const {
|
| - return num_current_failed_connections_ + num_previous_failed_connections_;
|
| - }
|
| - int64_t num_succeeded_connections() const {
|
| - return num_current_succeeded_connections_ +
|
| - num_previous_succeeded_connections_;
|
| - }
|
| -
|
| - void DoCreateWebSocket(mojom::WebSocketRequest request) {
|
| - WebSocketManager::DoCreateWebSocket(MSG_ROUTING_NONE, std::move(request));
|
| - }
|
| -
|
| - private:
|
| - WebSocketImpl* CreateWebSocketImpl(WebSocketImpl::Delegate* delegate,
|
| - mojom::WebSocketRequest request,
|
| - int frame_id,
|
| - base::TimeDelta delay) override {
|
| - TestWebSocketImpl* impl =
|
| - new TestWebSocketImpl(delegate, std::move(request), frame_id, delay);
|
| - // We keep a vector of sockets here to track their creation order.
|
| - sockets_.push_back(impl);
|
| - return impl;
|
| - }
|
| -
|
| - void OnLostConnectionToClient(WebSocketImpl* impl) override {
|
| - auto it = std::find(sockets_.begin(), sockets_.end(),
|
| - static_cast<TestWebSocketImpl*>(impl));
|
| - ASSERT_TRUE(it != sockets_.end());
|
| - sockets_.erase(it);
|
| -
|
| - WebSocketManager::OnLostConnectionToClient(impl);
|
| - }
|
| -
|
| - std::vector<TestWebSocketImpl*> sockets_;
|
| -};
|
| -
|
| -class WebSocketManagerTest : public ::testing::Test {
|
| - public:
|
| - WebSocketManagerTest()
|
| - : thread_bundle_(TestBrowserThreadBundle::IO_MAINLOOP) {
|
| - websocket_manager_.reset(new TestWebSocketManager());
|
| - }
|
| -
|
| - void AddMultipleChannels(int number_of_channels) {
|
| - for (int i = 0; i < number_of_channels; ++i) {
|
| - mojom::WebSocketPtr websocket;
|
| - websocket_manager_->DoCreateWebSocket(mojo::GetProxy(&websocket));
|
| - }
|
| - }
|
| -
|
| - void AddAndCancelMultipleChannels(int number_of_channels) {
|
| - for (int i = 0; i < number_of_channels; ++i) {
|
| - mojom::WebSocketPtr websocket;
|
| - websocket_manager_->DoCreateWebSocket(mojo::GetProxy(&websocket));
|
| - websocket_manager_->sockets().back()->SimulateConnectionError();
|
| - }
|
| - }
|
| -
|
| - TestWebSocketManager* websocket_manager() { return websocket_manager_.get(); }
|
| -
|
| - private:
|
| - TestBrowserThreadBundle thread_bundle_;
|
| - std::unique_ptr<TestWebSocketManager> websocket_manager_;
|
| -};
|
| -
|
| -TEST_F(WebSocketManagerTest, Construct) {
|
| - // Do nothing.
|
| -}
|
| -
|
| -TEST_F(WebSocketManagerTest, CreateWebSocket) {
|
| - mojom::WebSocketPtr websocket;
|
| -
|
| - websocket_manager()->DoCreateWebSocket(mojo::GetProxy(&websocket));
|
| -
|
| - EXPECT_EQ(1U, websocket_manager()->sockets().size());
|
| -}
|
| -
|
| -TEST_F(WebSocketManagerTest, SendFrameButNotConnectedYet) {
|
| - mojom::WebSocketPtr websocket;
|
| -
|
| - websocket_manager()->DoCreateWebSocket(mojo::GetProxy(&websocket));
|
| -
|
| - // This should not crash.
|
| - mojo::Array<uint8_t> data;
|
| - websocket->SendFrame(
|
| - true, mojom::WebSocketMessageType::TEXT, std::move(data));
|
| -}
|
| -
|
| -TEST_F(WebSocketManagerTest, DelayFor4thPendingConnectionIsZero) {
|
| - AddMultipleChannels(4);
|
| -
|
| - EXPECT_EQ(4, websocket_manager()->num_pending_connections());
|
| - EXPECT_EQ(0, websocket_manager()->num_failed_connections());
|
| - EXPECT_EQ(0, websocket_manager()->num_succeeded_connections());
|
| -
|
| - ASSERT_EQ(4U, websocket_manager()->sockets().size());
|
| - EXPECT_EQ(base::TimeDelta(), websocket_manager()->sockets()[3]->delay());
|
| -}
|
| -
|
| -TEST_F(WebSocketManagerTest, DelayFor8thPendingConnectionIsNonZero) {
|
| - AddMultipleChannels(8);
|
| -
|
| - EXPECT_EQ(8, websocket_manager()->num_pending_connections());
|
| - EXPECT_EQ(0, websocket_manager()->num_failed_connections());
|
| - EXPECT_EQ(0, websocket_manager()->num_succeeded_connections());
|
| -
|
| - ASSERT_EQ(8U, websocket_manager()->sockets().size());
|
| - EXPECT_LT(base::TimeDelta(), websocket_manager()->sockets()[7]->delay());
|
| -}
|
| -
|
| -TEST_F(WebSocketManagerTest, DelayFor17thPendingConnection) {
|
| - AddMultipleChannels(17);
|
| -
|
| - EXPECT_EQ(17, websocket_manager()->num_pending_connections());
|
| - EXPECT_EQ(0, websocket_manager()->num_failed_connections());
|
| - EXPECT_EQ(0, websocket_manager()->num_succeeded_connections());
|
| -
|
| - ASSERT_EQ(17U, websocket_manager()->sockets().size());
|
| - EXPECT_LE(base::TimeDelta::FromMilliseconds(1000),
|
| - websocket_manager()->sockets()[16]->delay());
|
| - EXPECT_GE(base::TimeDelta::FromMilliseconds(5000),
|
| - websocket_manager()->sockets()[16]->delay());
|
| -}
|
| -
|
| -// The 256th connection is rejected by per-renderer WebSocket throttling.
|
| -// This is not counted as a failure.
|
| -TEST_F(WebSocketManagerTest, Rejects256thPendingConnection) {
|
| - AddMultipleChannels(256);
|
| -
|
| - EXPECT_EQ(255, websocket_manager()->num_pending_connections());
|
| - EXPECT_EQ(0, websocket_manager()->num_failed_connections());
|
| - EXPECT_EQ(0, websocket_manager()->num_succeeded_connections());
|
| -
|
| - ASSERT_EQ(255U, websocket_manager()->sockets().size());
|
| -}
|
| -
|
| -TEST_F(WebSocketManagerTest, DelayIsZeroAfter3FailedConnections) {
|
| - AddAndCancelMultipleChannels(3);
|
| -
|
| - EXPECT_EQ(0, websocket_manager()->num_pending_connections());
|
| - EXPECT_EQ(3, websocket_manager()->num_failed_connections());
|
| - EXPECT_EQ(0, websocket_manager()->num_succeeded_connections());
|
| -
|
| - AddMultipleChannels(1);
|
| -
|
| - ASSERT_EQ(1U, websocket_manager()->sockets().size());
|
| - EXPECT_EQ(base::TimeDelta(), websocket_manager()->sockets()[0]->delay());
|
| -}
|
| -
|
| -TEST_F(WebSocketManagerTest, DelayIsNonZeroAfter7FailedConnections) {
|
| - AddAndCancelMultipleChannels(7);
|
| -
|
| - EXPECT_EQ(0, websocket_manager()->num_pending_connections());
|
| - EXPECT_EQ(7, websocket_manager()->num_failed_connections());
|
| - EXPECT_EQ(0, websocket_manager()->num_succeeded_connections());
|
| -
|
| - AddMultipleChannels(1);
|
| -
|
| - ASSERT_EQ(1U, websocket_manager()->sockets().size());
|
| - EXPECT_LT(base::TimeDelta(), websocket_manager()->sockets()[0]->delay());
|
| -}
|
| -
|
| -TEST_F(WebSocketManagerTest, DelayAfter16FailedConnections) {
|
| - AddAndCancelMultipleChannels(16);
|
| -
|
| - EXPECT_EQ(0, websocket_manager()->num_pending_connections());
|
| - EXPECT_EQ(16, websocket_manager()->num_failed_connections());
|
| - EXPECT_EQ(0, websocket_manager()->num_succeeded_connections());
|
| -
|
| - AddMultipleChannels(1);
|
| -
|
| - ASSERT_EQ(1U, websocket_manager()->sockets().size());
|
| - EXPECT_LE(base::TimeDelta::FromMilliseconds(1000),
|
| - websocket_manager()->sockets()[0]->delay());
|
| - EXPECT_GE(base::TimeDelta::FromMilliseconds(5000),
|
| - websocket_manager()->sockets()[0]->delay());
|
| -}
|
| -
|
| -TEST_F(WebSocketManagerTest, NotRejectedAfter255FailedConnections) {
|
| - AddAndCancelMultipleChannels(255);
|
| -
|
| - EXPECT_EQ(0, websocket_manager()->num_pending_connections());
|
| - EXPECT_EQ(255, websocket_manager()->num_failed_connections());
|
| - EXPECT_EQ(0, websocket_manager()->num_succeeded_connections());
|
| -
|
| - AddMultipleChannels(1);
|
| -
|
| - EXPECT_EQ(1, websocket_manager()->num_pending_connections());
|
| - EXPECT_EQ(255, websocket_manager()->num_failed_connections());
|
| - EXPECT_EQ(0, websocket_manager()->num_succeeded_connections());
|
| -}
|
| -
|
| -} // namespace
|
| -} // namespace content
|
|
|