Chromium Code Reviews| 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_; |