Index: chrome/browser/policy/network_configuration_updater_unittest.cc |
diff --git a/chrome/browser/policy/network_configuration_updater_unittest.cc b/chrome/browser/policy/network_configuration_updater_unittest.cc |
index bf6d71bab965d220f5a6cfeaf5266e387a05cac9..d03ee8bba15102e5679763fa3960be4326d8aee3 100644 |
--- a/chrome/browser/policy/network_configuration_updater_unittest.cc |
+++ b/chrome/browser/policy/network_configuration_updater_unittest.cc |
@@ -23,8 +23,6 @@ class NetworkConfigurationUpdaterTest |
: public testing::TestWithParam<const char*> { |
protected: |
virtual void SetUp() OVERRIDE { |
- EXPECT_CALL(provider_, ProvideInternal(_)) |
- .WillRepeatedly(CopyPolicyMap(&policy_)); |
EXPECT_CALL(network_library_, LoadOncNetworks(_, "", _, _)) |
.WillRepeatedly(Return(true)); |
} |
@@ -40,13 +38,14 @@ class NetworkConfigurationUpdaterTest |
} |
chromeos::MockNetworkLibrary network_library_; |
- PolicyMap policy_; |
MockConfigurationPolicyProvider provider_; |
}; |
TEST_P(NetworkConfigurationUpdaterTest, InitialUpdate) { |
- policy_.Set(GetParam(), POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- Value::CreateStringValue(kFakeONC)); |
+ PolicyMap policy; |
+ policy.Set(GetParam(), POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
+ Value::CreateStringValue(kFakeONC)); |
+ provider_.UpdateChromePolicy(policy); |
EXPECT_CALL(network_library_, |
LoadOncNetworks(kFakeONC, "", NameToONCSource(GetParam()), _)) |
@@ -63,16 +62,17 @@ TEST_P(NetworkConfigurationUpdaterTest, PolicyChange) { |
EXPECT_CALL(network_library_, |
LoadOncNetworks(kFakeONC, "", NameToONCSource(GetParam()), _)) |
.WillOnce(Return(true)); |
- policy_.Set(GetParam(), POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
- Value::CreateStringValue(kFakeONC)); |
- provider_.NotifyPolicyUpdated(); |
+ PolicyMap policy; |
+ policy.Set(GetParam(), POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
+ Value::CreateStringValue(kFakeONC)); |
+ provider_.UpdateChromePolicy(policy); |
Mock::VerifyAndClearExpectations(&network_library_); |
// No update if the set the same value again. |
EXPECT_CALL(network_library_, |
LoadOncNetworks(kFakeONC, "", NameToONCSource(GetParam()), _)) |
.Times(0); |
- provider_.NotifyPolicyUpdated(); |
+ provider_.UpdateChromePolicy(policy); |
Mock::VerifyAndClearExpectations(&network_library_); |
// Another update is expected if the policy goes away. |
@@ -80,8 +80,8 @@ TEST_P(NetworkConfigurationUpdaterTest, PolicyChange) { |
LoadOncNetworks(NetworkConfigurationUpdater::kEmptyConfiguration, |
"", NameToONCSource(GetParam()), _)) |
.WillOnce(Return(true)); |
- policy_.Erase(GetParam()); |
- provider_.NotifyPolicyUpdated(); |
+ policy.Erase(GetParam()); |
+ provider_.UpdateChromePolicy(policy); |
Mock::VerifyAndClearExpectations(&network_library_); |
} |