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

Side by Side Diff: chrome/browser/media/router/discovery/mdns/dns_sd_registry_unittest.cc

Issue 2876703002: [mDns] Move dns_sd_registry from extension/api/mdns to media/router/discovery/mdns (Closed)
Patch Set: rebase with master Created 3 years, 7 months 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 unified diff | Download patch
« no previous file with comments | « chrome/browser/media/router/discovery/mdns/dns_sd_registry.cc ('k') | chrome/test/BUILD.gn » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 // Copyright 2013 The Chromium Authors. All rights reserved. 1 // Copyright 2013 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be 2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file. 3 // found in the LICENSE file.
4 4
5 #include "chrome/browser/extensions/api/mdns/dns_sd_registry.h" 5 #include "chrome/browser/media/router/discovery/mdns/dns_sd_registry.h"
6 #include "chrome/browser/extensions/api/mdns/dns_sd_delegate.h" 6 #include "chrome/browser/media/router/discovery/mdns/dns_sd_delegate.h"
7 #include "chrome/browser/extensions/api/mdns/dns_sd_device_lister.h" 7 #include "chrome/browser/media/router/discovery/mdns/dns_sd_device_lister.h"
8 #include "testing/gmock/include/gmock/gmock.h" 8 #include "testing/gmock/include/gmock/gmock.h"
9 #include "testing/gtest/include/gtest/gtest.h" 9 #include "testing/gtest/include/gtest/gtest.h"
10 10
11 namespace extensions { 11 namespace media_router {
12 12
13 class MockDnsSdDeviceLister : public DnsSdDeviceLister { 13 class MockDnsSdDeviceLister : public DnsSdDeviceLister {
14 public: 14 public:
15 MockDnsSdDeviceLister() : DnsSdDeviceLister(NULL, NULL, "") {} 15 MockDnsSdDeviceLister() : DnsSdDeviceLister(NULL, NULL, "") {}
16 ~MockDnsSdDeviceLister() override {} 16 ~MockDnsSdDeviceLister() override {}
17 17
18 MOCK_METHOD1(Discover, void(bool force_update)); 18 MOCK_METHOD1(Discover, void(bool force_update));
19 }; 19 };
20 20
21 class TestDnsSdRegistry : public DnsSdRegistry { 21 class TestDnsSdRegistry : public DnsSdRegistry {
(...skipping 11 matching lines...) Expand all
33 33
34 return service_data_map_[service_type]->GetListenerCount(); 34 return service_data_map_[service_type]->GetListenerCount();
35 } 35 }
36 36
37 MockDnsSdDeviceLister* GetLister(const std::string& service_type) { 37 MockDnsSdDeviceLister* GetLister(const std::string& service_type) {
38 return listers_.find(service_type) != listers_.end() 38 return listers_.find(service_type) != listers_.end()
39 ? listers_[service_type] 39 ? listers_[service_type]
40 : nullptr; 40 : nullptr;
41 } 41 }
42 42
43 DnsSdDelegate* GetDelegate() { 43 DnsSdDelegate* GetDelegate() { return delegate_; }
44 return delegate_;
45 }
46 44
47 protected: 45 protected:
48 DnsSdDeviceLister* CreateDnsSdDeviceLister( 46 DnsSdDeviceLister* CreateDnsSdDeviceLister(
49 DnsSdDelegate* delegate, 47 DnsSdDelegate* delegate,
50 const std::string& service_type, 48 const std::string& service_type,
51 local_discovery::ServiceDiscoverySharedClient* discovery_client) 49 local_discovery::ServiceDiscoverySharedClient* discovery_client)
52 override { 50 override {
53 delegate_ = delegate; 51 delegate_ = delegate;
54 MockDnsSdDeviceLister* lister = new MockDnsSdDeviceLister(); 52 MockDnsSdDeviceLister* lister = new MockDnsSdDeviceLister();
55 listers_[service_type] = lister; 53 listers_[service_type] = lister;
56 return lister; 54 return lister;
57 }; 55 };
58 56
59 private: 57 private:
60 std::map<std::string, MockDnsSdDeviceLister*> listers_; 58 std::map<std::string, MockDnsSdDeviceLister*> listers_;
61 // The last delegate used or NULL. 59 // The last delegate used or NULL.
62 DnsSdDelegate* delegate_; 60 DnsSdDelegate* delegate_;
63 }; 61 };
64 62
65 class MockDnsSdObserver : public DnsSdRegistry::DnsSdObserver { 63 class MockDnsSdObserver : public DnsSdRegistry::DnsSdObserver {
66 public: 64 public:
67 MOCK_METHOD2(OnDnsSdEvent, void(const std::string&, 65 MOCK_METHOD2(OnDnsSdEvent,
68 const DnsSdRegistry::DnsSdServiceList&)); 66 void(const std::string&,
67 const DnsSdRegistry::DnsSdServiceList&));
69 }; 68 };
70 69
71 class DnsSdRegistryTest : public testing::Test { 70 class DnsSdRegistryTest : public testing::Test {
72 public: 71 public:
73 DnsSdRegistryTest() {} 72 DnsSdRegistryTest() {}
74 ~DnsSdRegistryTest() override {} 73 ~DnsSdRegistryTest() override {}
75 74
76 void SetUp() override { 75 void SetUp() override {
77 registry_.reset(new TestDnsSdRegistry()); 76 registry_.reset(new TestDnsSdRegistry());
78 registry_->AddObserver(&observer_); 77 registry_->AddObserver(&observer_);
79 } 78 }
80 79
81 protected: 80 protected:
82 std::unique_ptr<TestDnsSdRegistry> registry_; 81 std::unique_ptr<TestDnsSdRegistry> registry_;
83 MockDnsSdObserver observer_; 82 MockDnsSdObserver observer_;
84 }; 83 };
85 84
86 // Tests registering 2 listeners and removing one. The device lister should 85 // Tests registering 2 listeners and removing one. The device lister should
87 // not be destroyed. 86 // not be destroyed.
88 TEST_F(DnsSdRegistryTest, RegisterUnregisterListeners) { 87 TEST_F(DnsSdRegistryTest, RegisterUnregisterListeners) {
89 const std::string service_type = "_testing._tcp.local"; 88 const std::string service_type = "_testing._tcp.local";
90 89
91 EXPECT_CALL(observer_, OnDnsSdEvent(service_type, 90 EXPECT_CALL(observer_,
92 DnsSdRegistry::DnsSdServiceList())).Times(2); 91 OnDnsSdEvent(service_type, DnsSdRegistry::DnsSdServiceList()))
92 .Times(2);
93 93
94 registry_->RegisterDnsSdListener(service_type); 94 registry_->RegisterDnsSdListener(service_type);
95 registry_->RegisterDnsSdListener(service_type); 95 registry_->RegisterDnsSdListener(service_type);
96 registry_->UnregisterDnsSdListener(service_type); 96 registry_->UnregisterDnsSdListener(service_type);
97 EXPECT_EQ(1, registry_->GetServiceListenerCount(service_type)); 97 EXPECT_EQ(1, registry_->GetServiceListenerCount(service_type));
98 98
99 registry_->UnregisterDnsSdListener(service_type); 99 registry_->UnregisterDnsSdListener(service_type);
100 EXPECT_EQ(0, registry_->GetServiceListenerCount(service_type)); 100 EXPECT_EQ(0, registry_->GetServiceListenerCount(service_type));
101 } 101 }
102 102
(...skipping 151 matching lines...) Expand 10 before | Expand all | Expand 10 after
254 registry_->RegisterDnsSdListener(service_type); 254 registry_->RegisterDnsSdListener(service_type);
255 service.ip_address = "192.168.0.100"; 255 service.ip_address = "192.168.0.100";
256 registry_->GetDelegate()->ServiceChanged(service_type, true, service); 256 registry_->GetDelegate()->ServiceChanged(service_type, true, service);
257 // Update with changed ip address. 257 // Update with changed ip address.
258 service.ip_address = "192.168.0.101"; 258 service.ip_address = "192.168.0.101";
259 registry_->GetDelegate()->ServiceChanged(service_type, false, service); 259 registry_->GetDelegate()->ServiceChanged(service_type, false, service);
260 // Update with no changes to the service. 260 // Update with no changes to the service.
261 registry_->GetDelegate()->ServiceChanged(service_type, false, service); 261 registry_->GetDelegate()->ServiceChanged(service_type, false, service);
262 } 262 }
263 263
264 } // namespace extensions 264 } // namespace media_router
OLDNEW
« no previous file with comments | « chrome/browser/media/router/discovery/mdns/dns_sd_registry.cc ('k') | chrome/test/BUILD.gn » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698