Index: chrome/utility/local_discovery/service_discovery_client_unittest.cc |
diff --git a/chrome/utility/local_discovery/service_discovery_client_unittest.cc b/chrome/utility/local_discovery/service_discovery_client_unittest.cc |
index 43914ee894bd57c01b19dc1b8a31c8ccc8e0eaca..9dcbcc348acd49f7455327ffb5be4ff1f174f8c5 100644 |
--- a/chrome/utility/local_discovery/service_discovery_client_unittest.cc |
+++ b/chrome/utility/local_discovery/service_discovery_client_unittest.cc |
@@ -183,12 +183,8 @@ class MockServiceWatcherClient { |
class ServiceDiscoveryTest : public ::testing::Test { |
public: |
ServiceDiscoveryTest() |
- : socket_factory_(new net::MockMDnsSocketFactory), |
- mdns_client_( |
- scoped_ptr<net::MDnsConnection::SocketFactory>( |
- socket_factory_)), |
- service_discovery_client_(&mdns_client_) { |
- mdns_client_.StartListening(); |
+ : service_discovery_client_(&mdns_client_) { |
+ mdns_client_.StartListening(&socket_factory_); |
} |
virtual ~ServiceDiscoveryTest() { |
@@ -209,7 +205,7 @@ class ServiceDiscoveryTest : public ::testing::Test { |
base::MessageLoop::current()->Quit(); |
} |
- net::MockMDnsSocketFactory* socket_factory_; |
+ net::MockMDnsSocketFactory socket_factory_; |
net::MDnsClientImpl mdns_client_; |
ServiceDiscoveryClientImpl service_discovery_client_; |
base::MessageLoop loop_; |
@@ -228,8 +224,7 @@ TEST_F(ServiceDiscoveryTest, AddRemoveService) { |
"hello._privet._tcp.local")) |
.Times(Exactly(1)); |
- socket_factory_->SimulateReceive( |
- kSamplePacketPTR, sizeof(kSamplePacketPTR)); |
+ socket_factory_.SimulateReceive(kSamplePacketPTR, sizeof(kSamplePacketPTR)); |
EXPECT_CALL(delegate, OnServiceUpdated(ServiceWatcher::UPDATE_REMOVED, |
"hello._privet._tcp.local")) |
@@ -247,15 +242,13 @@ TEST_F(ServiceDiscoveryTest, DiscoverNewServices) { |
watcher->Start(); |
- EXPECT_CALL(*socket_factory_, OnSendTo(_)) |
- .Times(2); |
+ EXPECT_CALL(socket_factory_, OnSendTo(_)).Times(2); |
watcher->DiscoverNewServices(false); |
}; |
TEST_F(ServiceDiscoveryTest, ReadCachedServices) { |
- socket_factory_->SimulateReceive( |
- kSamplePacketPTR, sizeof(kSamplePacketPTR)); |
+ socket_factory_.SimulateReceive(kSamplePacketPTR, sizeof(kSamplePacketPTR)); |
StrictMock<MockServiceWatcherClient> delegate; |
@@ -274,8 +267,7 @@ TEST_F(ServiceDiscoveryTest, ReadCachedServices) { |
TEST_F(ServiceDiscoveryTest, ReadCachedServicesMultiple) { |
- socket_factory_->SimulateReceive( |
- kSamplePacketPTR2, sizeof(kSamplePacketPTR2)); |
+ socket_factory_.SimulateReceive(kSamplePacketPTR2, sizeof(kSamplePacketPTR2)); |
StrictMock<MockServiceWatcherClient> delegate; |
scoped_ptr<ServiceWatcher> watcher = |
@@ -308,8 +300,7 @@ TEST_F(ServiceDiscoveryTest, OnServiceChanged) { |
"hello._privet._tcp.local")) |
.Times(Exactly(1)); |
- socket_factory_->SimulateReceive( |
- kSamplePacketPTR, sizeof(kSamplePacketPTR)); |
+ socket_factory_.SimulateReceive(kSamplePacketPTR, sizeof(kSamplePacketPTR)); |
base::MessageLoop::current()->RunUntilIdle(); |
@@ -317,11 +308,9 @@ TEST_F(ServiceDiscoveryTest, OnServiceChanged) { |
"hello._privet._tcp.local")) |
.Times(Exactly(1)); |
- socket_factory_->SimulateReceive( |
- kSamplePacketSRV, sizeof(kSamplePacketSRV)); |
+ socket_factory_.SimulateReceive(kSamplePacketSRV, sizeof(kSamplePacketSRV)); |
- socket_factory_->SimulateReceive( |
- kSamplePacketTXT, sizeof(kSamplePacketTXT)); |
+ socket_factory_.SimulateReceive(kSamplePacketTXT, sizeof(kSamplePacketTXT)); |
base::MessageLoop::current()->RunUntilIdle(); |
}; |
@@ -338,8 +327,7 @@ TEST_F(ServiceDiscoveryTest, SinglePacket) { |
"hello._privet._tcp.local")) |
.Times(Exactly(1)); |
- socket_factory_->SimulateReceive( |
- kSamplePacketPTR, sizeof(kSamplePacketPTR)); |
+ socket_factory_.SimulateReceive(kSamplePacketPTR, sizeof(kSamplePacketPTR)); |
// Reset the "already updated" flag. |
base::MessageLoop::current()->RunUntilIdle(); |
@@ -348,11 +336,9 @@ TEST_F(ServiceDiscoveryTest, SinglePacket) { |
"hello._privet._tcp.local")) |
.Times(Exactly(1)); |
- socket_factory_->SimulateReceive( |
- kSamplePacketSRV, sizeof(kSamplePacketSRV)); |
+ socket_factory_.SimulateReceive(kSamplePacketSRV, sizeof(kSamplePacketSRV)); |
- socket_factory_->SimulateReceive( |
- kSamplePacketTXT, sizeof(kSamplePacketTXT)); |
+ socket_factory_.SimulateReceive(kSamplePacketTXT, sizeof(kSamplePacketTXT)); |
base::MessageLoop::current()->RunUntilIdle(); |
}; |
@@ -401,13 +387,11 @@ class ServiceResolverTest : public ServiceDiscoveryTest { |
}; |
TEST_F(ServiceResolverTest, TxtAndSrvButNoA) { |
- EXPECT_CALL(*socket_factory_, OnSendTo(_)) |
- .Times(4); |
+ EXPECT_CALL(socket_factory_, OnSendTo(_)).Times(4); |
resolver_->StartResolving(); |
- socket_factory_->SimulateReceive( |
- kSamplePacketSRV, sizeof(kSamplePacketSRV)); |
+ socket_factory_.SimulateReceive(kSamplePacketSRV, sizeof(kSamplePacketSRV)); |
base::MessageLoop::current()->RunUntilIdle(); |
@@ -417,13 +401,11 @@ TEST_F(ServiceResolverTest, TxtAndSrvButNoA) { |
metadata_expected_, |
net::IPAddressNumber())); |
- socket_factory_->SimulateReceive( |
- kSamplePacketTXT, sizeof(kSamplePacketTXT)); |
+ socket_factory_.SimulateReceive(kSamplePacketTXT, sizeof(kSamplePacketTXT)); |
}; |
TEST_F(ServiceResolverTest, TxtSrvAndA) { |
- EXPECT_CALL(*socket_factory_, OnSendTo(_)) |
- .Times(4); |
+ EXPECT_CALL(socket_factory_, OnSendTo(_)).Times(4); |
resolver_->StartResolving(); |
@@ -433,16 +415,13 @@ TEST_F(ServiceResolverTest, TxtSrvAndA) { |
metadata_expected_, |
ip_address_expected_)); |
- socket_factory_->SimulateReceive( |
- kSamplePacketTXT, sizeof(kSamplePacketTXT)); |
+ socket_factory_.SimulateReceive(kSamplePacketTXT, sizeof(kSamplePacketTXT)); |
- socket_factory_->SimulateReceive( |
- kSamplePacketSRVA, sizeof(kSamplePacketSRVA)); |
+ socket_factory_.SimulateReceive(kSamplePacketSRVA, sizeof(kSamplePacketSRVA)); |
}; |
TEST_F(ServiceResolverTest, JustSrv) { |
- EXPECT_CALL(*socket_factory_, OnSendTo(_)) |
- .Times(4); |
+ EXPECT_CALL(socket_factory_, OnSendTo(_)).Times(4); |
resolver_->StartResolving(); |
@@ -452,8 +431,7 @@ TEST_F(ServiceResolverTest, JustSrv) { |
std::vector<std::string>(), |
ip_address_expected_)); |
- socket_factory_->SimulateReceive( |
- kSamplePacketSRVA, sizeof(kSamplePacketSRVA)); |
+ socket_factory_.SimulateReceive(kSamplePacketSRVA, sizeof(kSamplePacketSRVA)); |
// TODO(noamsml): When NSEC record support is added, change this to use an |
// NSEC record. |
@@ -461,13 +439,12 @@ TEST_F(ServiceResolverTest, JustSrv) { |
}; |
TEST_F(ServiceResolverTest, WithNothing) { |
- EXPECT_CALL(*socket_factory_, OnSendTo(_)) |
- .Times(4); |
+ EXPECT_CALL(socket_factory_, OnSendTo(_)).Times(4); |
resolver_->StartResolving(); |
EXPECT_CALL(*this, OnFinishedResolvingInternal( |
- ServiceResolver::STATUS_REQUEST_TIMEOUT, _, _, _)); |
+ ServiceResolver::STATUS_REQUEST_TIMEOUT, _, _, _)); |
// TODO(noamsml): When NSEC record support is added, change this to use an |
// NSEC record. |