Index: chrome/browser/policy/user_policy_cache_unittest.cc |
diff --git a/chrome/browser/policy/user_policy_cache_unittest.cc b/chrome/browser/policy/user_policy_cache_unittest.cc |
index 7f327ea2ccccb4a33248bcc412b61601ec48606d..3a9e449ccbe4ff7f8aa37e301fd90a54d5f36209 100644 |
--- a/chrome/browser/policy/user_policy_cache_unittest.cc |
+++ b/chrome/browser/policy/user_policy_cache_unittest.cc |
@@ -32,13 +32,13 @@ void DecodePolicy(const em::CloudPolicySettings& policy, |
Value* DecodeIntegerValue(google::protobuf::int64 value); |
ListValue* DecodeStringList(const em::StringList& string_list); |
-class MockConfigurationPolicyProviderObserver |
- : public ConfigurationPolicyProvider::Observer { |
+class MockCloudPolicyCacheBaseObserver |
+ : public CloudPolicyCacheBase::Observer { |
public: |
- MockConfigurationPolicyProviderObserver() {} |
- virtual ~MockConfigurationPolicyProviderObserver() {} |
- MOCK_METHOD0(OnUpdatePolicy, void()); |
- void OnProviderGoingAway() {} |
+ MockCloudPolicyCacheBaseObserver() {} |
+ virtual ~MockCloudPolicyCacheBaseObserver() {} |
+ MOCK_METHOD1(OnCacheUpdate, void(CloudPolicyCacheBase*)); |
+ void OnCacheGoingAway(CloudPolicyCacheBase*) {} |
}; |
// Tests the device management policy cache. |
@@ -103,14 +103,16 @@ class UserPolicyCacheTest : public testing::Test { |
em::PolicyFetchResponse* policy_response, |
bool expect_changed_policy) { |
scoped_ptr<em::PolicyFetchResponse> policy(policy_response); |
- ConfigurationPolicyObserverRegistrar registrar; |
- registrar.Init(cache->GetManagedPolicyProvider(), &observer); |
+ cache->AddObserver(&observer); |
+ |
if (expect_changed_policy) |
- EXPECT_CALL(observer, OnUpdatePolicy()).Times(1); |
+ EXPECT_CALL(observer, OnCacheUpdate(testing::_)).Times(1); |
Mattias Nissler (ping if slow)
2011/06/09 14:36:26
using testing::_ at the top of the file.
sfeuz
2011/06/13 06:53:53
Done.
|
else |
- EXPECT_CALL(observer, OnUpdatePolicy()).Times(0); |
+ EXPECT_CALL(observer, OnCacheUpdate(testing::_)).Times(0); |
cache->SetPolicy(*policy); |
testing::Mock::VerifyAndClearExpectations(&observer); |
+ |
+ cache->RemoveObserver(&observer); |
} |
FilePath test_file() { |
@@ -126,7 +128,7 @@ class UserPolicyCacheTest : public testing::Test { |
} |
MessageLoop loop_; |
- MockConfigurationPolicyProviderObserver observer; |
+ MockCloudPolicyCacheBaseObserver observer; |
private: |
ScopedTempDir temp_dir_; |