| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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 "chromeos/components/tether/host_scan_scheduler.h" | 5 #include "chromeos/components/tether/host_scan_scheduler.h" |
| 6 | 6 |
| 7 #include "base/memory/ptr_util.h" | 7 #include "base/memory/ptr_util.h" |
| 8 #include "chromeos/components/tether/host_scanner.h" | 8 #include "chromeos/components/tether/host_scanner.h" |
| 9 #include "chromeos/dbus/power_manager_client.h" | 9 #include "chromeos/dbus/power_manager_client.h" |
| 10 #include "chromeos/login/login_state.h" | 10 #include "chromeos/login/login_state.h" |
| 11 #include "chromeos/network/network_handler.h" | 11 #include "chromeos/network/network_handler.h" |
| 12 #include "chromeos/network/network_state.h" | 12 #include "chromeos/network/network_state.h" |
| 13 #include "chromeos/network/network_state_handler.h" | 13 #include "chromeos/network/network_state_handler.h" |
| 14 #include "components/cryptauth/cryptauth_device_manager.h" | 14 #include "components/cryptauth/cryptauth_device_manager.h" |
| 15 #include "testing/gmock/include/gmock/gmock.h" | 15 #include "testing/gmock/include/gmock/gmock.h" |
| 16 #include "testing/gtest/include/gtest/gtest.h" | 16 #include "testing/gtest/include/gtest/gtest.h" |
| 17 | 17 |
| 18 namespace chromeos { | 18 namespace chromeos { |
| 19 | 19 |
| 20 namespace tether { | 20 namespace tether { |
| 21 | 21 |
| 22 namespace {} // namespace | |
| 23 | |
| 24 class HostScanSchedulerTest : public testing::Test { | 22 class HostScanSchedulerTest : public testing::Test { |
| 25 protected: | 23 protected: |
| 26 class TestContext : public HostScanScheduler::Context { | 24 class TestDelegate : public HostScanScheduler::Delegate { |
| 27 public: | 25 public: |
| 28 TestContext(HostScanSchedulerTest* test) | 26 TestDelegate(HostScanSchedulerTest* test) |
| 29 : test_(test), | 27 : test_(test), |
| 30 observer_(nullptr), | 28 observer_(nullptr), |
| 31 is_authenticated_user_logged_in(true), | 29 is_authenticated_user_logged_in(true), |
| 32 is_network_connected_or_connecting(false), | 30 is_network_connected_or_connecting(false), |
| 33 are_tether_hosts_synced(true) {} | 31 are_tether_hosts_synced(true) {} |
| 34 | 32 |
| 35 void AddObserver(HostScanScheduler* host_scan_scheduler) override { | 33 void AddObserver(HostScanScheduler* host_scan_scheduler) override { |
| 36 observer_ = host_scan_scheduler; | 34 observer_ = host_scan_scheduler; |
| 37 } | 35 } |
| 38 | 36 |
| (...skipping 14 matching lines...) Expand all Loading... |
| 53 | 51 |
| 54 bool AreTetherHostsSynced() const override { | 52 bool AreTetherHostsSynced() const override { |
| 55 return are_tether_hosts_synced; | 53 return are_tether_hosts_synced; |
| 56 } | 54 } |
| 57 | 55 |
| 58 bool IsObserverSet() const { | 56 bool IsObserverSet() const { |
| 59 return observer_ != nullptr && | 57 return observer_ != nullptr && |
| 60 observer_ == test_->host_scan_scheduler_.get(); | 58 observer_ == test_->host_scan_scheduler_.get(); |
| 61 } | 59 } |
| 62 | 60 |
| 63 void SetIsAuthenticatedUserLoggedIn(bool value) { | 61 void set_is_authenticated_user_logged_in(bool value) { |
| 64 is_authenticated_user_logged_in = value; | 62 is_authenticated_user_logged_in = value; |
| 65 } | 63 } |
| 66 | 64 |
| 67 void SetIsNetworkConnectedOrConnecting(bool value) { | 65 void set_is_network_connected_or_connecting(bool value) { |
| 68 is_network_connected_or_connecting = value; | 66 is_network_connected_or_connecting = value; |
| 69 } | 67 } |
| 70 | 68 |
| 71 void AreTetherHostsSynced(bool value) { are_tether_hosts_synced = value; } | 69 void set_are_tether_hosts_synced(bool value) { |
| 70 are_tether_hosts_synced = value; |
| 71 } |
| 72 | 72 |
| 73 private: | 73 private: |
| 74 const HostScanSchedulerTest* test_; | 74 const HostScanSchedulerTest* test_; |
| 75 | 75 |
| 76 HostScanScheduler* observer_; | 76 HostScanScheduler* observer_; |
| 77 bool is_authenticated_user_logged_in; | 77 bool is_authenticated_user_logged_in; |
| 78 bool is_network_connected_or_connecting; | 78 bool is_network_connected_or_connecting; |
| 79 bool are_tether_hosts_synced; | 79 bool are_tether_hosts_synced; |
| 80 }; | 80 }; |
| 81 | 81 |
| 82 class MockHostScanner : public HostScanner { | 82 class FakeHostScanner : public HostScanner { |
| 83 public: | 83 public: |
| 84 MockHostScanner() : num_scans_started_(0) {} | 84 FakeHostScanner() : num_scans_started_(0) {} |
| 85 ~MockHostScanner() override {} | 85 ~FakeHostScanner() override {} |
| 86 | 86 |
| 87 void StartScan() override { num_scans_started_++; } | 87 void StartScan() override { num_scans_started_++; } |
| 88 | 88 |
| 89 int num_scans_started() { return num_scans_started_; } | 89 int num_scans_started() { return num_scans_started_; } |
| 90 | 90 |
| 91 private: | 91 private: |
| 92 int num_scans_started_; | 92 int num_scans_started_; |
| 93 }; | 93 }; |
| 94 | 94 |
| 95 HostScanSchedulerTest() {} | 95 HostScanSchedulerTest() {} |
| 96 | 96 |
| 97 void SetUp() override { | 97 void SetUp() override { |
| 98 test_context_ = new TestContext(this); | 98 test_delegate_ = new TestDelegate(this); |
| 99 mock_host_scanner_ = new MockHostScanner(); | 99 fake_host_scanner_ = new FakeHostScanner(); |
| 100 | 100 |
| 101 host_scan_scheduler_.reset(new HostScanScheduler( | 101 host_scan_scheduler_.reset( |
| 102 base::WrapUnique(test_context_), base::WrapUnique(mock_host_scanner_))); | 102 new HostScanScheduler(base::WrapUnique(test_delegate_), |
| 103 base::WrapUnique(fake_host_scanner_))); |
| 103 | 104 |
| 104 EXPECT_FALSE(test_context_->IsObserverSet()); | 105 EXPECT_FALSE(test_delegate_->IsObserverSet()); |
| 105 host_scan_scheduler_->InitializeAutomaticScans(); | 106 host_scan_scheduler_->InitializeAutomaticScans(); |
| 106 EXPECT_TRUE(test_context_->IsObserverSet()); | 107 EXPECT_TRUE(test_delegate_->IsObserverSet()); |
| 107 } | 108 } |
| 108 | 109 |
| 109 void TearDown() override { | 110 void TearDown() override { |
| 110 EXPECT_TRUE(test_context_->IsObserverSet()); | 111 EXPECT_TRUE(test_delegate_->IsObserverSet()); |
| 111 host_scan_scheduler_.reset(); | 112 host_scan_scheduler_.reset(); |
| 112 } | 113 } |
| 113 | 114 |
| 114 void TestScheduleScanNowIfPossible(bool is_authenticated_user_logged_in, | 115 void TestScheduleScanNowIfPossible(bool is_authenticated_user_logged_in, |
| 115 bool is_network_connected_or_connecting, | 116 bool is_network_connected_or_connecting, |
| 116 bool are_tether_hosts_synced, | 117 bool are_tether_hosts_synced, |
| 117 int num_expected_scans) { | 118 int num_expected_scans) { |
| 118 test_context_->SetIsAuthenticatedUserLoggedIn( | 119 test_delegate_->set_is_authenticated_user_logged_in( |
| 119 is_authenticated_user_logged_in); | 120 is_authenticated_user_logged_in); |
| 120 test_context_->SetIsNetworkConnectedOrConnecting( | 121 test_delegate_->set_is_network_connected_or_connecting( |
| 121 is_network_connected_or_connecting); | 122 is_network_connected_or_connecting); |
| 122 test_context_->AreTetherHostsSynced(are_tether_hosts_synced); | 123 test_delegate_->set_are_tether_hosts_synced(are_tether_hosts_synced); |
| 123 host_scan_scheduler_->ScheduleScanNowIfPossible(); | 124 host_scan_scheduler_->ScheduleScanNowIfPossible(); |
| 124 EXPECT_EQ(num_expected_scans, mock_host_scanner_->num_scans_started()); | 125 EXPECT_EQ(num_expected_scans, fake_host_scanner_->num_scans_started()); |
| 125 } | 126 } |
| 126 | 127 |
| 127 std::unique_ptr<HostScanScheduler> host_scan_scheduler_; | 128 std::unique_ptr<HostScanScheduler> host_scan_scheduler_; |
| 128 | 129 |
| 129 TestContext* test_context_; | 130 TestDelegate* test_delegate_; |
| 130 MockHostScanner* mock_host_scanner_; | 131 FakeHostScanner* fake_host_scanner_; |
| 131 }; | 132 }; |
| 132 | 133 |
| 133 TEST_F(HostScanSchedulerTest, TestObserverAddedAndRemoved) { | 134 TEST_F(HostScanSchedulerTest, TestObserverAddedAndRemoved) { |
| 134 // Intentionally empty. This test just ensures that adding/removing observers | 135 // Intentionally empty. This test just ensures that adding/removing observers |
| 135 // (in setUp() and tearDown()) works. | 136 // (in setUp() and tearDown()) works. |
| 136 } | 137 } |
| 137 | 138 |
| 138 TEST_F(HostScanSchedulerTest, TestScheduleScanNowIfPossible) { | 139 TEST_F(HostScanSchedulerTest, TestScheduleScanNowIfPossible) { |
| 139 EXPECT_EQ(0, mock_host_scanner_->num_scans_started()); | 140 EXPECT_EQ(0, fake_host_scanner_->num_scans_started()); |
| 140 | 141 |
| 141 // A scan should only be started when an authenticated user is logged in, | 142 // A scan should only be started when an authenticated user is logged in, |
| 142 // the network is not connected/connecting, and tether hosts are synced. | 143 // the network is not connected/connecting, and tether hosts are synced. |
| 143 TestScheduleScanNowIfPossible(false, false, false, 0); | 144 TestScheduleScanNowIfPossible(false, false, false, 0); |
| 144 TestScheduleScanNowIfPossible(true, false, false, 0); | 145 TestScheduleScanNowIfPossible(true, false, false, 0); |
| 145 TestScheduleScanNowIfPossible(false, true, false, 0); | 146 TestScheduleScanNowIfPossible(false, true, false, 0); |
| 146 TestScheduleScanNowIfPossible(false, false, true, 0); | 147 TestScheduleScanNowIfPossible(false, false, true, 0); |
| 147 TestScheduleScanNowIfPossible(true, true, false, 0); | 148 TestScheduleScanNowIfPossible(true, true, false, 0); |
| 148 TestScheduleScanNowIfPossible(true, false, true, 1); | 149 TestScheduleScanNowIfPossible(true, false, true, 1); |
| 149 TestScheduleScanNowIfPossible(false, true, true, 1); | 150 TestScheduleScanNowIfPossible(false, true, true, 1); |
| 150 TestScheduleScanNowIfPossible(true, true, true, 1); | 151 TestScheduleScanNowIfPossible(true, true, true, 1); |
| 151 } | 152 } |
| 152 | 153 |
| 153 TEST_F(HostScanSchedulerTest, TestLoggedInStateChange) { | 154 TEST_F(HostScanSchedulerTest, TestLoggedInStateChange) { |
| 154 EXPECT_EQ(0, mock_host_scanner_->num_scans_started()); | 155 EXPECT_EQ(0, fake_host_scanner_->num_scans_started()); |
| 155 | 156 |
| 156 test_context_->SetIsAuthenticatedUserLoggedIn(true); | 157 test_delegate_->set_is_authenticated_user_logged_in(true); |
| 157 test_context_->SetIsNetworkConnectedOrConnecting(false); | 158 test_delegate_->set_is_network_connected_or_connecting(false); |
| 158 test_context_->AreTetherHostsSynced(true); | 159 test_delegate_->set_are_tether_hosts_synced(true); |
| 159 | 160 |
| 160 host_scan_scheduler_->LoggedInStateChanged(); | 161 host_scan_scheduler_->LoggedInStateChanged(); |
| 161 EXPECT_EQ(1, mock_host_scanner_->num_scans_started()); | 162 EXPECT_EQ(1, fake_host_scanner_->num_scans_started()); |
| 162 | 163 |
| 163 // Change a condition so that it should not scan, and re-trigger; there should | 164 // Change a condition so that it should not scan, and re-trigger; there should |
| 164 // still be only 1 started scan. | 165 // still be only 1 started scan. |
| 165 test_context_->SetIsAuthenticatedUserLoggedIn(false); | 166 test_delegate_->set_is_authenticated_user_logged_in(false); |
| 166 host_scan_scheduler_->LoggedInStateChanged(); | 167 host_scan_scheduler_->LoggedInStateChanged(); |
| 167 EXPECT_EQ(1, mock_host_scanner_->num_scans_started()); | 168 EXPECT_EQ(1, fake_host_scanner_->num_scans_started()); |
| 168 } | 169 } |
| 169 | 170 |
| 170 TEST_F(HostScanSchedulerTest, TestSuspendDone) { | 171 TEST_F(HostScanSchedulerTest, TestSuspendDone) { |
| 171 EXPECT_EQ(0, mock_host_scanner_->num_scans_started()); | 172 EXPECT_EQ(0, fake_host_scanner_->num_scans_started()); |
| 172 | 173 |
| 173 test_context_->SetIsAuthenticatedUserLoggedIn(true); | 174 test_delegate_->set_is_authenticated_user_logged_in(true); |
| 174 test_context_->SetIsNetworkConnectedOrConnecting(false); | 175 test_delegate_->set_is_network_connected_or_connecting(false); |
| 175 test_context_->AreTetherHostsSynced(true); | 176 test_delegate_->set_are_tether_hosts_synced(true); |
| 176 | 177 |
| 177 host_scan_scheduler_->SuspendDone(base::TimeDelta::FromSeconds(1)); | 178 host_scan_scheduler_->SuspendDone(base::TimeDelta::FromSeconds(1)); |
| 178 EXPECT_EQ(1, mock_host_scanner_->num_scans_started()); | 179 EXPECT_EQ(1, fake_host_scanner_->num_scans_started()); |
| 179 | 180 |
| 180 // Change a condition so that it should not scan, and re-trigger; there should | 181 // Change a condition so that it should not scan, and re-trigger; there should |
| 181 // still be only 1 started scan. | 182 // still be only 1 started scan. |
| 182 test_context_->SetIsAuthenticatedUserLoggedIn(false); | 183 test_delegate_->set_is_authenticated_user_logged_in(false); |
| 183 host_scan_scheduler_->SuspendDone(base::TimeDelta::FromSeconds(1)); | 184 host_scan_scheduler_->SuspendDone(base::TimeDelta::FromSeconds(1)); |
| 184 EXPECT_EQ(1, mock_host_scanner_->num_scans_started()); | 185 EXPECT_EQ(1, fake_host_scanner_->num_scans_started()); |
| 185 } | 186 } |
| 186 | 187 |
| 187 TEST_F(HostScanSchedulerTest, TestNetworkConnectionStateChanged) { | 188 TEST_F(HostScanSchedulerTest, TestNetworkConnectionStateChanged) { |
| 188 EXPECT_EQ(0, mock_host_scanner_->num_scans_started()); | 189 EXPECT_EQ(0, fake_host_scanner_->num_scans_started()); |
| 189 | 190 |
| 190 test_context_->SetIsAuthenticatedUserLoggedIn(true); | 191 test_delegate_->set_is_authenticated_user_logged_in(true); |
| 191 test_context_->SetIsNetworkConnectedOrConnecting(false); | 192 test_delegate_->set_is_network_connected_or_connecting(false); |
| 192 test_context_->AreTetherHostsSynced(true); | 193 test_delegate_->set_are_tether_hosts_synced(true); |
| 193 | 194 |
| 194 host_scan_scheduler_->NetworkConnectionStateChanged(nullptr); | 195 host_scan_scheduler_->NetworkConnectionStateChanged(nullptr); |
| 195 EXPECT_EQ(1, mock_host_scanner_->num_scans_started()); | 196 EXPECT_EQ(1, fake_host_scanner_->num_scans_started()); |
| 196 | 197 |
| 197 // Change a condition so that it should not scan, and re-trigger; there should | 198 // Change a condition so that it should not scan, and re-trigger; there should |
| 198 // still be only 1 started scan. | 199 // still be only 1 started scan. |
| 199 test_context_->SetIsNetworkConnectedOrConnecting(true); | 200 test_delegate_->set_is_network_connected_or_connecting(true); |
| 200 host_scan_scheduler_->NetworkConnectionStateChanged(nullptr); | 201 host_scan_scheduler_->NetworkConnectionStateChanged(nullptr); |
| 201 EXPECT_EQ(1, mock_host_scanner_->num_scans_started()); | 202 EXPECT_EQ(1, fake_host_scanner_->num_scans_started()); |
| 202 } | 203 } |
| 203 | 204 |
| 204 TEST_F(HostScanSchedulerTest, TestOnSyncFinished) { | 205 TEST_F(HostScanSchedulerTest, TestOnSyncFinished) { |
| 205 EXPECT_EQ(0, mock_host_scanner_->num_scans_started()); | 206 EXPECT_EQ(0, fake_host_scanner_->num_scans_started()); |
| 206 | 207 |
| 207 test_context_->SetIsAuthenticatedUserLoggedIn(true); | 208 test_delegate_->set_is_authenticated_user_logged_in(true); |
| 208 test_context_->SetIsNetworkConnectedOrConnecting(false); | 209 test_delegate_->set_is_network_connected_or_connecting(false); |
| 209 test_context_->AreTetherHostsSynced(true); | 210 test_delegate_->set_are_tether_hosts_synced(true); |
| 210 | 211 |
| 211 host_scan_scheduler_->OnSyncFinished( | 212 host_scan_scheduler_->OnSyncFinished( |
| 212 cryptauth::CryptAuthDeviceManager::SyncResult::SUCCESS, | 213 cryptauth::CryptAuthDeviceManager::SyncResult::SUCCESS, |
| 213 cryptauth::CryptAuthDeviceManager::DeviceChangeResult::CHANGED); | 214 cryptauth::CryptAuthDeviceManager::DeviceChangeResult::CHANGED); |
| 214 EXPECT_EQ(1, mock_host_scanner_->num_scans_started()); | 215 EXPECT_EQ(1, fake_host_scanner_->num_scans_started()); |
| 215 | 216 |
| 216 // Change a condition so that it should not scan, and re-trigger; there should | 217 // Change a condition so that it should not scan, and re-trigger; there should |
| 217 // still be only 1 started scan. | 218 // still be only 1 started scan. |
| 218 test_context_->AreTetherHostsSynced(false); | 219 test_delegate_->set_are_tether_hosts_synced(false); |
| 219 host_scan_scheduler_->OnSyncFinished( | 220 host_scan_scheduler_->OnSyncFinished( |
| 220 cryptauth::CryptAuthDeviceManager::SyncResult::SUCCESS, | 221 cryptauth::CryptAuthDeviceManager::SyncResult::SUCCESS, |
| 221 cryptauth::CryptAuthDeviceManager::DeviceChangeResult::UNCHANGED); | 222 cryptauth::CryptAuthDeviceManager::DeviceChangeResult::UNCHANGED); |
| 222 EXPECT_EQ(1, mock_host_scanner_->num_scans_started()); | 223 EXPECT_EQ(1, fake_host_scanner_->num_scans_started()); |
| 223 } | 224 } |
| 224 | 225 |
| 225 } // namespace tether | 226 } // namespace tether |
| 226 | 227 |
| 227 } // namespace chromeos | 228 } // namespace chromeos |
| OLD | NEW |