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
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/extensions/api/force_installed_affiliated_extension_apitest.h"
#include <optional>
#include "base/files/file_path.h"
#include "base/json/json_writer.h"
#include "base/path_service.h"
#include "chrome/browser/ash/login/test/user_auth_config.h"
#include "chrome/browser/ash/policy/affiliation/affiliation_test_helper.h"
#include "chrome/browser/ash/policy/core/browser_policy_connector_ash.h"
#include "chrome/browser/browser_process.h"
#include "chrome/common/chrome_paths.h"
#include "chrome/test/base/mixin_based_in_process_browser_test.h"
#include "chrome/test/base/ui_test_utils.h"
#include "chromeos/ash/components/dbus/session_manager/fake_session_manager_client.h"
#include "components/prefs/pref_service.h"
#include "extensions/browser/extension_registry.h"
#include "extensions/test/result_catcher.h"
namespace {
base::FilePath GetTestDataDir() {
return base::PathService::CheckedGet(chrome::DIR_TEST_DATA);
}
} // namespace
namespace extensions {
ForceInstalledAffiliatedExtensionApiTest::
ForceInstalledAffiliatedExtensionApiTest(bool is_affiliated)
: test_install_attributes_(
ash::StubInstallAttributes::CreateCloudManaged("fake-domain",
"fake-id")) {
set_exit_when_last_browser_closes(false);
set_chromeos_user_ = false;
affiliation_mixin_.set_affiliated(is_affiliated);
cryptohome_mixin_.MarkUserAsExisting(affiliation_mixin_.account_id());
cryptohome_mixin_.ApplyAuthConfig(
affiliation_mixin_.account_id(),
ash::test::UserAuthConfig::Create(ash::test::kDefaultAuthSetup));
}
ForceInstalledAffiliatedExtensionApiTest::
~ForceInstalledAffiliatedExtensionApiTest() = default;
void ForceInstalledAffiliatedExtensionApiTest::SetUpCommandLine(
base::CommandLine* command_line) {
policy::AffiliationTestHelper::AppendCommandLineSwitchesForLoginManager(
command_line);
MixinBasedExtensionApiTest::SetUpCommandLine(command_line);
}
void ForceInstalledAffiliatedExtensionApiTest::
SetUpInProcessBrowserTestFixture() {
// Initialize clients here so they are available during setup. They will be
// shutdown in ChromeBrowserMain.
ash::SessionManagerClient::InitializeFakeInMemory();
// Init the user policy provider.
policy_provider_.SetDefaultReturns(
/*is_initialization_complete_return=*/true,
/*is_first_policy_load_complete_return=*/true);
policy_provider_.SetAutoRefresh();
policy::BrowserPolicyConnector::SetPolicyProviderForTesting(
&policy_provider_);
// Set retry delay to prevent timeouts.
policy::DeviceManagementService::SetRetryDelayForTesting(0);
MixinBasedExtensionApiTest::SetUpInProcessBrowserTestFixture();
}
void ForceInstalledAffiliatedExtensionApiTest::SetUpOnMainThread() {
// Log in user that was created with
// policy::AffiliationTestHelper::PreLoginUser() in the PRE_ test.
const base::Value::List& users =
g_browser_process->local_state()->GetList("LoggedInUsers");
if (!users.empty()) {
policy::AffiliationTestHelper::LoginUser(affiliation_mixin_.account_id());
}
force_install_mixin_.InitWithMockPolicyProvider(profile(), &policy_provider_);
MixinBasedExtensionApiTest::SetUpOnMainThread();
}
const extensions::Extension*
ForceInstalledAffiliatedExtensionApiTest::ForceInstallExtension(
const std::string& extension_path,
const std::string& pem_path) {
extensions::ExtensionId extension_id;
EXPECT_TRUE(force_install_mixin_.ForceInstallFromSourceDir(
GetTestDataDir().AppendASCII(extension_path),
GetTestDataDir().AppendASCII(pem_path),
ExtensionForceInstallMixin::WaitMode::kLoad, &extension_id));
return force_install_mixin_.GetInstalledExtension(extension_id);
}
void ForceInstalledAffiliatedExtensionApiTest::TestExtension(
Browser* browser,
const GURL& page_url,
const base::Value::Dict& custom_arg_value) {
DCHECK(page_url.is_valid()) << "page_url must be valid";
std::string custom_arg;
base::JSONWriter::Write(custom_arg_value, &custom_arg);
SetCustomArg(custom_arg);
extensions::ResultCatcher catcher;
ASSERT_TRUE(ui_test_utils::NavigateToURL(browser, GURL(page_url)));
ASSERT_TRUE(catcher.GetNextResult()) << catcher.message();
}
} // namespace extensions
|
