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
|
// Copyright 2013 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/policy/core/common/mock_configuration_policy_provider.h"
#include <memory>
#include <string>
#include <utility>
#include "base/run_loop.h"
#include "base/task/current_thread.h"
#include "build/build_config.h"
#include "components/policy/core/common/policy_bundle.h"
using testing::Invoke;
namespace policy {
MockConfigurationPolicyProvider::MockConfigurationPolicyProvider() = default;
MockConfigurationPolicyProvider::~MockConfigurationPolicyProvider() {
#if BUILDFLAG(IS_ANDROID)
ShutdownForTesting();
#endif // BUILDFLAG(IS_ANDROID)
}
void MockConfigurationPolicyProvider::UpdateChromePolicy(
const PolicyMap& policy) {
PolicyBundle bundle;
bundle.Get(PolicyNamespace(POLICY_DOMAIN_CHROME, std::string())) =
policy.Clone();
UpdatePolicy(std::move(bundle));
WaitForPoliciesUpdated(POLICY_DOMAIN_CHROME);
}
void MockConfigurationPolicyProvider::UpdateExtensionPolicy(
const PolicyMap& policy,
const std::string& extension_id) {
PolicyBundle bundle;
bundle.Get(PolicyNamespace(POLICY_DOMAIN_EXTENSIONS, extension_id)) =
policy.Clone();
UpdatePolicy(std::move(bundle));
WaitForPoliciesUpdated(POLICY_DOMAIN_EXTENSIONS);
}
void MockConfigurationPolicyProvider::SetAutoRefresh() {
EXPECT_CALL(*this, RefreshPolicies(testing::_))
.WillRepeatedly(Invoke(
this, &MockConfigurationPolicyProvider::RefreshWithSamePolicies));
}
void MockConfigurationPolicyProvider::RefreshWithSamePolicies() {
UpdatePolicy(policies().Clone());
}
void MockConfigurationPolicyProvider::OnPolicyUpdated(
const policy::PolicyNamespace& ns,
const policy::PolicyMap& previous,
const policy::PolicyMap& current) {
policy_service_->RemoveObserver(ns.domain, this);
if (ns.domain == POLICY_DOMAIN_CHROME) {
CHECK(chrome_policies_updated_callback_);
std::move(chrome_policies_updated_callback_).Run();
}
if (ns.domain == POLICY_DOMAIN_EXTENSIONS) {
CHECK(extension_policies_updated_callback_);
std::move(extension_policies_updated_callback_).Run();
}
}
void MockConfigurationPolicyProvider::WaitForPoliciesUpdated(
policy::PolicyDomain domain) {
if (!policy_service_) {
bool spin_run_loop = base::CurrentThread::IsSet();
#if BUILDFLAG(IS_IOS)
// On iOS, the UI message loop does not support RunUntilIdle().
spin_run_loop &= !base::CurrentUIThread::IsSet();
#endif // BUILDFLAG(IS_IOS)
if (spin_run_loop) {
base::RunLoop().RunUntilIdle();
}
return;
}
CHECK(!chrome_policies_updated_callback_);
CHECK(!extension_policies_updated_callback_);
base::RunLoop loop;
if (domain == POLICY_DOMAIN_CHROME) {
chrome_policies_updated_callback_ = loop.QuitClosure();
} else if (domain == POLICY_DOMAIN_EXTENSIONS) {
extension_policies_updated_callback_ = loop.QuitClosure();
} else {
NOTREACHED();
}
policy_service_->AddObserver(domain, this);
loop.Run();
}
void MockConfigurationPolicyProvider::SetupPolicyServiceForPolicyUpdates(
policy::PolicyService* policy_service) {
CHECK(!chrome_policies_updated_callback_);
CHECK(!extension_policies_updated_callback_);
policy_service_ = policy_service;
}
MockConfigurationPolicyObserver::MockConfigurationPolicyObserver() = default;
MockConfigurationPolicyObserver::~MockConfigurationPolicyObserver() = default;
} // namespace policy
|
