| 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 |