onboarding_ui_tracker_impl_unittest.cc 6.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198
  1. // Copyright 2020 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/components/phonehub/onboarding_ui_tracker_impl.h"
  5. #include <memory>
  6. #include "ash/components/phonehub/fake_feature_status_provider.h"
  7. #include "ash/components/phonehub/feature_status.h"
  8. #include "ash/components/phonehub/pref_names.h"
  9. #include "ash/services/multidevice_setup/public/cpp/fake_multidevice_setup_client.h"
  10. #include "base/bind.h"
  11. #include "base/callback.h"
  12. #include "base/logging.h"
  13. #include "components/prefs/testing_pref_service.h"
  14. #include "testing/gtest/include/gtest/gtest.h"
  15. namespace ash {
  16. namespace phonehub {
  17. namespace {
  18. using multidevice_setup::mojom::Feature;
  19. using multidevice_setup::mojom::FeatureState;
  20. class FakeObserver : public OnboardingUiTracker::Observer {
  21. public:
  22. FakeObserver() = default;
  23. ~FakeObserver() override = default;
  24. size_t num_calls() const { return num_calls_; }
  25. // OnboardingUiTracker::Observer:
  26. void OnShouldShowOnboardingUiChanged() override { ++num_calls_; }
  27. private:
  28. size_t num_calls_ = 0;
  29. };
  30. } // namespace
  31. class OnboardingUiTrackerImplTest : public testing::Test {
  32. protected:
  33. OnboardingUiTrackerImplTest() = default;
  34. OnboardingUiTrackerImplTest(const OnboardingUiTrackerImplTest&) = delete;
  35. OnboardingUiTrackerImplTest& operator=(const OnboardingUiTrackerImplTest&) =
  36. delete;
  37. ~OnboardingUiTrackerImplTest() override = default;
  38. // testing::Test:
  39. void SetUp() override {
  40. OnboardingUiTrackerImpl::RegisterPrefs(pref_service_.registry());
  41. fake_feature_status_provider_ =
  42. std::make_unique<FakeFeatureStatusProvider>();
  43. fake_feature_status_provider_->SetStatus(
  44. FeatureStatus::kNotEligibleForFeature);
  45. controller_ = std::make_unique<OnboardingUiTrackerImpl>(
  46. &pref_service_, fake_feature_status_provider_.get(),
  47. &fake_multidevice_setup_client_,
  48. base::BindRepeating(
  49. &OnboardingUiTrackerImplTest::ShowMultideviceSetupDialog,
  50. base::Unretained(this)));
  51. controller_->AddObserver(&fake_observer_);
  52. }
  53. void TearDown() override { controller_->RemoveObserver(&fake_observer_); }
  54. void ShowMultideviceSetupDialog() { ++setup_dialog_shown_count_; }
  55. size_t setup_dialog_shown_count() { return setup_dialog_shown_count_; }
  56. void SetStatus(FeatureStatus feature_status) {
  57. fake_feature_status_provider_->SetStatus(feature_status);
  58. }
  59. void SetFeatureState(Feature feature, FeatureState state) {
  60. fake_multidevice_setup_client_.SetFeatureState(feature, state);
  61. }
  62. void DismissSetupUi() { controller_->DismissSetupUi(); }
  63. bool ShouldShowOnboardingUi() const {
  64. return controller_->ShouldShowOnboardingUi();
  65. }
  66. size_t GetNumObserverCalls() const { return fake_observer_.num_calls(); }
  67. void HandleGetStarted() { controller_->HandleGetStarted(); }
  68. void InvokePendingSetFeatureEnabledStateCallback(bool expected_enabled) {
  69. fake_multidevice_setup_client_.InvokePendingSetFeatureEnabledStateCallback(
  70. Feature::kPhoneHub, expected_enabled, absl::nullopt, true);
  71. }
  72. size_t GetOnShouldShowOnboardingUiChangedCallCount() {
  73. return fake_observer_.num_calls();
  74. }
  75. private:
  76. size_t setup_dialog_shown_count_ = 0;
  77. TestingPrefServiceSimple pref_service_;
  78. multidevice_setup::FakeMultiDeviceSetupClient fake_multidevice_setup_client_;
  79. std::unique_ptr<FakeFeatureStatusProvider> fake_feature_status_provider_;
  80. FakeObserver fake_observer_;
  81. std::unique_ptr<OnboardingUiTracker> controller_;
  82. };
  83. TEST_F(OnboardingUiTrackerImplTest, ShouldShowUiWhenEligiblePhoneButNotSetup) {
  84. SetStatus(FeatureStatus::kNotEligibleForFeature);
  85. EXPECT_EQ(GetOnShouldShowOnboardingUiChangedCallCount(), 0U);
  86. SetStatus(FeatureStatus::kEligiblePhoneButNotSetUp);
  87. EXPECT_EQ(GetOnShouldShowOnboardingUiChangedCallCount(), 1U);
  88. EXPECT_TRUE(ShouldShowOnboardingUi());
  89. // User clicks get started button.
  90. EXPECT_EQ(setup_dialog_shown_count(), 0U);
  91. HandleGetStarted();
  92. EXPECT_EQ(setup_dialog_shown_count(), 1U);
  93. EXPECT_EQ(GetOnShouldShowOnboardingUiChangedCallCount(), 1U);
  94. // User dismisses setup flow. After dismissal, ShouldShowOnboardingUi() should
  95. // always be false.
  96. DismissSetupUi();
  97. EXPECT_EQ(GetOnShouldShowOnboardingUiChangedCallCount(), 2U);
  98. EXPECT_FALSE(ShouldShowOnboardingUi());
  99. // User clicks the get started button a second time; should still open.
  100. EXPECT_EQ(setup_dialog_shown_count(), 1U);
  101. HandleGetStarted();
  102. EXPECT_EQ(setup_dialog_shown_count(), 2U);
  103. EXPECT_EQ(GetOnShouldShowOnboardingUiChangedCallCount(), 2U);
  104. // User dismisses setup flow a second time.
  105. DismissSetupUi();
  106. EXPECT_EQ(GetOnShouldShowOnboardingUiChangedCallCount(), 2U);
  107. EXPECT_FALSE(ShouldShowOnboardingUi());
  108. }
  109. TEST_F(OnboardingUiTrackerImplTest, ShouldShowUiWhenDisabled) {
  110. SetStatus(FeatureStatus::kNotEligibleForFeature);
  111. EXPECT_EQ(GetOnShouldShowOnboardingUiChangedCallCount(), 0U);
  112. SetStatus(FeatureStatus::kDisabled);
  113. EXPECT_EQ(GetOnShouldShowOnboardingUiChangedCallCount(), 1U);
  114. EXPECT_TRUE(ShouldShowOnboardingUi());
  115. // User clicks get started button.
  116. HandleGetStarted();
  117. InvokePendingSetFeatureEnabledStateCallback(true);
  118. EXPECT_EQ(GetOnShouldShowOnboardingUiChangedCallCount(), 1U);
  119. // User dismisses setup flow. After dismissal, ShouldShowOnboardingUi() should
  120. // always be false.
  121. DismissSetupUi();
  122. EXPECT_EQ(GetOnShouldShowOnboardingUiChangedCallCount(), 2U);
  123. EXPECT_FALSE(ShouldShowOnboardingUi());
  124. // User clicks the get started button a second time; should still open.
  125. HandleGetStarted();
  126. InvokePendingSetFeatureEnabledStateCallback(true);
  127. EXPECT_EQ(GetOnShouldShowOnboardingUiChangedCallCount(), 2U);
  128. // User dismisses setup flow a second time.
  129. DismissSetupUi();
  130. EXPECT_EQ(GetOnShouldShowOnboardingUiChangedCallCount(), 2U);
  131. EXPECT_FALSE(ShouldShowOnboardingUi());
  132. }
  133. TEST_F(OnboardingUiTrackerImplTest, HideUiWhenFeatureIsEnabled) {
  134. SetStatus(FeatureStatus::kNotEligibleForFeature);
  135. EXPECT_EQ(GetOnShouldShowOnboardingUiChangedCallCount(), 0U);
  136. SetStatus(FeatureStatus::kDisabled);
  137. EXPECT_EQ(GetOnShouldShowOnboardingUiChangedCallCount(), 1U);
  138. EXPECT_TRUE(ShouldShowOnboardingUi());
  139. // Simulate feature disabled feature. Expect onboarding UI to still be
  140. // displayed.
  141. SetFeatureState(Feature::kPhoneHub, FeatureState::kDisabledByUser);
  142. EXPECT_EQ(GetOnShouldShowOnboardingUiChangedCallCount(), 1U);
  143. EXPECT_TRUE(ShouldShowOnboardingUi());
  144. // Toggle the feature to be enabled. Expect onboarding UI to no longer be
  145. // displayed.
  146. SetFeatureState(Feature::kPhoneHub, FeatureState::kEnabledByUser);
  147. EXPECT_EQ(GetOnShouldShowOnboardingUiChangedCallCount(), 2U);
  148. EXPECT_FALSE(ShouldShowOnboardingUi());
  149. // Toggle the feature back to disabled. Expect onboarding UI to still be
  150. // hidden.
  151. SetFeatureState(Feature::kPhoneHub, FeatureState::kDisabledByUser);
  152. EXPECT_EQ(GetOnShouldShowOnboardingUiChangedCallCount(), 2U);
  153. EXPECT_FALSE(ShouldShowOnboardingUi());
  154. }
  155. } // namespace phonehub
  156. } // namespace ash