123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239 |
- // Copyright 2019 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 "ash/assistant/assistant_state_controller.h"
- #include <memory>
- #include <string>
- #include "ash/assistant/assistant_controller_impl.h"
- #include "ash/session/session_controller_impl.h"
- #include "ash/shell.h"
- #include "ash/test/ash_test_base.h"
- #include "chromeos/ash/services/assistant/public/cpp/assistant_prefs.h"
- #include "components/prefs/pref_service.h"
- namespace ash {
- namespace {
- using assistant::prefs::AssistantOnboardingMode;
- using assistant::prefs::ConsentStatus;
- using assistant::prefs::kAssistantConsentStatus;
- using assistant::prefs::kAssistantContextEnabled;
- using assistant::prefs::kAssistantEnabled;
- using assistant::prefs::kAssistantHotwordAlwaysOn;
- using assistant::prefs::kAssistantHotwordEnabled;
- using assistant::prefs::kAssistantLaunchWithMicOpen;
- using assistant::prefs::kAssistantNotificationEnabled;
- using assistant::prefs::kAssistantOnboardingMode;
- using assistant::prefs::kAssistantOnboardingModeDefault;
- using assistant::prefs::kAssistantOnboardingModeEducation;
- class TestAssistantStateObserver : public AssistantStateObserver {
- public:
- TestAssistantStateObserver() = default;
- TestAssistantStateObserver(const TestAssistantStateObserver&) = delete;
- TestAssistantStateObserver& operator=(const TestAssistantStateObserver&) =
- delete;
- ~TestAssistantStateObserver() override = default;
- // AssistantStateObserver:
- void OnAssistantConsentStatusChanged(int consent_status) override {
- consent_status_ = consent_status;
- }
- void OnAssistantContextEnabled(bool context_enabled) override {
- context_enabled_ = context_enabled;
- }
- void OnAssistantSettingsEnabled(bool settings_enabled) override {
- settings_enabled_ = settings_enabled;
- }
- void OnAssistantHotwordAlwaysOn(bool hotword_always_on) override {
- hotword_always_on_ = hotword_always_on;
- }
- void OnAssistantHotwordEnabled(bool hotword_enabled) override {
- hotword_enabled_ = hotword_enabled;
- }
- void OnAssistantLaunchWithMicOpen(bool launch_with_mic_open) override {
- launch_with_mic_open_ = launch_with_mic_open;
- }
- void OnAssistantNotificationEnabled(bool notification_enabled) override {
- notification_enabled_ = notification_enabled;
- }
- void OnAssistantOnboardingModeChanged(
- AssistantOnboardingMode onboarding_mode) override {
- onboarding_mode_ = onboarding_mode;
- }
- int consent_status() const { return consent_status_; }
- bool context_enabled() const { return context_enabled_; }
- bool settings_enabled() const { return settings_enabled_; }
- bool hotword_always_on() const { return hotword_always_on_; }
- bool hotword_enabled() const { return hotword_enabled_; }
- bool launch_with_mic_open() const { return launch_with_mic_open_; }
- bool notification_enabled() const { return notification_enabled_; }
- AssistantOnboardingMode onboarding_mode() const { return onboarding_mode_; }
- private:
- int consent_status_ = ConsentStatus::kUnknown;
- bool context_enabled_ = false;
- bool settings_enabled_ = false;
- bool hotword_always_on_ = false;
- bool hotword_enabled_ = false;
- bool launch_with_mic_open_ = false;
- bool notification_enabled_ = false;
- AssistantOnboardingMode onboarding_mode_ = AssistantOnboardingMode::kDefault;
- };
- class AssistantStateControllerTest : public AshTestBase {
- public:
- AssistantStateControllerTest(const AssistantStateControllerTest&) = delete;
- AssistantStateControllerTest& operator=(const AssistantStateControllerTest&) =
- delete;
- protected:
- AssistantStateControllerTest() = default;
- ~AssistantStateControllerTest() override = default;
- // AshTestBase:
- void SetUp() override {
- AshTestBase::SetUp();
- prefs_ = Shell::Get()->session_controller()->GetPrimaryUserPrefService();
- DCHECK(prefs_);
- observer_ = std::make_unique<TestAssistantStateObserver>();
- }
- PrefService* prefs() { return prefs_; }
- TestAssistantStateObserver* observer() { return observer_.get(); }
- private:
- PrefService* prefs_ = nullptr;
- std::unique_ptr<TestAssistantStateObserver> observer_;
- };
- } // namespace
- TEST_F(AssistantStateControllerTest, InitObserver) {
- prefs()->SetInteger(kAssistantConsentStatus,
- ConsentStatus::kActivityControlAccepted);
- prefs()->SetBoolean(kAssistantContextEnabled, true);
- prefs()->SetBoolean(kAssistantEnabled, true);
- prefs()->SetBoolean(kAssistantHotwordAlwaysOn, true);
- prefs()->SetBoolean(kAssistantHotwordEnabled, true);
- prefs()->SetBoolean(kAssistantLaunchWithMicOpen, true);
- prefs()->SetBoolean(kAssistantNotificationEnabled, true);
- prefs()->SetString(kAssistantOnboardingMode, kAssistantOnboardingModeDefault);
- // The observer class should get an instant notification about the current
- // pref value.
- AssistantState::Get()->AddObserver(observer());
- EXPECT_EQ(observer()->consent_status(),
- ConsentStatus::kActivityControlAccepted);
- EXPECT_EQ(observer()->context_enabled(), true);
- EXPECT_EQ(observer()->settings_enabled(), true);
- EXPECT_EQ(observer()->hotword_always_on(), true);
- EXPECT_EQ(observer()->hotword_enabled(), true);
- EXPECT_EQ(observer()->launch_with_mic_open(), true);
- EXPECT_EQ(observer()->notification_enabled(), true);
- EXPECT_EQ(observer()->onboarding_mode(), AssistantOnboardingMode::kDefault);
- AssistantState::Get()->RemoveObserver(observer());
- }
- TEST_F(AssistantStateControllerTest, NotifyConsentStatus) {
- AssistantState::Get()->AddObserver(observer());
- prefs()->SetInteger(kAssistantConsentStatus, ConsentStatus::kUnauthorized);
- EXPECT_EQ(observer()->consent_status(), ConsentStatus::kUnauthorized);
- prefs()->SetInteger(kAssistantConsentStatus,
- ConsentStatus::kActivityControlAccepted);
- EXPECT_EQ(observer()->consent_status(),
- ConsentStatus::kActivityControlAccepted);
- AssistantState::Get()->RemoveObserver(observer());
- }
- TEST_F(AssistantStateControllerTest, NotifyContextEnabled) {
- AssistantState::Get()->AddObserver(observer());
- prefs()->SetBoolean(kAssistantContextEnabled, false);
- EXPECT_EQ(observer()->context_enabled(), false);
- prefs()->SetBoolean(kAssistantContextEnabled, true);
- EXPECT_EQ(observer()->context_enabled(), true);
- AssistantState::Get()->RemoveObserver(observer());
- }
- TEST_F(AssistantStateControllerTest, NotifySettingsEnabled) {
- AssistantState::Get()->AddObserver(observer());
- prefs()->SetBoolean(kAssistantEnabled, false);
- EXPECT_EQ(observer()->settings_enabled(), false);
- prefs()->SetBoolean(kAssistantEnabled, true);
- EXPECT_EQ(observer()->settings_enabled(), true);
- AssistantState::Get()->RemoveObserver(observer());
- }
- TEST_F(AssistantStateControllerTest, NotifyHotwordAlwaysOn) {
- AssistantState::Get()->AddObserver(observer());
- prefs()->SetBoolean(kAssistantHotwordAlwaysOn, false);
- EXPECT_EQ(observer()->hotword_always_on(), false);
- prefs()->SetBoolean(kAssistantHotwordAlwaysOn, true);
- EXPECT_EQ(observer()->hotword_always_on(), true);
- AssistantState::Get()->RemoveObserver(observer());
- }
- TEST_F(AssistantStateControllerTest, NotifyHotwordEnabled) {
- AssistantState::Get()->AddObserver(observer());
- prefs()->SetBoolean(kAssistantHotwordEnabled, false);
- EXPECT_EQ(observer()->hotword_enabled(), false);
- prefs()->SetBoolean(kAssistantHotwordEnabled, true);
- EXPECT_EQ(observer()->hotword_enabled(), true);
- AssistantState::Get()->RemoveObserver(observer());
- }
- TEST_F(AssistantStateControllerTest, NotifyLaunchWithMicOpen) {
- AssistantState::Get()->AddObserver(observer());
- prefs()->SetBoolean(kAssistantLaunchWithMicOpen, false);
- EXPECT_EQ(observer()->launch_with_mic_open(), false);
- prefs()->SetBoolean(kAssistantLaunchWithMicOpen, true);
- EXPECT_EQ(observer()->launch_with_mic_open(), true);
- AssistantState::Get()->RemoveObserver(observer());
- }
- TEST_F(AssistantStateControllerTest, NotifyNotificationEnabled) {
- AssistantState::Get()->AddObserver(observer());
- prefs()->SetBoolean(kAssistantNotificationEnabled, false);
- EXPECT_EQ(observer()->notification_enabled(), false);
- prefs()->SetBoolean(kAssistantNotificationEnabled, true);
- EXPECT_EQ(observer()->notification_enabled(), true);
- AssistantState::Get()->RemoveObserver(observer());
- }
- TEST_F(AssistantStateControllerTest, NotifyOnboardingModeChanged) {
- AssistantState::Get()->AddObserver(observer());
- prefs()->SetString(kAssistantOnboardingMode, kAssistantOnboardingModeDefault);
- EXPECT_EQ(observer()->onboarding_mode(), AssistantOnboardingMode::kDefault);
- prefs()->SetString(kAssistantOnboardingMode,
- kAssistantOnboardingModeEducation);
- EXPECT_EQ(observer()->onboarding_mode(), AssistantOnboardingMode::kEducation);
- AssistantState::Get()->RemoveObserver(observer());
- }
- } // namespace ash
|