| Index: chrome/browser/policy/cloud_policy_manager_unittest.cc
|
| diff --git a/chrome/browser/policy/cloud_policy_manager_unittest.cc b/chrome/browser/policy/cloud_policy_manager_unittest.cc
|
| index b833721b63770e37b1ac3b09ff3545d301ad7065..92c1b2f4c3c61fa7c02ae7f851cf2ef359cb3413 100644
|
| --- a/chrome/browser/policy/cloud_policy_manager_unittest.cc
|
| +++ b/chrome/browser/policy/cloud_policy_manager_unittest.cc
|
| @@ -53,7 +53,7 @@ class TestHarness : public PolicyProviderTestHarness {
|
| static PolicyProviderTestHarness* CreateRecommended();
|
|
|
| private:
|
| - MockCloudPolicyStore* store_;
|
| + MockCloudPolicyStore store_;
|
|
|
| DISALLOW_COPY_AND_ASSIGN(TestHarness);
|
| };
|
| @@ -68,11 +68,9 @@ void TestHarness::SetUp() {}
|
| ConfigurationPolicyProvider* TestHarness::CreateProvider(
|
| const PolicyDefinitionList* policy_definition_list) {
|
| // Create and initialize the store.
|
| - store_ = new MockCloudPolicyStore();
|
| - store_->NotifyStoreLoaded();
|
| - ConfigurationPolicyProvider* provider =
|
| - new CloudPolicyManager(scoped_ptr<CloudPolicyStore>(store_));
|
| - Mock::VerifyAndClearExpectations(store_);
|
| + store_.NotifyStoreLoaded();
|
| + ConfigurationPolicyProvider* provider = new CloudPolicyManager(&store_);
|
| + Mock::VerifyAndClearExpectations(&store_);
|
| return provider;
|
| }
|
|
|
| @@ -80,33 +78,33 @@ void TestHarness::InstallEmptyPolicy() {}
|
|
|
| void TestHarness::InstallStringPolicy(const std::string& policy_name,
|
| const std::string& policy_value) {
|
| - store_->policy_map_.Set(policy_name, policy_level(), policy_scope(),
|
| - base::Value::CreateStringValue(policy_value));
|
| + store_.policy_map_.Set(policy_name, policy_level(), policy_scope(),
|
| + base::Value::CreateStringValue(policy_value));
|
| }
|
|
|
| void TestHarness::InstallIntegerPolicy(const std::string& policy_name,
|
| int policy_value) {
|
| - store_->policy_map_.Set(policy_name, policy_level(), policy_scope(),
|
| - base::Value::CreateIntegerValue(policy_value));
|
| + store_.policy_map_.Set(policy_name, policy_level(), policy_scope(),
|
| + base::Value::CreateIntegerValue(policy_value));
|
| }
|
|
|
| void TestHarness::InstallBooleanPolicy(const std::string& policy_name,
|
| bool policy_value) {
|
| - store_->policy_map_.Set(policy_name, policy_level(), policy_scope(),
|
| - base::Value::CreateBooleanValue(policy_value));
|
| + store_.policy_map_.Set(policy_name, policy_level(), policy_scope(),
|
| + base::Value::CreateBooleanValue(policy_value));
|
| }
|
|
|
| void TestHarness::InstallStringListPolicy(const std::string& policy_name,
|
| const base::ListValue* policy_value) {
|
| - store_->policy_map_.Set(policy_name, policy_level(), policy_scope(),
|
| - policy_value->DeepCopy());
|
| + store_.policy_map_.Set(policy_name, policy_level(), policy_scope(),
|
| + policy_value->DeepCopy());
|
| }
|
|
|
| void TestHarness::InstallDictionaryPolicy(
|
| const std::string& policy_name,
|
| const base::DictionaryValue* policy_value) {
|
| - store_->policy_map_.Set(policy_name, policy_level(), policy_scope(),
|
| - policy_value->DeepCopy());
|
| + store_.policy_map_.Set(policy_name, policy_level(), policy_scope(),
|
| + policy_value->DeepCopy());
|
| }
|
|
|
| // static
|
| @@ -128,8 +126,8 @@ INSTANTIATE_TEST_CASE_P(
|
|
|
| class TestCloudPolicyManager : public CloudPolicyManager {
|
| public:
|
| - explicit TestCloudPolicyManager(scoped_ptr<CloudPolicyStore> store)
|
| - : CloudPolicyManager(store.Pass()) {}
|
| + explicit TestCloudPolicyManager(CloudPolicyStore* store)
|
| + : CloudPolicyManager(store) {}
|
| virtual ~TestCloudPolicyManager() {}
|
|
|
| // Publish the protected members for testing.
|
| @@ -161,12 +159,10 @@ class CloudPolicyManagerTest : public testing::Test {
|
| "http://www.example.com");
|
| policy_.Build();
|
|
|
| - store_ = new MockCloudPolicyStore();
|
| - EXPECT_CALL(*store_, Load());
|
| - manager_.reset(
|
| - new TestCloudPolicyManager(scoped_ptr<CloudPolicyStore>(store_)));
|
| + EXPECT_CALL(store_, Load());
|
| + manager_.reset(new TestCloudPolicyManager(&store_));
|
| manager_->Init();
|
| - Mock::VerifyAndClearExpectations(store_);
|
| + Mock::VerifyAndClearExpectations(&store_);
|
| manager_->AddObserver(&observer_);
|
| }
|
|
|
| @@ -185,7 +181,7 @@ class CloudPolicyManagerTest : public testing::Test {
|
|
|
| // Policy infrastructure.
|
| MockConfigurationPolicyObserver observer_;
|
| - MockCloudPolicyStore* store_;
|
| + MockCloudPolicyStore store_;
|
| scoped_ptr<TestCloudPolicyManager> manager_;
|
|
|
| private:
|
| @@ -201,10 +197,10 @@ TEST_F(CloudPolicyManagerTest, InitAndShutdown) {
|
| manager_->CheckAndPublishPolicy();
|
| Mock::VerifyAndClearExpectations(&observer_);
|
|
|
| - store_->policy_map_.CopyFrom(policy_map_);
|
| - store_->policy_.reset(new em::PolicyData(policy_.policy_data()));
|
| + store_.policy_map_.CopyFrom(policy_map_);
|
| + store_.policy_.reset(new em::PolicyData(policy_.policy_data()));
|
| EXPECT_CALL(observer_, OnUpdatePolicy(manager_.get()));
|
| - store_->NotifyStoreLoaded();
|
| + store_.NotifyStoreLoaded();
|
| Mock::VerifyAndClearExpectations(&observer_);
|
| EXPECT_TRUE(expected_bundle_.Equals(manager_->policies()));
|
| EXPECT_TRUE(manager_->IsInitializationComplete());
|
| @@ -227,7 +223,7 @@ TEST_F(CloudPolicyManagerTest, InitAndShutdown) {
|
|
|
| TEST_F(CloudPolicyManagerTest, RegistrationAndFetch) {
|
| EXPECT_CALL(observer_, OnUpdatePolicy(manager_.get()));
|
| - store_->NotifyStoreLoaded();
|
| + store_.NotifyStoreLoaded();
|
| Mock::VerifyAndClearExpectations(&observer_);
|
| EXPECT_TRUE(manager_->IsInitializationComplete());
|
|
|
| @@ -238,28 +234,28 @@ TEST_F(CloudPolicyManagerTest, RegistrationAndFetch) {
|
| client->NotifyRegistrationStateChanged();
|
|
|
| client->SetPolicy(policy_.policy());
|
| - EXPECT_CALL(*store_, Store(ProtoMatches(policy_.policy())));
|
| + EXPECT_CALL(store_, Store(ProtoMatches(policy_.policy())));
|
| client->NotifyPolicyFetched();
|
| - Mock::VerifyAndClearExpectations(store_);
|
| + Mock::VerifyAndClearExpectations(&store_);
|
|
|
| - store_->policy_map_.CopyFrom(policy_map_);
|
| + store_.policy_map_.CopyFrom(policy_map_);
|
| EXPECT_CALL(observer_, OnUpdatePolicy(manager_.get()));
|
| - store_->NotifyStoreLoaded();
|
| + store_.NotifyStoreLoaded();
|
| Mock::VerifyAndClearExpectations(&observer_);
|
| EXPECT_TRUE(expected_bundle_.Equals(manager_->policies()));
|
| }
|
|
|
| TEST_F(CloudPolicyManagerTest, Update) {
|
| EXPECT_CALL(observer_, OnUpdatePolicy(manager_.get()));
|
| - store_->NotifyStoreLoaded();
|
| + store_.NotifyStoreLoaded();
|
| Mock::VerifyAndClearExpectations(&observer_);
|
| EXPECT_TRUE(manager_->IsInitializationComplete());
|
| PolicyBundle empty_bundle;
|
| EXPECT_TRUE(empty_bundle.Equals(manager_->policies()));
|
|
|
| - store_->policy_map_.CopyFrom(policy_map_);
|
| + store_.policy_map_.CopyFrom(policy_map_);
|
| EXPECT_CALL(observer_, OnUpdatePolicy(manager_.get()));
|
| - store_->NotifyStoreLoaded();
|
| + store_.NotifyStoreLoaded();
|
| Mock::VerifyAndClearExpectations(&observer_);
|
| EXPECT_TRUE(expected_bundle_.Equals(manager_->policies()));
|
| EXPECT_TRUE(manager_->IsInitializationComplete());
|
| @@ -270,7 +266,7 @@ TEST_F(CloudPolicyManagerTest, RefreshNotRegistered) {
|
| manager_->InitializeService(scoped_ptr<CloudPolicyClient>(client));
|
|
|
| EXPECT_CALL(observer_, OnUpdatePolicy(manager_.get()));
|
| - store_->NotifyStoreLoaded();
|
| + store_.NotifyStoreLoaded();
|
| Mock::VerifyAndClearExpectations(&observer_);
|
|
|
| // A refresh on a non-registered store should not block.
|
| @@ -284,10 +280,10 @@ TEST_F(CloudPolicyManagerTest, RefreshSuccessful) {
|
| manager_->InitializeService(scoped_ptr<CloudPolicyClient>(client));
|
|
|
| // Simulate a store load.
|
| - store_->policy_.reset(new em::PolicyData(policy_.policy_data()));
|
| + store_.policy_.reset(new em::PolicyData(policy_.policy_data()));
|
| EXPECT_CALL(observer_, OnUpdatePolicy(manager_.get()));
|
| EXPECT_CALL(*client, SetupRegistration(_, _));
|
| - store_->NotifyStoreLoaded();
|
| + store_.NotifyStoreLoaded();
|
| Mock::VerifyAndClearExpectations(client);
|
| Mock::VerifyAndClearExpectations(&observer_);
|
|
|
| @@ -300,24 +296,24 @@ TEST_F(CloudPolicyManagerTest, RefreshSuccessful) {
|
| manager_->RefreshPolicies();
|
| Mock::VerifyAndClearExpectations(client);
|
| Mock::VerifyAndClearExpectations(&observer_);
|
| - store_->policy_map_.CopyFrom(policy_map_);
|
| + store_.policy_map_.CopyFrom(policy_map_);
|
|
|
| // A stray reload should be suppressed until the refresh completes.
|
| EXPECT_CALL(observer_, OnUpdatePolicy(_)).Times(0);
|
| - store_->NotifyStoreLoaded();
|
| + store_.NotifyStoreLoaded();
|
| Mock::VerifyAndClearExpectations(&observer_);
|
|
|
| // Respond to the policy fetch, which should trigger a write to |store_|.
|
| EXPECT_CALL(observer_, OnUpdatePolicy(_)).Times(0);
|
| - EXPECT_CALL(*store_, Store(_));
|
| + EXPECT_CALL(store_, Store(_));
|
| client->SetPolicy(policy_.policy());
|
| client->NotifyPolicyFetched();
|
| Mock::VerifyAndClearExpectations(&observer_);
|
| - Mock::VerifyAndClearExpectations(store_);
|
| + Mock::VerifyAndClearExpectations(&store_);
|
|
|
| // The load notification from |store_| should trigger the policy update.
|
| EXPECT_CALL(observer_, OnUpdatePolicy(manager_.get()));
|
| - store_->NotifyStoreLoaded();
|
| + store_.NotifyStoreLoaded();
|
| EXPECT_TRUE(expected_bundle_.Equals(manager_->policies()));
|
| Mock::VerifyAndClearExpectations(&observer_);
|
| }
|
|
|