Index: chrome/browser/ui/webui/policy_ui_browsertest.cc |
diff --git a/chrome/browser/ui/webui/policy_ui_browsertest.cc b/chrome/browser/ui/webui/policy_ui_browsertest.cc |
index 161a734a735981668475392a7b3ad5eda7b84366..6aa0568f46c77eb4d9677eee3608aeb52bf8e861 100644 |
--- a/chrome/browser/ui/webui/policy_ui_browsertest.cc |
+++ b/chrome/browser/ui/webui/policy_ui_browsertest.cc |
@@ -41,6 +41,7 @@ namespace { |
std::vector<std::string> PopulateExpectedPolicy( |
const std::string& name, |
const std::string& value, |
+ const std::string& source, |
const policy::PolicyMap::Entry* metadata, |
bool unknown) { |
std::vector<std::string> expected_policy; |
@@ -62,6 +63,8 @@ std::vector<std::string> PopulateExpectedPolicy( |
} else { |
expected_policy.push_back(std::string()); |
} |
+ // Populate expected source name. |
+ expected_policy.push_back(source); |
// Populate expected policy name. |
expected_policy.push_back(name); |
@@ -118,9 +121,9 @@ void PolicyUITest::SetUpInProcessBrowserTestFixture() { |
} |
void PolicyUITest::UpdateProviderPolicy(const policy::PolicyMap& policy) { |
- provider_.UpdateChromePolicy(policy); |
- base::RunLoop loop; |
- loop.RunUntilIdle(); |
+ provider_.UpdateChromePolicy(policy); |
+ base::RunLoop loop; |
+ loop.RunUntilIdle(); |
} |
void PolicyUITest::VerifyPolicies( |
@@ -187,7 +190,8 @@ IN_PROC_BROWSER_TEST_F(PolicyUITest, SendPolicyNames) { |
for (policy::Schema::Iterator it = chrome_schema.GetPropertiesIterator(); |
!it.IsAtEnd(); it.Advance()) { |
expected_policies.push_back( |
- PopulateExpectedPolicy(it.key(), std::string(), NULL, false)); |
+ PopulateExpectedPolicy( |
+ it.key(), std::string(), std::string(), NULL, false)); |
bartfab (slow)
2015/09/14 14:42:26
Nit: s/NULL/nullptr/
fhorschig
2015/09/16 13:52:04
Done.
|
} |
// Retrieve the contents of the policy table from the UI and verify that it |
@@ -234,11 +238,12 @@ IN_PROC_BROWSER_TEST_F(PolicyUITest, SendPolicyValues) { |
expected_values[policy::key::kShowHomeButton] = "true"; |
// Set the value of a policy that does not exist. |
const std::string kUnknownPolicy = "NoSuchThing"; |
- values.Set(kUnknownPolicy, |
- policy::POLICY_LEVEL_MANDATORY, |
- policy::POLICY_SCOPE_USER, |
- new base::FundamentalValue(true), |
- NULL); |
+ values.SetWithSource(kUnknownPolicy, |
+ policy::POLICY_LEVEL_MANDATORY, |
+ policy::POLICY_SCOPE_USER, |
+ new base::FundamentalValue(true), |
+ NULL, |
bartfab (slow)
2015/09/14 14:42:26
Nit: s/NULL/nullptr/
fhorschig
2015/09/16 13:52:04
Done.
|
+ policy::POLICY_SOURCE_CLOUD); |
expected_values[kUnknownPolicy] = "true"; |
UpdateProviderPolicy(values); |
@@ -261,12 +266,14 @@ IN_PROC_BROWSER_TEST_F(PolicyUITest, SendPolicyValues) { |
expected_policies.insert( |
metadata ? expected_policies.begin() + first_unset_position++ : |
expected_policies.end(), |
- PopulateExpectedPolicy(props.key(), value, metadata, false)); |
+ PopulateExpectedPolicy( |
+ props.key(), value, std::string(), metadata, false)); |
} |
expected_policies.insert( |
expected_policies.begin() + first_unset_position++, |
PopulateExpectedPolicy(kUnknownPolicy, |
expected_values[kUnknownPolicy], |
+ "Cloud", |
values.Get(kUnknownPolicy), |
true)); |
@@ -318,11 +325,12 @@ IN_PROC_BROWSER_TEST_F(PolicyUITest, ExtensionLoadAndSendPolicy) { |
for (policy::Schema::Iterator it = chrome_schema.GetPropertiesIterator(); |
!it.IsAtEnd(); it.Advance()) { |
expected_policies.push_back( |
- PopulateExpectedPolicy(it.key(), std::string(), NULL, false)); |
+ PopulateExpectedPolicy( |
+ it.key(), std::string(), std::string(), NULL, false)); |
} |
// Add newly added policy to expected policy list. |
expected_policies.push_back(PopulateExpectedPolicy( |
- newly_added_policy_name, std::string(), NULL, false)); |
+ newly_added_policy_name, std::string(), std::string(), NULL, false)); |
// Verify if policy UI includes policy that extension have. |
VerifyPolicies(expected_policies); |