assistant_state_controller_unittest.cc 8.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239
  1. // Copyright 2019 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/assistant/assistant_state_controller.h"
  5. #include <memory>
  6. #include <string>
  7. #include "ash/assistant/assistant_controller_impl.h"
  8. #include "ash/session/session_controller_impl.h"
  9. #include "ash/shell.h"
  10. #include "ash/test/ash_test_base.h"
  11. #include "chromeos/ash/services/assistant/public/cpp/assistant_prefs.h"
  12. #include "components/prefs/pref_service.h"
  13. namespace ash {
  14. namespace {
  15. using assistant::prefs::AssistantOnboardingMode;
  16. using assistant::prefs::ConsentStatus;
  17. using assistant::prefs::kAssistantConsentStatus;
  18. using assistant::prefs::kAssistantContextEnabled;
  19. using assistant::prefs::kAssistantEnabled;
  20. using assistant::prefs::kAssistantHotwordAlwaysOn;
  21. using assistant::prefs::kAssistantHotwordEnabled;
  22. using assistant::prefs::kAssistantLaunchWithMicOpen;
  23. using assistant::prefs::kAssistantNotificationEnabled;
  24. using assistant::prefs::kAssistantOnboardingMode;
  25. using assistant::prefs::kAssistantOnboardingModeDefault;
  26. using assistant::prefs::kAssistantOnboardingModeEducation;
  27. class TestAssistantStateObserver : public AssistantStateObserver {
  28. public:
  29. TestAssistantStateObserver() = default;
  30. TestAssistantStateObserver(const TestAssistantStateObserver&) = delete;
  31. TestAssistantStateObserver& operator=(const TestAssistantStateObserver&) =
  32. delete;
  33. ~TestAssistantStateObserver() override = default;
  34. // AssistantStateObserver:
  35. void OnAssistantConsentStatusChanged(int consent_status) override {
  36. consent_status_ = consent_status;
  37. }
  38. void OnAssistantContextEnabled(bool context_enabled) override {
  39. context_enabled_ = context_enabled;
  40. }
  41. void OnAssistantSettingsEnabled(bool settings_enabled) override {
  42. settings_enabled_ = settings_enabled;
  43. }
  44. void OnAssistantHotwordAlwaysOn(bool hotword_always_on) override {
  45. hotword_always_on_ = hotword_always_on;
  46. }
  47. void OnAssistantHotwordEnabled(bool hotword_enabled) override {
  48. hotword_enabled_ = hotword_enabled;
  49. }
  50. void OnAssistantLaunchWithMicOpen(bool launch_with_mic_open) override {
  51. launch_with_mic_open_ = launch_with_mic_open;
  52. }
  53. void OnAssistantNotificationEnabled(bool notification_enabled) override {
  54. notification_enabled_ = notification_enabled;
  55. }
  56. void OnAssistantOnboardingModeChanged(
  57. AssistantOnboardingMode onboarding_mode) override {
  58. onboarding_mode_ = onboarding_mode;
  59. }
  60. int consent_status() const { return consent_status_; }
  61. bool context_enabled() const { return context_enabled_; }
  62. bool settings_enabled() const { return settings_enabled_; }
  63. bool hotword_always_on() const { return hotword_always_on_; }
  64. bool hotword_enabled() const { return hotword_enabled_; }
  65. bool launch_with_mic_open() const { return launch_with_mic_open_; }
  66. bool notification_enabled() const { return notification_enabled_; }
  67. AssistantOnboardingMode onboarding_mode() const { return onboarding_mode_; }
  68. private:
  69. int consent_status_ = ConsentStatus::kUnknown;
  70. bool context_enabled_ = false;
  71. bool settings_enabled_ = false;
  72. bool hotword_always_on_ = false;
  73. bool hotword_enabled_ = false;
  74. bool launch_with_mic_open_ = false;
  75. bool notification_enabled_ = false;
  76. AssistantOnboardingMode onboarding_mode_ = AssistantOnboardingMode::kDefault;
  77. };
  78. class AssistantStateControllerTest : public AshTestBase {
  79. public:
  80. AssistantStateControllerTest(const AssistantStateControllerTest&) = delete;
  81. AssistantStateControllerTest& operator=(const AssistantStateControllerTest&) =
  82. delete;
  83. protected:
  84. AssistantStateControllerTest() = default;
  85. ~AssistantStateControllerTest() override = default;
  86. // AshTestBase:
  87. void SetUp() override {
  88. AshTestBase::SetUp();
  89. prefs_ = Shell::Get()->session_controller()->GetPrimaryUserPrefService();
  90. DCHECK(prefs_);
  91. observer_ = std::make_unique<TestAssistantStateObserver>();
  92. }
  93. PrefService* prefs() { return prefs_; }
  94. TestAssistantStateObserver* observer() { return observer_.get(); }
  95. private:
  96. PrefService* prefs_ = nullptr;
  97. std::unique_ptr<TestAssistantStateObserver> observer_;
  98. };
  99. } // namespace
  100. TEST_F(AssistantStateControllerTest, InitObserver) {
  101. prefs()->SetInteger(kAssistantConsentStatus,
  102. ConsentStatus::kActivityControlAccepted);
  103. prefs()->SetBoolean(kAssistantContextEnabled, true);
  104. prefs()->SetBoolean(kAssistantEnabled, true);
  105. prefs()->SetBoolean(kAssistantHotwordAlwaysOn, true);
  106. prefs()->SetBoolean(kAssistantHotwordEnabled, true);
  107. prefs()->SetBoolean(kAssistantLaunchWithMicOpen, true);
  108. prefs()->SetBoolean(kAssistantNotificationEnabled, true);
  109. prefs()->SetString(kAssistantOnboardingMode, kAssistantOnboardingModeDefault);
  110. // The observer class should get an instant notification about the current
  111. // pref value.
  112. AssistantState::Get()->AddObserver(observer());
  113. EXPECT_EQ(observer()->consent_status(),
  114. ConsentStatus::kActivityControlAccepted);
  115. EXPECT_EQ(observer()->context_enabled(), true);
  116. EXPECT_EQ(observer()->settings_enabled(), true);
  117. EXPECT_EQ(observer()->hotword_always_on(), true);
  118. EXPECT_EQ(observer()->hotword_enabled(), true);
  119. EXPECT_EQ(observer()->launch_with_mic_open(), true);
  120. EXPECT_EQ(observer()->notification_enabled(), true);
  121. EXPECT_EQ(observer()->onboarding_mode(), AssistantOnboardingMode::kDefault);
  122. AssistantState::Get()->RemoveObserver(observer());
  123. }
  124. TEST_F(AssistantStateControllerTest, NotifyConsentStatus) {
  125. AssistantState::Get()->AddObserver(observer());
  126. prefs()->SetInteger(kAssistantConsentStatus, ConsentStatus::kUnauthorized);
  127. EXPECT_EQ(observer()->consent_status(), ConsentStatus::kUnauthorized);
  128. prefs()->SetInteger(kAssistantConsentStatus,
  129. ConsentStatus::kActivityControlAccepted);
  130. EXPECT_EQ(observer()->consent_status(),
  131. ConsentStatus::kActivityControlAccepted);
  132. AssistantState::Get()->RemoveObserver(observer());
  133. }
  134. TEST_F(AssistantStateControllerTest, NotifyContextEnabled) {
  135. AssistantState::Get()->AddObserver(observer());
  136. prefs()->SetBoolean(kAssistantContextEnabled, false);
  137. EXPECT_EQ(observer()->context_enabled(), false);
  138. prefs()->SetBoolean(kAssistantContextEnabled, true);
  139. EXPECT_EQ(observer()->context_enabled(), true);
  140. AssistantState::Get()->RemoveObserver(observer());
  141. }
  142. TEST_F(AssistantStateControllerTest, NotifySettingsEnabled) {
  143. AssistantState::Get()->AddObserver(observer());
  144. prefs()->SetBoolean(kAssistantEnabled, false);
  145. EXPECT_EQ(observer()->settings_enabled(), false);
  146. prefs()->SetBoolean(kAssistantEnabled, true);
  147. EXPECT_EQ(observer()->settings_enabled(), true);
  148. AssistantState::Get()->RemoveObserver(observer());
  149. }
  150. TEST_F(AssistantStateControllerTest, NotifyHotwordAlwaysOn) {
  151. AssistantState::Get()->AddObserver(observer());
  152. prefs()->SetBoolean(kAssistantHotwordAlwaysOn, false);
  153. EXPECT_EQ(observer()->hotword_always_on(), false);
  154. prefs()->SetBoolean(kAssistantHotwordAlwaysOn, true);
  155. EXPECT_EQ(observer()->hotword_always_on(), true);
  156. AssistantState::Get()->RemoveObserver(observer());
  157. }
  158. TEST_F(AssistantStateControllerTest, NotifyHotwordEnabled) {
  159. AssistantState::Get()->AddObserver(observer());
  160. prefs()->SetBoolean(kAssistantHotwordEnabled, false);
  161. EXPECT_EQ(observer()->hotword_enabled(), false);
  162. prefs()->SetBoolean(kAssistantHotwordEnabled, true);
  163. EXPECT_EQ(observer()->hotword_enabled(), true);
  164. AssistantState::Get()->RemoveObserver(observer());
  165. }
  166. TEST_F(AssistantStateControllerTest, NotifyLaunchWithMicOpen) {
  167. AssistantState::Get()->AddObserver(observer());
  168. prefs()->SetBoolean(kAssistantLaunchWithMicOpen, false);
  169. EXPECT_EQ(observer()->launch_with_mic_open(), false);
  170. prefs()->SetBoolean(kAssistantLaunchWithMicOpen, true);
  171. EXPECT_EQ(observer()->launch_with_mic_open(), true);
  172. AssistantState::Get()->RemoveObserver(observer());
  173. }
  174. TEST_F(AssistantStateControllerTest, NotifyNotificationEnabled) {
  175. AssistantState::Get()->AddObserver(observer());
  176. prefs()->SetBoolean(kAssistantNotificationEnabled, false);
  177. EXPECT_EQ(observer()->notification_enabled(), false);
  178. prefs()->SetBoolean(kAssistantNotificationEnabled, true);
  179. EXPECT_EQ(observer()->notification_enabled(), true);
  180. AssistantState::Get()->RemoveObserver(observer());
  181. }
  182. TEST_F(AssistantStateControllerTest, NotifyOnboardingModeChanged) {
  183. AssistantState::Get()->AddObserver(observer());
  184. prefs()->SetString(kAssistantOnboardingMode, kAssistantOnboardingModeDefault);
  185. EXPECT_EQ(observer()->onboarding_mode(), AssistantOnboardingMode::kDefault);
  186. prefs()->SetString(kAssistantOnboardingMode,
  187. kAssistantOnboardingModeEducation);
  188. EXPECT_EQ(observer()->onboarding_mode(), AssistantOnboardingMode::kEducation);
  189. AssistantState::Get()->RemoveObserver(observer());
  190. }
  191. } // namespace ash