Index: net/udp/udp_socket_unittest.cc |
diff --git a/net/udp/udp_socket_unittest.cc b/net/udp/udp_socket_unittest.cc |
index c7ab74dd4ae92e004baf6879bec74709a96c55ba..cb4a495c70386703a4ec6b6e69dd7ab59dad0a31 100644 |
--- a/net/udp/udp_socket_unittest.cc |
+++ b/net/udp/udp_socket_unittest.cc |
@@ -23,7 +23,9 @@ |
#include "net/log/test_net_log.h" |
#include "net/log/test_net_log_entry.h" |
#include "net/log/test_net_log_util.h" |
+#include "net/test/gtest_util.h" |
#include "net/test/net_test_suite.h" |
+#include "testing/gmock/include/gmock/gmock.h" |
#include "testing/gtest/include/gtest/gtest.h" |
#include "testing/platform_test.h" |
@@ -31,6 +33,9 @@ |
#include <TargetConditionals.h> |
#endif |
+using net::test::IsError; |
+using net::test::IsOk; |
+ |
namespace net { |
namespace { |
@@ -167,7 +172,7 @@ void UDPSocketTest::ConnectTest(bool use_nonblocking_io) { |
#endif |
server->AllowAddressReuse(); |
int rv = server->Listen(bind_address); |
- ASSERT_EQ(OK, rv); |
+ ASSERT_THAT(rv, IsOk()); |
// Setup the client. |
IPEndPoint server_address; |
@@ -182,7 +187,7 @@ void UDPSocketTest::ConnectTest(bool use_nonblocking_io) { |
#endif |
rv = client->Connect(server_address); |
- EXPECT_EQ(OK, rv); |
+ EXPECT_THAT(rv, IsOk()); |
// Client sends to the server. |
rv = WriteSocket(client.get(), simple_message); |
@@ -206,7 +211,7 @@ void UDPSocketTest::ConnectTest(bool use_nonblocking_io) { |
rv = server->RecvFrom( |
buffer_.get(), kMaxRead, &recv_from_address_, |
base::Bind(&ReadCompleteCallback, &read_result, run_loop.QuitClosure())); |
- EXPECT_EQ(ERR_IO_PENDING, rv); |
+ EXPECT_THAT(rv, IsError(ERR_IO_PENDING)); |
// Client sends to the server. |
base::ThreadTaskRunnerHandle::Get()->PostTask( |
@@ -297,9 +302,9 @@ TEST_F(UDPSocketTest, Broadcast) { |
server2->AllowBroadcast(); |
int rv = server1->Listen(listen_address); |
- EXPECT_EQ(OK, rv); |
+ EXPECT_THAT(rv, IsOk()); |
rv = server2->Listen(listen_address); |
- EXPECT_EQ(OK, rv); |
+ EXPECT_THAT(rv, IsOk()); |
rv = SendToSocket(server1.get(), first_message, broadcast_address); |
ASSERT_EQ(static_cast<int>(first_message.size()), rv); |
@@ -364,10 +369,10 @@ TEST_F(UDPSocketTest, ConnectRandomBind) { |
NULL, |
NetLog::Source()); |
sockets.push_back(socket); |
- EXPECT_EQ(OK, socket->Connect(peer_address)); |
+ EXPECT_THAT(socket->Connect(peer_address), IsOk()); |
IPEndPoint client_address; |
- EXPECT_EQ(OK, socket->GetLocalAddress(&client_address)); |
+ EXPECT_THAT(socket->GetLocalAddress(&client_address), IsOk()); |
used_ports.push_back(client_address.port()); |
} |
@@ -382,11 +387,11 @@ TEST_F(UDPSocketTest, ConnectRandomBind) { |
// Create a socket with random binding policy and connect. |
std::unique_ptr<UDPClientSocket> test_socket(new UDPClientSocket( |
DatagramSocket::RANDOM_BIND, rand_int_cb, NULL, NetLog::Source())); |
- EXPECT_EQ(OK, test_socket->Connect(peer_address)); |
+ EXPECT_THAT(test_socket->Connect(peer_address), IsOk()); |
// Make sure that the last port number in the |used_ports| was used. |
IPEndPoint client_address; |
- EXPECT_EQ(OK, test_socket->GetLocalAddress(&client_address)); |
+ EXPECT_THAT(test_socket->GetLocalAddress(&client_address), IsOk()); |
EXPECT_EQ(used_ports.back(), client_address.port()); |
STLDeleteElements(&sockets); |
@@ -413,7 +418,7 @@ TEST_F(UDPSocketTest, MAYBE_ConnectFail) { |
base::Bind(&PrivilegedRand), |
NULL, NetLog::Source())); |
int rv = socket->Open(peer_address.GetFamily()); |
- EXPECT_EQ(OK, rv); |
+ EXPECT_THAT(rv, IsOk()); |
rv = socket->Connect(peer_address); |
// Connect should have failed since we couldn't bind to that port, |
EXPECT_NE(OK, rv); |
@@ -441,14 +446,14 @@ TEST_F(UDPSocketTest, VerifyConnectBindsAddr) { |
UDPServerSocket server1(NULL, NetLog::Source()); |
server1.AllowAddressReuse(); |
int rv = server1.Listen(bind_address); |
- ASSERT_EQ(OK, rv); |
+ ASSERT_THAT(rv, IsOk()); |
// Setup the second server to listen. |
CreateUDPAddress("127.0.0.1", kPort2, &bind_address); |
UDPServerSocket server2(NULL, NetLog::Source()); |
server2.AllowAddressReuse(); |
rv = server2.Listen(bind_address); |
- ASSERT_EQ(OK, rv); |
+ ASSERT_THAT(rv, IsOk()); |
// Setup the client, connected to server 1. |
IPEndPoint server_address; |
@@ -458,7 +463,7 @@ TEST_F(UDPSocketTest, VerifyConnectBindsAddr) { |
NULL, |
NetLog::Source()); |
rv = client.Connect(server_address); |
- EXPECT_EQ(OK, rv); |
+ EXPECT_THAT(rv, IsOk()); |
// Client sends to server1. |
rv = WriteSocket(&client, simple_message); |
@@ -471,7 +476,7 @@ TEST_F(UDPSocketTest, VerifyConnectBindsAddr) { |
// Get the client's address. |
IPEndPoint client_address; |
rv = client.GetLocalAddress(&client_address); |
- EXPECT_EQ(OK, rv); |
+ EXPECT_THAT(rv, IsOk()); |
// Server2 sends reply. |
rv = SendToSocket(&server2, foreign_message, |
@@ -528,7 +533,7 @@ TEST_F(UDPSocketTest, ClientGetLocalPeerAddresses) { |
IPEndPoint fetched_local_address; |
rv = client.GetLocalAddress(&fetched_local_address); |
- EXPECT_EQ(OK, rv); |
+ EXPECT_THAT(rv, IsOk()); |
// TODO(mbelshe): figure out how to verify the IP and port. |
// The port is dynamically generated by the udp stack. |
@@ -538,7 +543,7 @@ TEST_F(UDPSocketTest, ClientGetLocalPeerAddresses) { |
IPEndPoint fetched_remote_address; |
rv = client.GetPeerAddress(&fetched_remote_address); |
- EXPECT_EQ(OK, rv); |
+ EXPECT_THAT(rv, IsOk()); |
EXPECT_EQ(remote_address, fetched_remote_address); |
} |
@@ -549,7 +554,7 @@ TEST_F(UDPSocketTest, ServerGetLocalAddress) { |
CreateUDPAddress("127.0.0.1", 0, &bind_address); |
UDPServerSocket server(NULL, NetLog::Source()); |
int rv = server.Listen(bind_address); |
- EXPECT_EQ(OK, rv); |
+ EXPECT_THAT(rv, IsOk()); |
IPEndPoint local_address; |
rv = server.GetLocalAddress(&local_address); |
@@ -565,7 +570,7 @@ TEST_F(UDPSocketTest, ServerGetPeerAddress) { |
CreateUDPAddress("127.0.0.1", 0, &bind_address); |
UDPServerSocket server(NULL, NetLog::Source()); |
int rv = server.Listen(bind_address); |
- EXPECT_EQ(OK, rv); |
+ EXPECT_THAT(rv, IsOk()); |
IPEndPoint peer_address; |
rv = server.GetPeerAddress(&peer_address); |
@@ -578,7 +583,7 @@ TEST_F(UDPSocketTest, CloseWithPendingRead) { |
CreateUDPAddress("127.0.0.1", 0, &bind_address); |
UDPServerSocket server(NULL, NetLog::Source()); |
int rv = server.Listen(bind_address); |
- EXPECT_EQ(OK, rv); |
+ EXPECT_THAT(rv, IsOk()); |
TestCompletionCallback callback; |
IPEndPoint from; |
@@ -612,12 +617,12 @@ TEST_F(UDPSocketTest, MAYBE_JoinMulticastGroup) { |
RandIntCallback(), |
NULL, |
NetLog::Source()); |
- EXPECT_EQ(OK, socket.Open(bind_address.GetFamily())); |
- EXPECT_EQ(OK, socket.Bind(bind_address)); |
- EXPECT_EQ(OK, socket.JoinGroup(group_ip)); |
+ EXPECT_THAT(socket.Open(bind_address.GetFamily()), IsOk()); |
+ EXPECT_THAT(socket.Bind(bind_address), IsOk()); |
+ EXPECT_THAT(socket.JoinGroup(group_ip), IsOk()); |
// Joining group multiple times. |
EXPECT_NE(OK, socket.JoinGroup(group_ip)); |
- EXPECT_EQ(OK, socket.LeaveGroup(group_ip)); |
+ EXPECT_THAT(socket.LeaveGroup(group_ip), IsOk()); |
// Leaving group multiple times. |
EXPECT_NE(OK, socket.LeaveGroup(group_ip)); |
@@ -634,15 +639,15 @@ TEST_F(UDPSocketTest, MulticastOptions) { |
NULL, |
NetLog::Source()); |
// Before binding. |
- EXPECT_EQ(OK, socket.SetMulticastLoopbackMode(false)); |
- EXPECT_EQ(OK, socket.SetMulticastLoopbackMode(true)); |
- EXPECT_EQ(OK, socket.SetMulticastTimeToLive(0)); |
- EXPECT_EQ(OK, socket.SetMulticastTimeToLive(3)); |
+ EXPECT_THAT(socket.SetMulticastLoopbackMode(false), IsOk()); |
+ EXPECT_THAT(socket.SetMulticastLoopbackMode(true), IsOk()); |
+ EXPECT_THAT(socket.SetMulticastTimeToLive(0), IsOk()); |
+ EXPECT_THAT(socket.SetMulticastTimeToLive(3), IsOk()); |
EXPECT_NE(OK, socket.SetMulticastTimeToLive(-1)); |
- EXPECT_EQ(OK, socket.SetMulticastInterface(0)); |
+ EXPECT_THAT(socket.SetMulticastInterface(0), IsOk()); |
- EXPECT_EQ(OK, socket.Open(bind_address.GetFamily())); |
- EXPECT_EQ(OK, socket.Bind(bind_address)); |
+ EXPECT_THAT(socket.Open(bind_address.GetFamily()), IsOk()); |
+ EXPECT_THAT(socket.Bind(bind_address), IsOk()); |
EXPECT_NE(OK, socket.SetMulticastLoopbackMode(false)); |
EXPECT_NE(OK, socket.SetMulticastTimeToLive(0)); |
@@ -663,7 +668,7 @@ TEST_F(UDPSocketTest, SetDSCP) { |
// We need a real IP, but we won't actually send anything to it. |
CreateUDPAddress("8.8.8.8", 9999, &bind_address); |
int rv = client.Open(bind_address.GetFamily()); |
- EXPECT_EQ(OK, rv); |
+ EXPECT_THAT(rv, IsOk()); |
rv = client.Connect(bind_address); |
if (rv != OK) { |
@@ -671,7 +676,7 @@ TEST_F(UDPSocketTest, SetDSCP) { |
CreateUDPAddress("127.0.0.1", 9999, &bind_address); |
rv = client.Connect(bind_address); |
} |
- EXPECT_EQ(OK, rv); |
+ EXPECT_THAT(rv, IsOk()); |
client.SetDiffServCodePoint(DSCP_NO_CHANGE); |
client.SetDiffServCodePoint(DSCP_AF41); |
@@ -771,13 +776,13 @@ TEST_F(UDPSocketTest, SetDSCPFake) { |
NULL, |
NetLog::Source()); |
int rv = client.SetDiffServCodePoint(DSCP_AF41); |
- EXPECT_EQ(ERR_SOCKET_NOT_CONNECTED, rv); |
+ EXPECT_THAT(rv, IsError(ERR_SOCKET_NOT_CONNECTED)); |
rv = client.Open(bind_address.GetFamily()); |
- EXPECT_EQ(OK, rv); |
+ EXPECT_THAT(rv, IsOk()); |
rv = client.Connect(bind_address); |
- EXPECT_EQ(OK, rv); |
+ EXPECT_THAT(rv, IsOk()); |
QwaveAPI& qos(QwaveAPI::Get()); |
qos.create_handle_func_ = FakeQOSCreateHandleFAIL; |
@@ -787,24 +792,24 @@ TEST_F(UDPSocketTest, SetDSCPFake) { |
qos.set_flow_func_ = FakeQOSSetFlow; |
qos.qwave_supported_ = true; |
- EXPECT_EQ(OK, client.SetDiffServCodePoint(DSCP_NO_CHANGE)); |
+ EXPECT_THAT(client.SetDiffServCodePoint(DSCP_NO_CHANGE), IsOk()); |
EXPECT_EQ(ERROR_NOT_SUPPORTED, client.SetDiffServCodePoint(DSCP_AF41)); |
qos.create_handle_func_ = FakeQOSCreateHandle; |
g_expected_dscp = DSCP_AF41; |
g_expected_traffic_type = QOSTrafficTypeAudioVideo; |
- EXPECT_EQ(OK, client.SetDiffServCodePoint(DSCP_AF41)); |
+ EXPECT_THAT(client.SetDiffServCodePoint(DSCP_AF41), IsOk()); |
g_expected_dscp = DSCP_DEFAULT; |
g_expected_traffic_type = QOSTrafficTypeBestEffort; |
- EXPECT_EQ(OK, client.SetDiffServCodePoint(DSCP_DEFAULT)); |
+ EXPECT_THAT(client.SetDiffServCodePoint(DSCP_DEFAULT), IsOk()); |
g_expected_dscp = DSCP_CS2; |
g_expected_traffic_type = QOSTrafficTypeExcellentEffort; |
- EXPECT_EQ(OK, client.SetDiffServCodePoint(DSCP_CS2)); |
+ EXPECT_THAT(client.SetDiffServCodePoint(DSCP_CS2), IsOk()); |
g_expected_dscp = DSCP_CS3; |
g_expected_traffic_type = QOSTrafficTypeExcellentEffort; |
- EXPECT_EQ(OK, client.SetDiffServCodePoint(DSCP_NO_CHANGE)); |
+ EXPECT_THAT(client.SetDiffServCodePoint(DSCP_NO_CHANGE), IsOk()); |
g_expected_dscp = DSCP_DEFAULT; |
g_expected_traffic_type = QOSTrafficTypeBestEffort; |
- EXPECT_EQ(OK, client.SetDiffServCodePoint(DSCP_DEFAULT)); |
+ EXPECT_THAT(client.SetDiffServCodePoint(DSCP_DEFAULT), IsOk()); |
client.Close(); |
} |
#endif |