chrome/browser/policy/assistant_policy_browsertest.cc

#include "chrome/browser/policy/policy_test_utils.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/browser.h"
#include "chromeos/services/assistant/public/cpp/assistant_prefs.h"
#include "components/policy/core/common/policy_map.h"
#include "components/policy/policy_constants.h"
#include "components/prefs/pref_service.h"

namespace policy {

class AssistantPolicyTest : public PolicyTest {};

IN_PROC_BROWSER_TEST_F(AssistantPolicyTest, AssistantContextEnabled) {
  PrefService* prefs = browser()->profile()->GetPrefs();
  EXPECT_FALSE(prefs->IsManagedPreference(
      chromeos::assistant::prefs::kAssistantContextEnabled));
  EXPECT_FALSE(
      prefs->GetBoolean(chromeos::assistant::prefs::kAssistantContextEnabled));
  prefs->SetBoolean(chromeos::assistant::prefs::kAssistantContextEnabled, true);
  EXPECT_TRUE(
      prefs->GetBoolean(chromeos::assistant::prefs::kAssistantContextEnabled));

  // Verifies that the Assistant context can be forced to always disabled.
  PolicyMap policies;
  policies.Set(key::kVoiceInteractionContextEnabled, POLICY_LEVEL_MANDATORY,
               POLICY_SCOPE_USER, POLICY_SOURCE_CLOUD,
               std::make_unique<base::Value>(false), nullptr);
  UpdateProviderPolicy(policies);
  EXPECT_TRUE(prefs->IsManagedPreference(
      chromeos::assistant::prefs::kAssistantContextEnabled));
  EXPECT_FALSE(
      prefs->GetBoolean(chromeos::assistant::prefs::kAssistantContextEnabled));
  prefs->SetBoolean(chromeos::assistant::prefs::kAssistantContextEnabled, true);
  EXPECT_FALSE(
      prefs->GetBoolean(chromeos::assistant::prefs::kAssistantContextEnabled));

  // Verifies that the Assistant context can be forced to always enabled.
  policies.Set(key::kVoiceInteractionContextEnabled, POLICY_LEVEL_MANDATORY,
               POLICY_SCOPE_USER, POLICY_SOURCE_CLOUD,
               std::make_unique<base::Value>(true), nullptr);
  UpdateProviderPolicy(policies);
  EXPECT_TRUE(prefs->IsManagedPreference(
      chromeos::assistant::prefs::kAssistantContextEnabled));
  EXPECT_TRUE(
      prefs->GetBoolean(chromeos::assistant::prefs::kAssistantContextEnabled));
  prefs->SetBoolean(chromeos::assistant::prefs::kAssistantContextEnabled,
                    false);
  EXPECT_TRUE(
      prefs->GetBoolean(chromeos::assistant::prefs::kAssistantContextEnabled));
}

IN_PROC_BROWSER_TEST_F(AssistantPolicyTest, AssistantHotwordEnabled) {
  PrefService* prefs = browser()->profile()->GetPrefs();
  EXPECT_FALSE(prefs->IsManagedPreference(
      chromeos::assistant::prefs::kAssistantHotwordEnabled));
  EXPECT_FALSE(
      prefs->GetBoolean(chromeos::assistant::prefs::kAssistantHotwordEnabled));
  prefs->SetBoolean(chromeos::assistant::prefs::kAssistantHotwordEnabled, true);
  EXPECT_TRUE(
      prefs->GetBoolean(chromeos::assistant::prefs::kAssistantHotwordEnabled));

  // Verifies that the Assistant hotword can be forced to always disabled.
  PolicyMap policies;
  policies.Set(key::kVoiceInteractionHotwordEnabled, POLICY_LEVEL_MANDATORY,
               POLICY_SCOPE_USER, POLICY_SOURCE_CLOUD,
               std::make_unique<base::Value>(false), nullptr);
  UpdateProviderPolicy(policies);
  EXPECT_TRUE(prefs->IsManagedPreference(
      chromeos::assistant::prefs::kAssistantHotwordEnabled));
  EXPECT_FALSE(
      prefs->GetBoolean(chromeos::assistant::prefs::kAssistantHotwordEnabled));
  prefs->SetBoolean(chromeos::assistant::prefs::kAssistantHotwordEnabled, true);
  EXPECT_FALSE(
      prefs->GetBoolean(chromeos::assistant::prefs::kAssistantHotwordEnabled));

  // Verifies that the Assistant hotword can be forced to always enabled.
  policies.Set(key::kVoiceInteractionHotwordEnabled, POLICY_LEVEL_MANDATORY,
               POLICY_SCOPE_USER, POLICY_SOURCE_CLOUD,
               std::make_unique<base::Value>(true), nullptr);
  UpdateProviderPolicy(policies);
  EXPECT_TRUE(prefs->IsManagedPreference(
      chromeos::assistant::prefs::kAssistantHotwordEnabled));
  EXPECT_TRUE(
      prefs->GetBoolean(chromeos::assistant::prefs::kAssistantHotwordEnabled));
  prefs->SetBoolean(chromeos::assistant::prefs::kAssistantHotwordEnabled,
                    false);
  EXPECT_TRUE(
      prefs->GetBoolean(chromeos::assistant::prefs::kAssistantHotwordEnabled));
}

}  // namespace policy