Tether: only scan when asked to by NetworkStateHandler.

chromeos/components/tether/host_scan_scheduler_unittest.cc

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chromeos/components/tether/host_scan_scheduler.h"
 
 #include "base/memory/ptr_util.h"
+#include "base/run_loop.h"
+#include "base/test/scoped_task_environment.h"
 #include "chromeos/components/tether/host_scanner.h"
+#include "chromeos/dbus/dbus_thread_manager.h"
 #include "chromeos/dbus/power_manager_client.h"
 #include "chromeos/login/login_state.h"
 #include "chromeos/network/network_handler.h"
 #include "chromeos/network/network_state.h"
 #include "chromeos/network/network_state_handler.h"
+#include "chromeos/network/network_state_test.h"
+#include "chromeos/network/network_type_pattern.h"
 #include "components/cryptauth/cryptauth_device_manager.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace chromeos {
 
 namespace tether {
 
-class HostScanSchedulerTest : public testing::Test {
+namespace {
+class FakeHostScanner : public HostScanner {
+ public:
+  FakeHostScanner()
+      : HostScanner(nullptr,
+                    nullptr,
+                    nullptr,
+                    nullptr,
+                    nullptr,
+                    nullptr,
+                    nullptr,
+                    nullptr),
+        num_scans_started_(0) {}
+  ~FakeHostScanner() override {}
+
+  void StartScan() override {
+    is_scan_active_ = true;
+    has_recently_scanned_ = true;
+    num_scans_started_++;
+  }
+
+  void StopScan() { is_scan_active_ = false; }
+
+  bool IsScanActive() override { return is_scan_active_; }
+
+  bool HasRecentlyScanned() override { return has_recently_scanned_; }
+
+  void set_has_recently_scanned(bool has_recently_scanned) {
+    has_recently_scanned_ = has_recently_scanned;
+  }
+
+  int num_scans_started() { return num_scans_started_; }
+
+ private:
+  bool is_scan_active_ = false;
+  bool has_recently_scanned_ = false;
+  int num_scans_started_ = 0;
+};
+}  // namespace
+
+class HostScanSchedulerTest : public NetworkStateTest {
  protected:
-  class TestDelegate : public HostScanScheduler::Delegate {
-   public:
-    explicit TestDelegate(HostScanSchedulerTest* test)
-        : test_(test),
-          observer_(nullptr),
-          is_authenticated_user_logged_in(true),
-          is_network_connected_or_connecting(false),
-          are_tether_hosts_synced(true) {}
+  void SetUp() override {
+    DBusThreadManager::Initialize();
+    NetworkStateTest::SetUp();
 
-    void AddObserver(HostScanScheduler* host_scan_scheduler) override {
-      observer_ = host_scan_scheduler;
-    }
+    network_state_handler()->SetTetherTechnologyState(
+        NetworkStateHandler::TECHNOLOGY_ENABLED);
 
-    void RemoveObserver(HostScanScheduler* host_scan_scheduler) override {
-      if (observer_ == host_scan_scheduler) {
-        observer_ = nullptr;
-      }
-      EXPECT_FALSE(IsObserverSet());
-    }
+    fake_host_scanner_ = base::MakeUnique<FakeHostScanner>();
 
-    bool IsAuthenticatedUserLoggedIn() const override {
-      return is_authenticated_user_logged_in;
-    }
-
-    bool IsNetworkConnectedOrConnecting() const override {
-      return is_network_connected_or_connecting;
-    }
-
-    bool AreTetherHostsSynced() const override {
-      return are_tether_hosts_synced;
-    }
-
-    bool IsObserverSet() const {
-      return observer_ != nullptr &&
-             observer_ == test_->host_scan_scheduler_.get();
-    }
-
-    void set_is_authenticated_user_logged_in(bool value) {
-      is_authenticated_user_logged_in = value;
-    }
-
-    void set_is_network_connected_or_connecting(bool value) {
-      is_network_connected_or_connecting = value;
-    }
-
-    void set_are_tether_hosts_synced(bool value) {
-      are_tether_hosts_synced = value;
-    }
-
-   private:
-    const HostScanSchedulerTest* test_;
-
-    HostScanScheduler* observer_;
-    bool is_authenticated_user_logged_in;
-    bool is_network_connected_or_connecting;
-    bool are_tether_hosts_synced;
-  };
-
-  class FakeHostScanner : public HostScanner {
-   public:
-    FakeHostScanner()
-        : HostScanner(nullptr,
-                      nullptr,
-                      nullptr,
-                      nullptr,
-                      nullptr,
-                      nullptr,
-                      nullptr),
-          num_scans_started_(0) {}
-    ~FakeHostScanner() override {}
-
-    void StartScan() override { num_scans_started_++; }
-
-    int num_scans_started() { return num_scans_started_; }
-
-   private:
-    int num_scans_started_;
-  };
-
-  HostScanSchedulerTest() {}
-
-  void SetUp() override {
-    test_delegate_ = new TestDelegate(this);
-    fake_host_scanner_ = new FakeHostScanner();
-
-    host_scan_scheduler_.reset(
-        new HostScanScheduler(base::WrapUnique(test_delegate_),
-                              base::WrapUnique(fake_host_scanner_)));
-
-    EXPECT_FALSE(test_delegate_->IsObserverSet());
-    host_scan_scheduler_->InitializeAutomaticScans();
-    EXPECT_TRUE(test_delegate_->IsObserverSet());
+    host_scan_scheduler_ = base::WrapUnique(new HostScanScheduler(
+        network_state_handler(), fake_host_scanner_.get()));
   }
 
   void TearDown() override {
-    EXPECT_TRUE(test_delegate_->IsObserverSet());
     host_scan_scheduler_.reset();
+
+    ShutdownNetworkState();
+    NetworkStateTest::TearDown();
+    DBusThreadManager::Shutdown();
   }
 
-  void TestScheduleScanNowIfPossible(bool is_authenticated_user_logged_in,
-                                     bool is_network_connected_or_connecting,
-                                     bool are_tether_hosts_synced,
-                                     int num_expected_scans) {
-    test_delegate_->set_is_authenticated_user_logged_in(
-        is_authenticated_user_logged_in);
-    test_delegate_->set_is_network_connected_or_connecting(
-        is_network_connected_or_connecting);
-    test_delegate_->set_are_tether_hosts_synced(are_tether_hosts_synced);
-    host_scan_scheduler_->ScheduleScanNowIfPossible();
-    EXPECT_EQ(num_expected_scans, fake_host_scanner_->num_scans_started());
-  }
+  base::test::ScopedTaskEnvironment scoped_task_environment_;
+
+  std::unique_ptr<FakeHostScanner> fake_host_scanner_;
 
   std::unique_ptr<HostScanScheduler> host_scan_scheduler_;
-
-  TestDelegate* test_delegate_;
-  FakeHostScanner* fake_host_scanner_;
 };
 
-TEST_F(HostScanSchedulerTest, TestObserverAddedAndRemoved) {
-  // Intentionally empty. This test just ensures that adding/removing observers
-  // (in setUp() and tearDown()) works.
-}
+TEST_F(HostScanSchedulerTest, TestScanRequested) {
+  EXPECT_EQ(0, fake_host_scanner_->num_scans_started());
+  EXPECT_FALSE(
+      network_state_handler()->GetScanningByType(NetworkTypePattern::Tether()));
 
-TEST_F(HostScanSchedulerTest, TestScheduleScanNowIfPossible) {
-  EXPECT_EQ(0, fake_host_scanner_->num_scans_started());
+  // Begin scanning.
+  host_scan_scheduler_->ScanRequested();
+  EXPECT_EQ(1, fake_host_scanner_->num_scans_started());
+  EXPECT_TRUE(
+      network_state_handler()->GetScanningByType(NetworkTypePattern::Tether()));
 
-  // A scan should only be started when an authenticated user is logged in,
-  // the network is not connected/connecting, and tether hosts are synced.
-  TestScheduleScanNowIfPossible(false, false, false, 0);
-  TestScheduleScanNowIfPossible(true, false, false, 0);
-  TestScheduleScanNowIfPossible(false, true, false, 0);
-  TestScheduleScanNowIfPossible(false, false, true, 0);
-  TestScheduleScanNowIfPossible(true, true, false, 0);
-  TestScheduleScanNowIfPossible(true, false, true, 1);
-  TestScheduleScanNowIfPossible(false, true, true, 1);
-  TestScheduleScanNowIfPossible(true, true, true, 1);
-}
+  // Should not begin a new scan while a scan is active.
+  host_scan_scheduler_->ScanRequested();
+  EXPECT_EQ(1, fake_host_scanner_->num_scans_started());
+  EXPECT_TRUE(
+      network_state_handler()->GetScanningByType(NetworkTypePattern::Tether()));
 
-TEST_F(HostScanSchedulerTest, TestLoggedInStateChange) {
-  EXPECT_EQ(0, fake_host_scanner_->num_scans_started());
+  fake_host_scanner_->StopScan();
+  host_scan_scheduler_->ScanFinished();
+  EXPECT_EQ(1, fake_host_scanner_->num_scans_started());
+  EXPECT_FALSE(
+      network_state_handler()->GetScanningByType(NetworkTypePattern::Tether()));
 
-  test_delegate_->set_is_authenticated_user_logged_in(true);
-  test_delegate_->set_is_network_connected_or_connecting(false);
-  test_delegate_->set_are_tether_hosts_synced(true);
+  // Should not begin a new scan when a scan has finished recently.
+  host_scan_scheduler_->ScanRequested();
+  EXPECT_EQ(1, fake_host_scanner_->num_scans_started());
+  EXPECT_FALSE(
+      network_state_handler()->GetScanningByType(NetworkTypePattern::Tether()));
 
-  host_scan_scheduler_->LoggedInStateChanged();
-  EXPECT_EQ(1, fake_host_scanner_->num_scans_started());
-
-  // Change a condition so that it should not scan, and re-trigger; there should
-  // still be only 1 started scan.
-  test_delegate_->set_is_authenticated_user_logged_in(false);
-  host_scan_scheduler_->LoggedInStateChanged();
-  EXPECT_EQ(1, fake_host_scanner_->num_scans_started());
-}
-
-TEST_F(HostScanSchedulerTest, TestSuspendDone) {
-  EXPECT_EQ(0, fake_host_scanner_->num_scans_started());
-
-  test_delegate_->set_is_authenticated_user_logged_in(true);
-  test_delegate_->set_is_network_connected_or_connecting(false);
-  test_delegate_->set_are_tether_hosts_synced(true);
-
-  host_scan_scheduler_->SuspendDone(base::TimeDelta::FromSeconds(1));
-  EXPECT_EQ(1, fake_host_scanner_->num_scans_started());
-
-  // Change a condition so that it should not scan, and re-trigger; there should
-  // still be only 1 started scan.
-  test_delegate_->set_is_authenticated_user_logged_in(false);
-  host_scan_scheduler_->SuspendDone(base::TimeDelta::FromSeconds(1));
-  EXPECT_EQ(1, fake_host_scanner_->num_scans_started());
-}
-
-TEST_F(HostScanSchedulerTest, TestNetworkConnectionStateChanged) {
-  EXPECT_EQ(0, fake_host_scanner_->num_scans_started());
-
-  test_delegate_->set_is_authenticated_user_logged_in(true);
-  test_delegate_->set_is_network_connected_or_connecting(false);
-  test_delegate_->set_are_tether_hosts_synced(true);
-
-  host_scan_scheduler_->NetworkConnectionStateChanged(nullptr);
-  EXPECT_EQ(1, fake_host_scanner_->num_scans_started());
-
-  // Change a condition so that it should not scan, and re-trigger; there should
-  // still be only 1 started scan.
-  test_delegate_->set_is_network_connected_or_connecting(true);
-  host_scan_scheduler_->NetworkConnectionStateChanged(nullptr);
-  EXPECT_EQ(1, fake_host_scanner_->num_scans_started());
-}
-
-TEST_F(HostScanSchedulerTest, TestOnSyncFinished) {
-  EXPECT_EQ(0, fake_host_scanner_->num_scans_started());
-
-  test_delegate_->set_is_authenticated_user_logged_in(true);
-  test_delegate_->set_is_network_connected_or_connecting(false);
-  test_delegate_->set_are_tether_hosts_synced(true);
-
-  host_scan_scheduler_->OnSyncFinished(
-      cryptauth::CryptAuthDeviceManager::SyncResult::SUCCESS,
-      cryptauth::CryptAuthDeviceManager::DeviceChangeResult::CHANGED);
-  EXPECT_EQ(1, fake_host_scanner_->num_scans_started());
-
-  // Change a condition so that it should not scan, and re-trigger; there should
-  // still be only 1 started scan.
-  test_delegate_->set_are_tether_hosts_synced(false);
-  host_scan_scheduler_->OnSyncFinished(
-      cryptauth::CryptAuthDeviceManager::SyncResult::SUCCESS,
-      cryptauth::CryptAuthDeviceManager::DeviceChangeResult::UNCHANGED);
-  EXPECT_EQ(1, fake_host_scanner_->num_scans_started());
+  // A new scan should be allowed once a scan is not active and has not finished
+  // recently.
+  fake_host_scanner_->set_has_recently_scanned(false);
+  host_scan_scheduler_->ScanRequested();
+  EXPECT_EQ(2, fake_host_scanner_->num_scans_started());
+  EXPECT_TRUE(
+      network_state_handler()->GetScanningByType(NetworkTypePattern::Tether()));
 }
 
 }  // namespace tether
 
 }  // namespace chromeos