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 |
index 640d8b3bd5d3cb374aeadb43a9f3694c6ca60595..e0f479997221bf3bde129b18e2df25b960e842ca 100644 |
--- a/net/socket/websocket_endpoint_lock_manager_unittest.cc |
+++ b/net/socket/websocket_endpoint_lock_manager_unittest.cc |
@@ -14,8 +14,12 @@ |
#include "net/socket/next_proto.h" |
#include "net/socket/socket_test_util.h" |
#include "net/socket/stream_socket.h" |
+#include "net/test/gtest_util.h" |
+#include "testing/gmock/include/gmock/gmock.h" |
#include "testing/gtest/include/gtest/gtest.h" |
+using net::test::IsOk; |
+ |
namespace net { |
namespace { |
@@ -159,7 +163,7 @@ TEST_F(WebSocketEndpointLockManagerTest, GetInstanceWorks) { |
TEST_F(WebSocketEndpointLockManagerTest, LockEndpointReturnsOkOnce) { |
FakeWaiter waiters[2]; |
- EXPECT_EQ(OK, instance()->LockEndpoint(DummyEndpoint(), &waiters[0])); |
+ EXPECT_THAT(instance()->LockEndpoint(DummyEndpoint(), &waiters[0]), IsOk()); |
EXPECT_EQ(ERR_IO_PENDING, |
instance()->LockEndpoint(DummyEndpoint(), &waiters[1])); |
@@ -168,7 +172,7 @@ TEST_F(WebSocketEndpointLockManagerTest, LockEndpointReturnsOkOnce) { |
TEST_F(WebSocketEndpointLockManagerTest, GotEndpointLockNotCalledOnOk) { |
FakeWaiter waiter; |
- EXPECT_EQ(OK, instance()->LockEndpoint(DummyEndpoint(), &waiter)); |
+ EXPECT_THAT(instance()->LockEndpoint(DummyEndpoint(), &waiter), IsOk()); |
RunUntilIdle(); |
EXPECT_FALSE(waiter.called()); |
@@ -177,7 +181,7 @@ TEST_F(WebSocketEndpointLockManagerTest, GotEndpointLockNotCalledOnOk) { |
TEST_F(WebSocketEndpointLockManagerTest, GotEndpointLockNotCalledImmediately) { |
FakeWaiter waiters[2]; |
- EXPECT_EQ(OK, instance()->LockEndpoint(DummyEndpoint(), &waiters[0])); |
+ EXPECT_THAT(instance()->LockEndpoint(DummyEndpoint(), &waiters[0]), IsOk()); |
EXPECT_EQ(ERR_IO_PENDING, |
instance()->LockEndpoint(DummyEndpoint(), &waiters[1])); |
RunUntilIdle(); |
@@ -188,7 +192,7 @@ TEST_F(WebSocketEndpointLockManagerTest, GotEndpointLockNotCalledImmediately) { |
TEST_F(WebSocketEndpointLockManagerTest, GotEndpointLockCalledWhenUnlocked) { |
FakeWaiter waiters[2]; |
- EXPECT_EQ(OK, instance()->LockEndpoint(DummyEndpoint(), &waiters[0])); |
+ EXPECT_THAT(instance()->LockEndpoint(DummyEndpoint(), &waiters[0]), IsOk()); |
EXPECT_EQ(ERR_IO_PENDING, |
instance()->LockEndpoint(DummyEndpoint(), &waiters[1])); |
instance()->UnlockEndpoint(DummyEndpoint()); |
@@ -201,7 +205,8 @@ TEST_F(WebSocketEndpointLockManagerTest, GotEndpointLockCalledWhenUnlocked) { |
TEST_F(WebSocketEndpointLockManagerTest, |
EndpointUnlockedIfWaiterAlreadyDeleted) { |
FakeWaiter first_lock_holder; |
- EXPECT_EQ(OK, instance()->LockEndpoint(DummyEndpoint(), &first_lock_holder)); |
+ EXPECT_THAT(instance()->LockEndpoint(DummyEndpoint(), &first_lock_holder), |
+ IsOk()); |
{ |
FakeWaiter short_lived_waiter; |
@@ -213,7 +218,8 @@ TEST_F(WebSocketEndpointLockManagerTest, |
RunUntilIdle(); |
FakeWaiter second_lock_holder; |
- EXPECT_EQ(OK, instance()->LockEndpoint(DummyEndpoint(), &second_lock_holder)); |
+ EXPECT_THAT(instance()->LockEndpoint(DummyEndpoint(), &second_lock_holder), |
+ IsOk()); |
UnlockDummyEndpoint(1); |
} |
@@ -221,7 +227,7 @@ TEST_F(WebSocketEndpointLockManagerTest, |
TEST_F(WebSocketEndpointLockManagerTest, RememberSocketWorks) { |
FakeWaiter waiters[2]; |
FakeStreamSocket dummy_socket; |
- EXPECT_EQ(OK, instance()->LockEndpoint(DummyEndpoint(), &waiters[0])); |
+ EXPECT_THAT(instance()->LockEndpoint(DummyEndpoint(), &waiters[0]), IsOk()); |
EXPECT_EQ(ERR_IO_PENDING, |
instance()->LockEndpoint(DummyEndpoint(), &waiters[1])); |
@@ -239,7 +245,7 @@ TEST_F(WebSocketEndpointLockManagerTest, SocketAssociationForgottenOnUnlock) { |
FakeWaiter waiter; |
FakeStreamSocket dummy_socket; |
- EXPECT_EQ(OK, instance()->LockEndpoint(DummyEndpoint(), &waiter)); |
+ EXPECT_THAT(instance()->LockEndpoint(DummyEndpoint(), &waiter), IsOk()); |
instance()->RememberSocket(&dummy_socket, DummyEndpoint()); |
instance()->UnlockEndpoint(DummyEndpoint()); |
RunUntilIdle(); |
@@ -251,7 +257,7 @@ TEST_F(WebSocketEndpointLockManagerTest, SocketAssociationForgottenOnUnlock) { |
TEST_F(WebSocketEndpointLockManagerTest, NextWaiterCanCallRememberSocketAgain) { |
FakeWaiter waiters[2]; |
FakeStreamSocket dummy_sockets[2]; |
- EXPECT_EQ(OK, instance()->LockEndpoint(DummyEndpoint(), &waiters[0])); |
+ EXPECT_THAT(instance()->LockEndpoint(DummyEndpoint(), &waiters[0]), IsOk()); |
EXPECT_EQ(ERR_IO_PENDING, |
instance()->LockEndpoint(DummyEndpoint(), &waiters[1])); |
@@ -270,7 +276,7 @@ TEST_F(WebSocketEndpointLockManagerTest, |
FakeWaiter waiters[3]; |
FakeStreamSocket dummy_socket; |
- EXPECT_EQ(OK, instance()->LockEndpoint(DummyEndpoint(), &waiters[0])); |
+ EXPECT_THAT(instance()->LockEndpoint(DummyEndpoint(), &waiters[0]), IsOk()); |
EXPECT_EQ(ERR_IO_PENDING, |
instance()->LockEndpoint(DummyEndpoint(), &waiters[1])); |
EXPECT_EQ(ERR_IO_PENDING, |
@@ -288,7 +294,7 @@ TEST_F(WebSocketEndpointLockManagerTest, |
// UnlockEndpoint() should always be asynchronous. |
TEST_F(WebSocketEndpointLockManagerTest, UnlockEndpointIsAsynchronous) { |
FakeWaiter waiters[2]; |
- EXPECT_EQ(OK, instance()->LockEndpoint(DummyEndpoint(), &waiters[0])); |
+ EXPECT_THAT(instance()->LockEndpoint(DummyEndpoint(), &waiters[0]), IsOk()); |
EXPECT_EQ(ERR_IO_PENDING, |
instance()->LockEndpoint(DummyEndpoint(), &waiters[1])); |
@@ -315,7 +321,7 @@ TEST_F(WebSocketEndpointLockManagerTest, UnlockEndpointIsDelayed) { |
instance()->SetUnlockDelayForTesting(unlock_delay); |
FakeWaiter fake_waiter; |
BlockingWaiter blocking_waiter; |
- EXPECT_EQ(OK, instance()->LockEndpoint(DummyEndpoint(), &fake_waiter)); |
+ EXPECT_THAT(instance()->LockEndpoint(DummyEndpoint(), &fake_waiter), IsOk()); |
EXPECT_EQ(ERR_IO_PENDING, |
instance()->LockEndpoint(DummyEndpoint(), &blocking_waiter)); |