Index: net/socket/websocket_transport_client_socket_pool_unittest.cc |
diff --git a/net/socket/websocket_transport_client_socket_pool_unittest.cc b/net/socket/websocket_transport_client_socket_pool_unittest.cc |
index 48026dd4320470b922f0e4a5454ad5127fca42c0..4afd76407be227fe07235562b8e4221a55215957 100644 |
--- a/net/socket/websocket_transport_client_socket_pool_unittest.cc |
+++ b/net/socket/websocket_transport_client_socket_pool_unittest.cc |
@@ -544,7 +544,7 @@ TEST_F(WebSocketTransportClientSocketPoolTest, |
EXPECT_TRUE(handle.socket()); |
IPEndPoint endpoint; |
handle.socket()->GetLocalAddress(&endpoint); |
- EXPECT_EQ(kIPv4AddressSize, endpoint.address().size()); |
+ EXPECT_TRUE(endpoint.address().IsIPv4()); |
EXPECT_EQ(2, client_socket_factory_.allocation_count()); |
} |
@@ -587,7 +587,7 @@ TEST_F(WebSocketTransportClientSocketPoolTest, |
EXPECT_TRUE(handle.socket()); |
IPEndPoint endpoint; |
handle.socket()->GetLocalAddress(&endpoint); |
- EXPECT_EQ(kIPv6AddressSize, endpoint.address().size()); |
+ EXPECT_TRUE(endpoint.address().IsIPv6()); |
EXPECT_EQ(2, client_socket_factory_.allocation_count()); |
} |
@@ -620,7 +620,7 @@ TEST_F(WebSocketTransportClientSocketPoolTest, |
EXPECT_TRUE(handle.socket()); |
IPEndPoint endpoint; |
handle.socket()->GetLocalAddress(&endpoint); |
- EXPECT_EQ(kIPv6AddressSize, endpoint.address().size()); |
+ EXPECT_TRUE(endpoint.address().IsIPv6()); |
EXPECT_EQ(1, client_socket_factory_.allocation_count()); |
} |
@@ -651,7 +651,7 @@ TEST_F(WebSocketTransportClientSocketPoolTest, IPv4HasNoFallback) { |
EXPECT_TRUE(handle.socket()); |
IPEndPoint endpoint; |
handle.socket()->GetLocalAddress(&endpoint); |
- EXPECT_EQ(kIPv4AddressSize, endpoint.address().size()); |
+ EXPECT_TRUE(endpoint.address().IsIPv4()); |
EXPECT_EQ(1, client_socket_factory_.allocation_count()); |
} |