Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(63)

Unified Diff: chrome/utility/local_discovery/service_discovery_client_unittest.cc

Issue 87693002: Refactored to make MDnsSocketFactory return array of sockets. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Created 7 years, 1 month ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
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.

Powered by Google App Engine
This is Rietveld 408576698