1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
|
// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "ash/constants/ash_features.h"
#include "ash/constants/ash_pref_names.h"
#include "chrome/browser/ash/policy/handlers/configuration_policy_handler_ash.h"
#include "chrome/browser/ash/system_web_apps/apps/personalization_app/personalization_app_utils.h"
#include "chrome/browser/policy/policy_test_utils.h"
#include "chrome/browser/policy/profile_policy_connector.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/browser.h"
#include "components/policy/policy_constants.h"
#include "components/prefs/pref_service.h"
#include "content/public/test/browser_test.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace policy {
class GenAIWallpaperPolicyTest : public PolicyTest {};
IN_PROC_BROWSER_TEST_F(GenAIWallpaperPolicyTest,
EnableFeatureIfGenAIWallpaperPolicyUnset) {
Profile* profile = browser()->profile();
profile->GetProfilePolicyConnector()->OverrideIsManagedForTesting(true);
UpdateProviderPolicy(PolicyMap());
EXPECT_TRUE(
ash::personalization_app::IsManagedSeaPenWallpaperEnabled(profile));
EXPECT_FALSE(
ash::personalization_app::IsManagedSeaPenWallpaperFeedbackEnabled(
profile));
}
IN_PROC_BROWSER_TEST_F(GenAIWallpaperPolicyTest,
EnableFeatureIfGenAIWallpaperPolicyEnabled) {
Profile* profile = browser()->profile();
PolicyMap policies;
profile->GetProfilePolicyConnector()->OverrideIsManagedForTesting(true);
policies.Set(key::kGenAIWallpaperSettings, POLICY_LEVEL_MANDATORY,
POLICY_SCOPE_USER, POLICY_SOURCE_CLOUD, base::Value(0), nullptr);
UpdateProviderPolicy(policies);
EXPECT_TRUE(profile->GetPrefs()->IsManagedPreference(
ash::prefs::kGenAIWallpaperSettings));
EXPECT_EQ(
profile->GetPrefs()->GetInteger(ash::prefs::kGenAIWallpaperSettings),
static_cast<int>(
ash::personalization_app::ManagedSeaPenSettings::kAllowed));
EXPECT_TRUE(
ash::personalization_app::IsManagedSeaPenWallpaperEnabled(profile));
EXPECT_TRUE(ash::personalization_app::IsManagedSeaPenWallpaperFeedbackEnabled(
profile));
}
IN_PROC_BROWSER_TEST_F(
GenAIWallpaperPolicyTest,
EnableFeatureIfGenAIWallpaperPolicyEnabledWithoutLogging) {
Profile* profile = browser()->profile();
PolicyMap policies;
profile->GetProfilePolicyConnector()->OverrideIsManagedForTesting(true);
policies.Set(key::kGenAIWallpaperSettings, POLICY_LEVEL_MANDATORY,
POLICY_SCOPE_USER, POLICY_SOURCE_CLOUD, base::Value(1), nullptr);
UpdateProviderPolicy(policies);
EXPECT_TRUE(profile->GetPrefs()->IsManagedPreference(
ash::prefs::kGenAIWallpaperSettings));
EXPECT_EQ(
profile->GetPrefs()->GetInteger(ash::prefs::kGenAIWallpaperSettings),
static_cast<int>(ash::personalization_app::ManagedSeaPenSettings::
kAllowedWithoutLogging));
EXPECT_TRUE(
ash::personalization_app::IsManagedSeaPenWallpaperEnabled(profile));
EXPECT_FALSE(
ash::personalization_app::IsManagedSeaPenWallpaperFeedbackEnabled(
profile));
}
IN_PROC_BROWSER_TEST_F(GenAIWallpaperPolicyTest,
DisableFeatureIfGenAIWallpaperPolicyDisabled) {
Profile* profile = browser()->profile();
PolicyMap policies;
profile->GetProfilePolicyConnector()->OverrideIsManagedForTesting(true);
policies.Set(key::kGenAIWallpaperSettings, POLICY_LEVEL_MANDATORY,
POLICY_SCOPE_USER, POLICY_SOURCE_CLOUD, base::Value(2), nullptr);
UpdateProviderPolicy(policies);
EXPECT_TRUE(profile->GetPrefs()->IsManagedPreference(
ash::prefs::kGenAIWallpaperSettings));
EXPECT_EQ(
profile->GetPrefs()->GetInteger(ash::prefs::kGenAIWallpaperSettings),
static_cast<int>(
ash::personalization_app::ManagedSeaPenSettings::kDisabled));
EXPECT_FALSE(
ash::personalization_app::IsManagedSeaPenWallpaperEnabled(profile));
EXPECT_FALSE(
ash::personalization_app::IsManagedSeaPenWallpaperFeedbackEnabled(
profile));
}
} // namespace policy
|
