host_device_timestamp_manager_impl_unittest.cc 5.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142
  1. // Copyright 2018 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "ash/services/multidevice_setup/host_device_timestamp_manager_impl.h"
  5. #include <memory>
  6. #include "ash/components/multidevice/remote_device_test_util.h"
  7. #include "ash/services/multidevice_setup/fake_host_status_provider.h"
  8. #include "base/test/simple_test_clock.h"
  9. #include "base/time/time.h"
  10. #include "components/sync_preferences/testing_pref_service_syncable.h"
  11. #include "testing/gtest/include/gtest/gtest.h"
  12. namespace ash {
  13. namespace multidevice_setup {
  14. namespace {
  15. const base::Time kTestTime = base::Time::FromJavaTime(1500000000000);
  16. const base::Time kLaterTime = kTestTime + base::Milliseconds(123456789);
  17. } // namespace
  18. class HostDeviceTimestampManagerImplTest : public testing::Test {
  19. public:
  20. HostDeviceTimestampManagerImplTest(
  21. const HostDeviceTimestampManagerImplTest&) = delete;
  22. HostDeviceTimestampManagerImplTest& operator=(
  23. const HostDeviceTimestampManagerImplTest&) = delete;
  24. protected:
  25. HostDeviceTimestampManagerImplTest() = default;
  26. ~HostDeviceTimestampManagerImplTest() override = default;
  27. void SetUp() override {
  28. test_pref_service_ =
  29. std::make_unique<sync_preferences::TestingPrefServiceSyncable>();
  30. HostDeviceTimestampManagerImpl::RegisterPrefs(
  31. test_pref_service_->registry());
  32. fake_host_status_provider_ = std::make_unique<FakeHostStatusProvider>();
  33. test_clock_ = std::make_unique<base::SimpleTestClock>();
  34. SetNow(kTestTime);
  35. manager_ = HostDeviceTimestampManagerImpl::Factory::Create(
  36. fake_host_status_provider_.get(), test_pref_service_.get(),
  37. test_clock_.get());
  38. }
  39. // If the status corresponds to a set host, we set a dummy default device
  40. // because the device info is irrelevant.
  41. void SetHostStatus(mojom::HostStatus host_status) {
  42. if (host_status == mojom::HostStatus::kNoEligibleHosts ||
  43. host_status == mojom::HostStatus::kEligibleHostExistsButNoHostSet) {
  44. fake_host_status_provider_->SetHostWithStatus(
  45. host_status, absl::nullopt /* host_device */);
  46. return;
  47. }
  48. fake_host_status_provider_->SetHostWithStatus(
  49. host_status, multidevice::RemoteDeviceRefBuilder()
  50. .SetPublicKey("fake-phone-key")
  51. .SetName("Fake Phone Name")
  52. .Build());
  53. }
  54. void SetNow(const base::Time now) { test_clock_->SetNow(now); }
  55. HostDeviceTimestampManager* manager() { return manager_.get(); }
  56. private:
  57. std::unique_ptr<sync_preferences::TestingPrefServiceSyncable>
  58. test_pref_service_;
  59. std::unique_ptr<FakeHostStatusProvider> fake_host_status_provider_;
  60. std::unique_ptr<base::SimpleTestClock> test_clock_;
  61. std::unique_ptr<HostDeviceTimestampManager> manager_;
  62. };
  63. TEST_F(HostDeviceTimestampManagerImplTest,
  64. RecordsWhetherHostWasSetFromThisChromebook) {
  65. EXPECT_FALSE(manager()->WasHostSetFromThisChromebook());
  66. // Discover potential host.
  67. SetHostStatus(mojom::HostStatus::kEligibleHostExistsButNoHostSet);
  68. // Set up host.
  69. SetHostStatus(
  70. mojom::HostStatus::kHostSetLocallyButWaitingForBackendConfirmation);
  71. EXPECT_TRUE(manager()->WasHostSetFromThisChromebook());
  72. // Verify host.
  73. SetHostStatus(mojom::HostStatus::kHostVerified);
  74. EXPECT_TRUE(manager()->WasHostSetFromThisChromebook());
  75. // Forget the host.
  76. SetHostStatus(mojom::HostStatus::kEligibleHostExistsButNoHostSet);
  77. EXPECT_FALSE(manager()->WasHostSetFromThisChromebook());
  78. // Set verified host from a different Chromebook
  79. SetHostStatus(mojom::HostStatus::kHostVerified);
  80. EXPECT_FALSE(manager()->WasHostSetFromThisChromebook());
  81. }
  82. TEST_F(HostDeviceTimestampManagerImplTest, RecordsCorrectCompletionTime) {
  83. EXPECT_FALSE(manager()->GetLatestSetupFlowCompletionTimestamp());
  84. SetHostStatus(
  85. mojom::HostStatus::kHostSetLocallyButWaitingForBackendConfirmation);
  86. EXPECT_EQ(kTestTime, manager()->GetLatestSetupFlowCompletionTimestamp());
  87. // Forget the host.
  88. SetHostStatus(mojom::HostStatus::kEligibleHostExistsButNoHostSet);
  89. // Set up later and check that the new time replaces the old.
  90. SetNow(kLaterTime);
  91. SetHostStatus(
  92. mojom::HostStatus::kHostSetLocallyButWaitingForBackendConfirmation);
  93. EXPECT_EQ(kLaterTime, manager()->GetLatestSetupFlowCompletionTimestamp());
  94. }
  95. TEST_F(HostDeviceTimestampManagerImplTest, RecordsCorrectVerificationTime) {
  96. EXPECT_FALSE(manager()->GetLatestVerificationTimestamp());
  97. // Update with verified host.
  98. SetHostStatus(mojom::HostStatus::kHostVerified);
  99. EXPECT_EQ(kTestTime, manager()->GetLatestVerificationTimestamp());
  100. // Forget the host.
  101. SetHostStatus(mojom::HostStatus::kEligibleHostExistsButNoHostSet);
  102. // Set up later and check that the new time replaces the old.
  103. SetNow(kLaterTime);
  104. SetHostStatus(mojom::HostStatus::kHostVerified);
  105. EXPECT_EQ(kLaterTime, manager()->GetLatestVerificationTimestamp());
  106. }
  107. TEST_F(HostDeviceTimestampManagerImplTest,
  108. StoresCompletionsAndVerificationSimultaneously) {
  109. SetHostStatus(
  110. mojom::HostStatus::kHostSetLocallyButWaitingForBackendConfirmation);
  111. SetNow(kLaterTime);
  112. SetHostStatus(mojom::HostStatus::kHostVerified);
  113. EXPECT_EQ(kTestTime, manager()->GetLatestSetupFlowCompletionTimestamp());
  114. EXPECT_EQ(kLaterTime, manager()->GetLatestVerificationTimestamp());
  115. }
  116. } // namespace multidevice_setup
  117. } // namespace ash