Index: net/socket/websocket_endpoint_lock_manager_unittest.cc |
diff --git a/net/socket/websocket_endpoint_lock_manager_unittest.cc b/net/socket/websocket_endpoint_lock_manager_unittest.cc |
deleted file mode 100644 |
index dbe8494f4f88baaf304d4b9a4e1c933b629e92a3..0000000000000000000000000000000000000000 |
--- a/net/socket/websocket_endpoint_lock_manager_unittest.cc |
+++ /dev/null |
@@ -1,321 +0,0 @@ |
-// Copyright 2014 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 "net/socket/websocket_endpoint_lock_manager.h" |
- |
-#include "base/message_loop/message_loop.h" |
-#include "base/run_loop.h" |
-#include "base/time/time.h" |
-#include "net/base/net_errors.h" |
-#include "net/socket/next_proto.h" |
-#include "net/socket/socket_test_util.h" |
-#include "net/socket/stream_socket.h" |
-#include "testing/gtest/include/gtest/gtest.h" |
- |
-namespace net { |
- |
-namespace { |
- |
-// A StreamSocket implementation with no functionality at all. |
-// TODO(ricea): If you need to use this in another file, please move it to |
-// socket_test_util.h. |
-class FakeStreamSocket : public StreamSocket { |
- public: |
- FakeStreamSocket() {} |
- |
- // StreamSocket implementation |
- int Connect(const CompletionCallback& callback) override { |
- return ERR_FAILED; |
- } |
- |
- void Disconnect() override { return; } |
- |
- bool IsConnected() const override { return false; } |
- |
- bool IsConnectedAndIdle() const override { return false; } |
- |
- int GetPeerAddress(IPEndPoint* address) const override { return ERR_FAILED; } |
- |
- int GetLocalAddress(IPEndPoint* address) const override { return ERR_FAILED; } |
- |
- const BoundNetLog& NetLog() const override { return bound_net_log_; } |
- |
- void SetSubresourceSpeculation() override { return; } |
- void SetOmniboxSpeculation() override { return; } |
- |
- bool WasEverUsed() const override { return false; } |
- |
- bool UsingTCPFastOpen() const override { return false; } |
- |
- bool WasNpnNegotiated() const override { return false; } |
- |
- NextProto GetNegotiatedProtocol() const override { return kProtoUnknown; } |
- |
- bool GetSSLInfo(SSLInfo* ssl_info) override { return false; } |
- |
- // Socket implementation |
- int Read(IOBuffer* buf, |
- int buf_len, |
- const CompletionCallback& callback) override { |
- return ERR_FAILED; |
- } |
- |
- int Write(IOBuffer* buf, |
- int buf_len, |
- const CompletionCallback& callback) override { |
- return ERR_FAILED; |
- } |
- |
- int SetReceiveBufferSize(int32 size) override { return ERR_FAILED; } |
- |
- int SetSendBufferSize(int32 size) override { return ERR_FAILED; } |
- |
- private: |
- BoundNetLog bound_net_log_; |
- |
- DISALLOW_COPY_AND_ASSIGN(FakeStreamSocket); |
-}; |
- |
-class FakeWaiter : public WebSocketEndpointLockManager::Waiter { |
- public: |
- FakeWaiter() : called_(false) {} |
- |
- void GotEndpointLock() override { |
- CHECK(!called_); |
- called_ = true; |
- } |
- |
- bool called() const { return called_; } |
- |
- private: |
- bool called_; |
-}; |
- |
-class BlockingWaiter : public FakeWaiter { |
- public: |
- void WaitForLock() { |
- while (!called()) { |
- run_loop_.Run(); |
- } |
- } |
- |
- void GotEndpointLock() override { |
- FakeWaiter::GotEndpointLock(); |
- run_loop_.Quit(); |
- } |
- |
- private: |
- base::RunLoop run_loop_; |
-}; |
- |
-class WebSocketEndpointLockManagerTest : public ::testing::Test { |
- protected: |
- WebSocketEndpointLockManagerTest() |
- : instance_(WebSocketEndpointLockManager::GetInstance()) {} |
- ~WebSocketEndpointLockManagerTest() override { |
- // Permit any pending asynchronous unlock operations to complete. |
- RunUntilIdle(); |
- // If this check fails then subsequent tests may fail. |
- CHECK(instance_->IsEmpty()); |
- } |
- |
- WebSocketEndpointLockManager* instance() const { return instance_; } |
- |
- IPEndPoint DummyEndpoint() { |
- IPAddressNumber ip_address_number; |
- CHECK(ParseIPLiteralToNumber("127.0.0.1", &ip_address_number)); |
- return IPEndPoint(ip_address_number, 80); |
- } |
- |
- void UnlockDummyEndpoint(int times) { |
- for (int i = 0; i < times; ++i) { |
- instance()->UnlockEndpoint(DummyEndpoint()); |
- RunUntilIdle(); |
- } |
- } |
- |
- static void RunUntilIdle() { base::RunLoop().RunUntilIdle(); } |
- |
- WebSocketEndpointLockManager* const instance_; |
- ScopedWebSocketEndpointZeroUnlockDelay zero_unlock_delay_; |
-}; |
- |
-TEST_F(WebSocketEndpointLockManagerTest, GetInstanceWorks) { |
- // All the work is done by the test framework. |
-} |
- |
-TEST_F(WebSocketEndpointLockManagerTest, LockEndpointReturnsOkOnce) { |
- FakeWaiter waiters[2]; |
- EXPECT_EQ(OK, instance()->LockEndpoint(DummyEndpoint(), &waiters[0])); |
- EXPECT_EQ(ERR_IO_PENDING, |
- instance()->LockEndpoint(DummyEndpoint(), &waiters[1])); |
- |
- UnlockDummyEndpoint(2); |
-} |
- |
-TEST_F(WebSocketEndpointLockManagerTest, GotEndpointLockNotCalledOnOk) { |
- FakeWaiter waiter; |
- EXPECT_EQ(OK, instance()->LockEndpoint(DummyEndpoint(), &waiter)); |
- RunUntilIdle(); |
- EXPECT_FALSE(waiter.called()); |
- |
- UnlockDummyEndpoint(1); |
-} |
- |
-TEST_F(WebSocketEndpointLockManagerTest, GotEndpointLockNotCalledImmediately) { |
- FakeWaiter waiters[2]; |
- EXPECT_EQ(OK, instance()->LockEndpoint(DummyEndpoint(), &waiters[0])); |
- EXPECT_EQ(ERR_IO_PENDING, |
- instance()->LockEndpoint(DummyEndpoint(), &waiters[1])); |
- RunUntilIdle(); |
- EXPECT_FALSE(waiters[1].called()); |
- |
- UnlockDummyEndpoint(2); |
-} |
- |
-TEST_F(WebSocketEndpointLockManagerTest, GotEndpointLockCalledWhenUnlocked) { |
- FakeWaiter waiters[2]; |
- EXPECT_EQ(OK, instance()->LockEndpoint(DummyEndpoint(), &waiters[0])); |
- EXPECT_EQ(ERR_IO_PENDING, |
- instance()->LockEndpoint(DummyEndpoint(), &waiters[1])); |
- instance()->UnlockEndpoint(DummyEndpoint()); |
- RunUntilIdle(); |
- EXPECT_TRUE(waiters[1].called()); |
- |
- UnlockDummyEndpoint(1); |
-} |
- |
-TEST_F(WebSocketEndpointLockManagerTest, |
- EndpointUnlockedIfWaiterAlreadyDeleted) { |
- FakeWaiter first_lock_holder; |
- EXPECT_EQ(OK, instance()->LockEndpoint(DummyEndpoint(), &first_lock_holder)); |
- |
- { |
- FakeWaiter short_lived_waiter; |
- EXPECT_EQ(ERR_IO_PENDING, |
- instance()->LockEndpoint(DummyEndpoint(), &short_lived_waiter)); |
- } |
- |
- instance()->UnlockEndpoint(DummyEndpoint()); |
- RunUntilIdle(); |
- |
- FakeWaiter second_lock_holder; |
- EXPECT_EQ(OK, instance()->LockEndpoint(DummyEndpoint(), &second_lock_holder)); |
- |
- UnlockDummyEndpoint(1); |
-} |
- |
-TEST_F(WebSocketEndpointLockManagerTest, RememberSocketWorks) { |
- FakeWaiter waiters[2]; |
- FakeStreamSocket dummy_socket; |
- EXPECT_EQ(OK, instance()->LockEndpoint(DummyEndpoint(), &waiters[0])); |
- EXPECT_EQ(ERR_IO_PENDING, |
- instance()->LockEndpoint(DummyEndpoint(), &waiters[1])); |
- |
- instance()->RememberSocket(&dummy_socket, DummyEndpoint()); |
- instance()->UnlockSocket(&dummy_socket); |
- RunUntilIdle(); |
- EXPECT_TRUE(waiters[1].called()); |
- |
- UnlockDummyEndpoint(1); |
-} |
- |
-// UnlockEndpoint() should cause any sockets remembered for this endpoint |
-// to be forgotten. |
-TEST_F(WebSocketEndpointLockManagerTest, SocketAssociationForgottenOnUnlock) { |
- FakeWaiter waiter; |
- FakeStreamSocket dummy_socket; |
- |
- EXPECT_EQ(OK, instance()->LockEndpoint(DummyEndpoint(), &waiter)); |
- instance()->RememberSocket(&dummy_socket, DummyEndpoint()); |
- instance()->UnlockEndpoint(DummyEndpoint()); |
- RunUntilIdle(); |
- EXPECT_TRUE(instance()->IsEmpty()); |
-} |
- |
-// When ownership of the endpoint is passed to a new waiter, the new waiter can |
-// call RememberSocket() again. |
-TEST_F(WebSocketEndpointLockManagerTest, NextWaiterCanCallRememberSocketAgain) { |
- FakeWaiter waiters[2]; |
- FakeStreamSocket dummy_sockets[2]; |
- EXPECT_EQ(OK, instance()->LockEndpoint(DummyEndpoint(), &waiters[0])); |
- EXPECT_EQ(ERR_IO_PENDING, |
- instance()->LockEndpoint(DummyEndpoint(), &waiters[1])); |
- |
- instance()->RememberSocket(&dummy_sockets[0], DummyEndpoint()); |
- instance()->UnlockEndpoint(DummyEndpoint()); |
- RunUntilIdle(); |
- EXPECT_TRUE(waiters[1].called()); |
- instance()->RememberSocket(&dummy_sockets[1], DummyEndpoint()); |
- |
- UnlockDummyEndpoint(1); |
-} |
- |
-// Calling UnlockSocket() after UnlockEndpoint() does nothing. |
-TEST_F(WebSocketEndpointLockManagerTest, |
- UnlockSocketAfterUnlockEndpointDoesNothing) { |
- FakeWaiter waiters[3]; |
- FakeStreamSocket dummy_socket; |
- |
- EXPECT_EQ(OK, instance()->LockEndpoint(DummyEndpoint(), &waiters[0])); |
- EXPECT_EQ(ERR_IO_PENDING, |
- instance()->LockEndpoint(DummyEndpoint(), &waiters[1])); |
- EXPECT_EQ(ERR_IO_PENDING, |
- instance()->LockEndpoint(DummyEndpoint(), &waiters[2])); |
- instance()->RememberSocket(&dummy_socket, DummyEndpoint()); |
- instance()->UnlockEndpoint(DummyEndpoint()); |
- instance()->UnlockSocket(&dummy_socket); |
- RunUntilIdle(); |
- EXPECT_TRUE(waiters[1].called()); |
- EXPECT_FALSE(waiters[2].called()); |
- |
- UnlockDummyEndpoint(2); |
-} |
- |
-// UnlockEndpoint() should always be asynchronous. |
-TEST_F(WebSocketEndpointLockManagerTest, UnlockEndpointIsAsynchronous) { |
- FakeWaiter waiters[2]; |
- EXPECT_EQ(OK, instance()->LockEndpoint(DummyEndpoint(), &waiters[0])); |
- EXPECT_EQ(ERR_IO_PENDING, |
- instance()->LockEndpoint(DummyEndpoint(), &waiters[1])); |
- |
- instance()->UnlockEndpoint(DummyEndpoint()); |
- EXPECT_FALSE(waiters[1].called()); |
- RunUntilIdle(); |
- EXPECT_TRUE(waiters[1].called()); |
- |
- UnlockDummyEndpoint(1); |
-} |
- |
-// UnlockEndpoint() should normally have a delay. |
-TEST_F(WebSocketEndpointLockManagerTest, UnlockEndpointIsDelayed) { |
- using base::TimeTicks; |
- |
- // This 1ms delay is too short for very slow environments (usually those |
- // running memory checkers). In those environments, the code takes >1ms to run |
- // and no delay is needed. Rather than increase the delay and slow down the |
- // test everywhere, the test doesn't explicitly verify that a delay has been |
- // applied. Instead it just verifies that the whole thing took >=1ms. 1ms is |
- // easily enough for normal compiles even on Android, so the fact that there |
- // is a delay is still checked on every platform. |
- const base::TimeDelta unlock_delay = base::TimeDelta::FromMilliseconds(1); |
- instance()->SetUnlockDelayForTesting(unlock_delay); |
- FakeWaiter fake_waiter; |
- BlockingWaiter blocking_waiter; |
- EXPECT_EQ(OK, instance()->LockEndpoint(DummyEndpoint(), &fake_waiter)); |
- EXPECT_EQ(ERR_IO_PENDING, |
- instance()->LockEndpoint(DummyEndpoint(), &blocking_waiter)); |
- |
- TimeTicks before_unlock = TimeTicks::Now(); |
- instance()->UnlockEndpoint(DummyEndpoint()); |
- blocking_waiter.WaitForLock(); |
- TimeTicks after_unlock = TimeTicks::Now(); |
- EXPECT_GE(after_unlock - before_unlock, unlock_delay); |
- instance()->SetUnlockDelayForTesting(base::TimeDelta()); |
- UnlockDummyEndpoint(1); |
-} |
- |
-} // namespace |
- |
-} // namespace net |