OLD | NEW |
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 Loading... |
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 Loading... |
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 |
OLD | NEW |