Index: chrome/browser/policy/user_cloud_policy_store_unittest.cc |
diff --git a/chrome/browser/policy/user_cloud_policy_store_unittest.cc b/chrome/browser/policy/user_cloud_policy_store_unittest.cc |
deleted file mode 100644 |
index 6391d71521edf1283af4e3f0c5bb7e2848576d47..0000000000000000000000000000000000000000 |
--- a/chrome/browser/policy/user_cloud_policy_store_unittest.cc |
+++ /dev/null |
@@ -1,319 +0,0 @@ |
-// Copyright (c) 2012 The Chromium Authors. All rights reserved. |
-// Use of this source code is governed by a BSD-style license that can be |
-// found in the LICENSE file. |
- |
-#include "chrome/browser/policy/user_cloud_policy_store.h" |
- |
-#include "base/file_util.h" |
-#include "base/files/scoped_temp_dir.h" |
-#include "base/message_loop.h" |
-#include "base/run_loop.h" |
-#include "chrome/browser/policy/mock_cloud_policy_store.h" |
-#include "chrome/browser/policy/policy_builder.h" |
-#include "chrome/browser/signin/signin_manager.h" |
-#include "chrome/browser/signin/signin_manager_factory.h" |
-#include "chrome/browser/signin/signin_manager_fake.h" |
-#include "chrome/test/base/testing_profile.h" |
-#include "content/public/test/test_browser_thread.h" |
-#include "policy/policy_constants.h" |
-#include "testing/gmock/include/gmock/gmock.h" |
-#include "testing/gtest/include/gtest/gtest.h" |
- |
-using testing::AllOf; |
-using testing::Eq; |
-using testing::Property; |
- |
-namespace policy { |
- |
-namespace { |
- |
-void RunUntilIdle() { |
- base::RunLoop run_loop; |
- run_loop.RunUntilIdle(); |
-} |
- |
-class UserCloudPolicyStoreTest : public testing::Test { |
- public: |
- UserCloudPolicyStoreTest() |
- : loop_(MessageLoop::TYPE_UI), |
- ui_thread_(content::BrowserThread::UI, &loop_), |
- file_thread_(content::BrowserThread::FILE, &loop_), |
- profile_(new TestingProfile()) {} |
- |
- virtual void SetUp() OVERRIDE { |
- ASSERT_TRUE(tmp_dir_.CreateUniqueTempDir()); |
- SigninManager* signin = static_cast<SigninManager*>( |
- SigninManagerFactory::GetInstance()->SetTestingFactoryAndUse( |
- profile_.get(), FakeSigninManager::Build)); |
- signin->SetAuthenticatedUsername(PolicyBuilder::kFakeUsername); |
- store_.reset(new UserCloudPolicyStore(profile_.get(), policy_file())); |
- store_->AddObserver(&observer_); |
- |
- policy_.payload().mutable_showhomebutton()->set_value(true); |
- policy_.payload().mutable_syncdisabled()->set_value(true); |
- policy_.Build(); |
- } |
- |
- virtual void TearDown() OVERRIDE { |
- store_->RemoveObserver(&observer_); |
- store_.reset(); |
- RunUntilIdle(); |
- } |
- |
- base::FilePath policy_file() { |
- return tmp_dir_.path().AppendASCII("policy"); |
- } |
- |
- // Verifies that store_->policy_map() has the appropriate entries. |
- void VerifyPolicyMap(CloudPolicyStore* store) { |
- EXPECT_EQ(2U, store->policy_map().size()); |
- const PolicyMap::Entry* entry = |
- store->policy_map().Get(key::kShowHomeButton); |
- ASSERT_TRUE(entry); |
- EXPECT_TRUE(base::FundamentalValue(true).Equals(entry->value)); |
- ASSERT_TRUE(store->policy_map().Get(key::kSyncDisabled)); |
- } |
- |
- // Install an expectation on |observer_| for an error code. |
- void ExpectError(CloudPolicyStore* store, CloudPolicyStore::Status error) { |
- EXPECT_CALL(observer_, |
- OnStoreError(AllOf(Eq(store), |
- Property(&CloudPolicyStore::status, |
- Eq(error))))); |
- } |
- |
- UserPolicyBuilder policy_; |
- MockCloudPolicyStoreObserver observer_; |
- scoped_ptr<UserCloudPolicyStore> store_; |
- |
- // CloudPolicyValidator() requires a FILE thread so declare one here. Both |
- // |ui_thread_| and |file_thread_| share the same MessageLoop |loop_| so |
- // callers can use RunLoop to manage both virtual threads. |
- MessageLoop loop_; |
- content::TestBrowserThread ui_thread_; |
- content::TestBrowserThread file_thread_; |
- |
- scoped_ptr<TestingProfile> profile_; |
- base::ScopedTempDir tmp_dir_; |
- |
- DISALLOW_COPY_AND_ASSIGN(UserCloudPolicyStoreTest); |
-}; |
- |
-TEST_F(UserCloudPolicyStoreTest, LoadWithNoFile) { |
- EXPECT_FALSE(store_->policy()); |
- EXPECT_TRUE(store_->policy_map().empty()); |
- |
- EXPECT_CALL(observer_, OnStoreLoaded(store_.get())); |
- store_->Load(); |
- RunUntilIdle(); |
- |
- EXPECT_FALSE(store_->policy()); |
- EXPECT_TRUE(store_->policy_map().empty()); |
-} |
- |
-TEST_F(UserCloudPolicyStoreTest, LoadWithInvalidFile) { |
- EXPECT_FALSE(store_->policy()); |
- EXPECT_TRUE(store_->policy_map().empty()); |
- |
- // Create a bogus file. |
- ASSERT_TRUE(file_util::CreateDirectory(policy_file().DirName())); |
- std::string bogus_data = "bogus_data"; |
- int size = bogus_data.size(); |
- ASSERT_EQ(size, file_util::WriteFile(policy_file(), |
- bogus_data.c_str(), |
- bogus_data.size())); |
- |
- ExpectError(store_.get(), CloudPolicyStore::STATUS_LOAD_ERROR); |
- store_->Load(); |
- RunUntilIdle(); |
- |
- EXPECT_FALSE(store_->policy()); |
- EXPECT_TRUE(store_->policy_map().empty()); |
-} |
- |
-TEST_F(UserCloudPolicyStoreTest, LoadImmediatelyWithNoFile) { |
- EXPECT_FALSE(store_->policy()); |
- EXPECT_TRUE(store_->policy_map().empty()); |
- |
- EXPECT_CALL(observer_, OnStoreLoaded(store_.get())); |
- store_->LoadImmediately(); // Should load without running the message loop. |
- |
- EXPECT_FALSE(store_->policy()); |
- EXPECT_TRUE(store_->policy_map().empty()); |
-} |
- |
-TEST_F(UserCloudPolicyStoreTest, LoadImmediatelyWithInvalidFile) { |
- EXPECT_FALSE(store_->policy()); |
- EXPECT_TRUE(store_->policy_map().empty()); |
- |
- // Create a bogus file. |
- ASSERT_TRUE(file_util::CreateDirectory(policy_file().DirName())); |
- std::string bogus_data = "bogus_data"; |
- int size = bogus_data.size(); |
- ASSERT_EQ(size, file_util::WriteFile(policy_file(), |
- bogus_data.c_str(), |
- bogus_data.size())); |
- |
- ExpectError(store_.get(), CloudPolicyStore::STATUS_LOAD_ERROR); |
- store_->LoadImmediately(); // Should load without running the message loop. |
- |
- EXPECT_FALSE(store_->policy()); |
- EXPECT_TRUE(store_->policy_map().empty()); |
-} |
- |
-TEST_F(UserCloudPolicyStoreTest, Store) { |
- EXPECT_FALSE(store_->policy()); |
- EXPECT_TRUE(store_->policy_map().empty()); |
- |
- // Store a simple policy and make sure it ends up as the currently active |
- // policy. |
- EXPECT_CALL(observer_, OnStoreLoaded(store_.get())); |
- store_->Store(policy_.policy()); |
- RunUntilIdle(); |
- |
- // Policy should be decoded and stored. |
- ASSERT_TRUE(store_->policy()); |
- EXPECT_EQ(policy_.policy_data().SerializeAsString(), |
- store_->policy()->SerializeAsString()); |
- VerifyPolicyMap(store_.get()); |
- EXPECT_EQ(CloudPolicyStore::STATUS_OK, store_->status()); |
-} |
- |
-TEST_F(UserCloudPolicyStoreTest, StoreThenClear) { |
- EXPECT_FALSE(store_->policy()); |
- EXPECT_TRUE(store_->policy_map().empty()); |
- |
- // Store a simple policy and make sure the file exists. |
- // policy. |
- EXPECT_CALL(observer_, OnStoreLoaded(store_.get())); |
- store_->Store(policy_.policy()); |
- RunUntilIdle(); |
- |
- EXPECT_TRUE(store_->policy()); |
- EXPECT_FALSE(store_->policy_map().empty()); |
- |
- // Policy file should exist. |
- ASSERT_TRUE(file_util::PathExists(policy_file())); |
- |
- EXPECT_CALL(observer_, OnStoreLoaded(store_.get())); |
- store_->Clear(); |
- RunUntilIdle(); |
- |
- // Policy file should not exist. |
- ASSERT_TRUE(!file_util::PathExists(policy_file())); |
- |
- // Policy should be gone. |
- EXPECT_FALSE(store_->policy()); |
- EXPECT_TRUE(store_->policy_map().empty()); |
- EXPECT_EQ(CloudPolicyStore::STATUS_OK, store_->status()); |
-} |
- |
-TEST_F(UserCloudPolicyStoreTest, StoreTwoTimes) { |
- EXPECT_FALSE(store_->policy()); |
- EXPECT_TRUE(store_->policy_map().empty()); |
- |
- // Store a simple policy then store a second policy before the first one |
- // finishes validating, and make sure the second policy ends up as the active |
- // policy. |
- EXPECT_CALL(observer_, OnStoreLoaded(store_.get())).Times(2); |
- |
- UserPolicyBuilder first_policy; |
- first_policy.payload().mutable_showhomebutton()->set_value(false); |
- first_policy.Build(); |
- store_->Store(first_policy.policy()); |
- RunUntilIdle(); |
- |
- store_->Store(policy_.policy()); |
- RunUntilIdle(); |
- |
- // Policy should be decoded and stored. |
- ASSERT_TRUE(store_->policy()); |
- EXPECT_EQ(policy_.policy_data().SerializeAsString(), |
- store_->policy()->SerializeAsString()); |
- VerifyPolicyMap(store_.get()); |
- EXPECT_EQ(CloudPolicyStore::STATUS_OK, store_->status()); |
-} |
- |
-TEST_F(UserCloudPolicyStoreTest, StoreThenLoad) { |
- // Store a simple policy and make sure it can be read back in. |
- // policy. |
- EXPECT_CALL(observer_, OnStoreLoaded(store_.get())); |
- store_->Store(policy_.policy()); |
- RunUntilIdle(); |
- |
- // Now, make sure the policy can be read back in from a second store. |
- scoped_ptr<UserCloudPolicyStore> store2( |
- new UserCloudPolicyStore(profile_.get(), policy_file())); |
- store2->AddObserver(&observer_); |
- EXPECT_CALL(observer_, OnStoreLoaded(store2.get())); |
- store2->Load(); |
- RunUntilIdle(); |
- |
- ASSERT_TRUE(store2->policy()); |
- EXPECT_EQ(policy_.policy_data().SerializeAsString(), |
- store2->policy()->SerializeAsString()); |
- VerifyPolicyMap(store2.get()); |
- EXPECT_EQ(CloudPolicyStore::STATUS_OK, store2->status()); |
- store2->RemoveObserver(&observer_); |
-} |
- |
-TEST_F(UserCloudPolicyStoreTest, StoreThenLoadImmediately) { |
- // Store a simple policy and make sure it can be read back in. |
- // policy. |
- EXPECT_CALL(observer_, OnStoreLoaded(store_.get())); |
- store_->Store(policy_.policy()); |
- RunUntilIdle(); |
- |
- // Now, make sure the policy can be read back in from a second store. |
- scoped_ptr<UserCloudPolicyStore> store2( |
- new UserCloudPolicyStore(profile_.get(), policy_file())); |
- store2->AddObserver(&observer_); |
- EXPECT_CALL(observer_, OnStoreLoaded(store2.get())); |
- store2->LoadImmediately(); // Should load without running the message loop. |
- |
- ASSERT_TRUE(store2->policy()); |
- EXPECT_EQ(policy_.policy_data().SerializeAsString(), |
- store2->policy()->SerializeAsString()); |
- VerifyPolicyMap(store2.get()); |
- EXPECT_EQ(CloudPolicyStore::STATUS_OK, store2->status()); |
- store2->RemoveObserver(&observer_); |
-} |
- |
-TEST_F(UserCloudPolicyStoreTest, StoreValidationError) { |
- // Create an invalid policy (no policy type). |
- policy_.policy_data().clear_policy_type(); |
- policy_.Build(); |
- |
- // Store policy. |
- ExpectError(store_.get(), CloudPolicyStore::STATUS_VALIDATION_ERROR); |
- store_->Store(policy_.policy()); |
- RunUntilIdle(); |
- ASSERT_FALSE(store_->policy()); |
-} |
- |
-TEST_F(UserCloudPolicyStoreTest, LoadValidationError) { |
- // Force a validation error by changing the username after policy is stored. |
- EXPECT_CALL(observer_, OnStoreLoaded(store_.get())); |
- store_->Store(policy_.policy()); |
- RunUntilIdle(); |
- |
- // Sign out, and sign back in as a different user, and try to load the profile |
- // data (should fail due to mismatched username). |
- SigninManagerFactory::GetForProfile(profile_.get())->SignOut(); |
- SigninManagerFactory::GetForProfile(profile_.get())->SetAuthenticatedUsername( |
- "foobar@foobar.com"); |
- |
- scoped_ptr<UserCloudPolicyStore> store2( |
- new UserCloudPolicyStore(profile_.get(), policy_file())); |
- store2->AddObserver(&observer_); |
- ExpectError(store2.get(), CloudPolicyStore::STATUS_VALIDATION_ERROR); |
- store2->Load(); |
- RunUntilIdle(); |
- |
- ASSERT_FALSE(store2->policy()); |
- store2->RemoveObserver(&observer_); |
-} |
- |
-} // namespace |
- |
-} // namespace policy |