| // 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/policy_map.h" |
| |
| #include "base/callback.h" |
| #include "base/memory/weak_ptr.h" |
| #include "chrome/browser/policy/external_data_manager.h" |
| #include "policy/policy_constants.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace policy { |
| |
| namespace { |
| |
| // Utility functions for the tests. |
| void SetPolicy(PolicyMap* map, const char* name, Value* value) { |
| map->Set(name, POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, value, NULL); |
| } |
| |
| void SetPolicy(PolicyMap* map, |
| const char* name, |
| ExternalDataFetcher* external_data_fetcher) { |
| map->Set(name, POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
| NULL, external_data_fetcher); |
| } |
| |
| } // namespace |
| |
| class PolicyMapTest : public testing::Test { |
| protected: |
| scoped_ptr<ExternalDataFetcher> CreateExternalDataFetcher( |
| const std::string& policy) const; |
| }; |
| |
| scoped_ptr<ExternalDataFetcher> PolicyMapTest::CreateExternalDataFetcher( |
| const std::string& policy) const { |
| return make_scoped_ptr( |
| new ExternalDataFetcher(base::WeakPtr<ExternalDataManager>(), policy)); |
| } |
| |
| TEST_F(PolicyMapTest, SetAndGet) { |
| PolicyMap map; |
| SetPolicy(&map, key::kHomepageLocation, Value::CreateStringValue("aaa")); |
| StringValue expected("aaa"); |
| EXPECT_TRUE(expected.Equals(map.GetValue(key::kHomepageLocation))); |
| SetPolicy(&map, key::kHomepageLocation, Value::CreateStringValue("bbb")); |
| StringValue expected_b("bbb"); |
| EXPECT_TRUE(expected_b.Equals(map.GetValue(key::kHomepageLocation))); |
| SetPolicy(&map, key::kHomepageLocation, |
| CreateExternalDataFetcher("dummy").release()); |
| EXPECT_FALSE(map.GetValue(key::kHomepageLocation)); |
| const PolicyMap::Entry* entry = map.Get(key::kHomepageLocation); |
| ASSERT_TRUE(entry != NULL); |
| EXPECT_EQ(POLICY_LEVEL_MANDATORY, entry->level); |
| EXPECT_EQ(POLICY_SCOPE_USER, entry->scope); |
| EXPECT_TRUE(ExternalDataFetcher::Equals( |
| entry->external_data_fetcher, CreateExternalDataFetcher("dummy").get())); |
| map.Set(key::kHomepageLocation, POLICY_LEVEL_RECOMMENDED, |
| POLICY_SCOPE_MACHINE, NULL, NULL); |
| EXPECT_FALSE(map.GetValue(key::kHomepageLocation)); |
| entry = map.Get(key::kHomepageLocation); |
| ASSERT_TRUE(entry != NULL); |
| EXPECT_EQ(POLICY_LEVEL_RECOMMENDED, entry->level); |
| EXPECT_EQ(POLICY_SCOPE_MACHINE, entry->scope); |
| EXPECT_FALSE(entry->external_data_fetcher); |
| } |
| |
| TEST_F(PolicyMapTest, Equals) { |
| PolicyMap a; |
| SetPolicy(&a, key::kHomepageLocation, Value::CreateStringValue("aaa")); |
| PolicyMap a2; |
| SetPolicy(&a2, key::kHomepageLocation, Value::CreateStringValue("aaa")); |
| PolicyMap b; |
| SetPolicy(&b, key::kHomepageLocation, Value::CreateStringValue("bbb")); |
| PolicyMap c; |
| SetPolicy(&c, key::kHomepageLocation, Value::CreateStringValue("aaa")); |
| SetPolicy(&c, key::kHomepageIsNewTabPage, Value::CreateBooleanValue(true)); |
| PolicyMap d; |
| SetPolicy(&d, key::kHomepageLocation, |
| CreateExternalDataFetcher("ddd").release()); |
| PolicyMap d2; |
| SetPolicy(&d2, key::kHomepageLocation, |
| CreateExternalDataFetcher("ddd").release()); |
| PolicyMap e; |
| SetPolicy(&e, key::kHomepageLocation, |
| CreateExternalDataFetcher("eee").release()); |
| EXPECT_FALSE(a.Equals(b)); |
| EXPECT_FALSE(a.Equals(c)); |
| EXPECT_FALSE(a.Equals(d)); |
| EXPECT_FALSE(a.Equals(e)); |
| EXPECT_FALSE(b.Equals(a)); |
| EXPECT_FALSE(b.Equals(c)); |
| EXPECT_FALSE(b.Equals(d)); |
| EXPECT_FALSE(b.Equals(e)); |
| EXPECT_FALSE(c.Equals(a)); |
| EXPECT_FALSE(c.Equals(b)); |
| EXPECT_FALSE(c.Equals(d)); |
| EXPECT_FALSE(c.Equals(e)); |
| EXPECT_FALSE(d.Equals(a)); |
| EXPECT_FALSE(d.Equals(b)); |
| EXPECT_FALSE(d.Equals(c)); |
| EXPECT_FALSE(d.Equals(e)); |
| EXPECT_FALSE(e.Equals(a)); |
| EXPECT_FALSE(e.Equals(b)); |
| EXPECT_FALSE(e.Equals(c)); |
| EXPECT_FALSE(e.Equals(d)); |
| EXPECT_TRUE(a.Equals(a2)); |
| EXPECT_TRUE(a2.Equals(a)); |
| EXPECT_TRUE(d.Equals(d2)); |
| EXPECT_TRUE(d2.Equals(d)); |
| PolicyMap empty1; |
| PolicyMap empty2; |
| EXPECT_TRUE(empty1.Equals(empty2)); |
| EXPECT_TRUE(empty2.Equals(empty1)); |
| EXPECT_FALSE(empty1.Equals(a)); |
| EXPECT_FALSE(a.Equals(empty1)); |
| } |
| |
| TEST_F(PolicyMapTest, Swap) { |
| PolicyMap a; |
| SetPolicy(&a, key::kHomepageLocation, Value::CreateStringValue("aaa")); |
| SetPolicy(&a, key::kAlternateErrorPagesEnabled, |
| CreateExternalDataFetcher("dummy").release()); |
| PolicyMap b; |
| SetPolicy(&b, key::kHomepageLocation, Value::CreateStringValue("bbb")); |
| SetPolicy(&b, key::kHomepageIsNewTabPage, Value::CreateBooleanValue(true)); |
| |
| a.Swap(&b); |
| base::StringValue expected("bbb"); |
| EXPECT_TRUE(expected.Equals(a.GetValue(key::kHomepageLocation))); |
| base::FundamentalValue expected_bool(true); |
| EXPECT_TRUE(expected_bool.Equals(a.GetValue(key::kHomepageIsNewTabPage))); |
| EXPECT_FALSE(a.GetValue(key::kAlternateErrorPagesEnabled)); |
| EXPECT_FALSE(a.Get(key::kAlternateErrorPagesEnabled)); |
| StringValue expected_a("aaa"); |
| EXPECT_TRUE(expected_a.Equals(b.GetValue(key::kHomepageLocation))); |
| EXPECT_FALSE(b.GetValue(key::kHomepageIsNewTabPage)); |
| EXPECT_FALSE(a.GetValue(key::kAlternateErrorPagesEnabled)); |
| const PolicyMap::Entry* entry = b.Get(key::kAlternateErrorPagesEnabled); |
| ASSERT_TRUE(entry); |
| EXPECT_TRUE(ExternalDataFetcher::Equals( |
| CreateExternalDataFetcher("dummy").get(), entry->external_data_fetcher)); |
| |
| b.Clear(); |
| a.Swap(&b); |
| PolicyMap empty; |
| EXPECT_TRUE(a.Equals(empty)); |
| EXPECT_FALSE(b.Equals(empty)); |
| } |
| |
| TEST_F(PolicyMapTest, MergeFrom) { |
| PolicyMap a; |
| a.Set(key::kHomepageLocation, POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
| Value::CreateStringValue("google.com"), NULL); |
| a.Set(key::kShowHomeButton, POLICY_LEVEL_MANDATORY, POLICY_SCOPE_MACHINE, |
| Value::CreateBooleanValue(true), NULL); |
| a.Set(key::kAlternateErrorPagesEnabled, |
| POLICY_LEVEL_MANDATORY, POLICY_SCOPE_MACHINE, |
| NULL, CreateExternalDataFetcher("a").release()); |
| a.Set(key::kBookmarkBarEnabled, POLICY_LEVEL_RECOMMENDED, POLICY_SCOPE_USER, |
| Value::CreateBooleanValue(false), NULL); |
| a.Set(key::kDefaultSearchProviderSearchURL, |
| POLICY_LEVEL_RECOMMENDED, POLICY_SCOPE_MACHINE, |
| Value::CreateStringValue("google.com/q={x}"), NULL); |
| |
| PolicyMap b; |
| b.Set(key::kHomepageLocation, POLICY_LEVEL_MANDATORY, POLICY_SCOPE_MACHINE, |
| Value::CreateStringValue("chromium.org"), NULL); |
| b.Set(key::kShowHomeButton, POLICY_LEVEL_MANDATORY, POLICY_SCOPE_MACHINE, |
| Value::CreateBooleanValue(false), NULL); |
| b.Set(key::kAlternateErrorPagesEnabled, |
| POLICY_LEVEL_MANDATORY, POLICY_SCOPE_MACHINE, |
| NULL, CreateExternalDataFetcher("b").release()); |
| b.Set(key::kBookmarkBarEnabled, POLICY_LEVEL_RECOMMENDED, |
| POLICY_SCOPE_MACHINE, Value::CreateBooleanValue(true), NULL); |
| b.Set(key::kDefaultSearchProviderSearchURL, |
| POLICY_LEVEL_MANDATORY, |
| POLICY_SCOPE_MACHINE, |
| Value::CreateStringValue(std::string()), |
| NULL); |
| b.Set(key::kDisableSpdy, |
| POLICY_LEVEL_RECOMMENDED, |
| POLICY_SCOPE_USER, |
| Value::CreateBooleanValue(true), |
| NULL); |
| |
| a.MergeFrom(b); |
| |
| PolicyMap c; |
| // POLICY_SCOPE_MACHINE over POLICY_SCOPE_USER. |
| c.Set(key::kHomepageLocation, POLICY_LEVEL_MANDATORY, POLICY_SCOPE_MACHINE, |
| Value::CreateStringValue("chromium.org"), NULL); |
| // |a| has precedence over |b|. |
| c.Set(key::kShowHomeButton, POLICY_LEVEL_MANDATORY, POLICY_SCOPE_MACHINE, |
| Value::CreateBooleanValue(true), NULL); |
| c.Set(key::kAlternateErrorPagesEnabled, |
| POLICY_LEVEL_MANDATORY, POLICY_SCOPE_MACHINE, |
| NULL, CreateExternalDataFetcher("a").release()); |
| // POLICY_SCOPE_MACHINE over POLICY_SCOPE_USER for POLICY_LEVEL_RECOMMENDED. |
| c.Set(key::kBookmarkBarEnabled, POLICY_LEVEL_RECOMMENDED, |
| POLICY_SCOPE_MACHINE, Value::CreateBooleanValue(true), NULL); |
| // POLICY_LEVEL_MANDATORY over POLICY_LEVEL_RECOMMENDED. |
| c.Set(key::kDefaultSearchProviderSearchURL, |
| POLICY_LEVEL_MANDATORY, |
| POLICY_SCOPE_MACHINE, |
| Value::CreateStringValue(std::string()), |
| NULL); |
| // Merge new ones. |
| c.Set(key::kDisableSpdy, POLICY_LEVEL_RECOMMENDED, POLICY_SCOPE_USER, |
| Value::CreateBooleanValue(true), NULL); |
| |
| EXPECT_TRUE(a.Equals(c)); |
| } |
| |
| TEST_F(PolicyMapTest, GetDifferingKeys) { |
| PolicyMap a; |
| a.Set(key::kHomepageLocation, POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
| Value::CreateStringValue("google.com"), NULL); |
| a.Set(key::kSearchSuggestEnabled, |
| POLICY_LEVEL_MANDATORY, POLICY_SCOPE_MACHINE, |
| NULL, CreateExternalDataFetcher("dummy").release()); |
| a.Set(key::kShowHomeButton, POLICY_LEVEL_MANDATORY, POLICY_SCOPE_MACHINE, |
| Value::CreateBooleanValue(true), NULL); |
| a.Set(key::kAlternateErrorPagesEnabled, |
| POLICY_LEVEL_MANDATORY, POLICY_SCOPE_MACHINE, |
| NULL, CreateExternalDataFetcher("a").release()); |
| a.Set(key::kBookmarkBarEnabled, POLICY_LEVEL_RECOMMENDED, POLICY_SCOPE_USER, |
| Value::CreateBooleanValue(false), NULL); |
| a.Set(key::kDefaultSearchProviderSearchURL, |
| POLICY_LEVEL_RECOMMENDED, POLICY_SCOPE_MACHINE, |
| Value::CreateStringValue("google.com/q={x}"), NULL); |
| a.Set(key::kDisable3DAPIs, POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
| Value::CreateBooleanValue(true), NULL); |
| |
| PolicyMap b; |
| b.Set(key::kHomepageLocation, POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
| Value::CreateStringValue("google.com"), NULL); |
| b.Set(key::kSearchSuggestEnabled, |
| POLICY_LEVEL_MANDATORY, POLICY_SCOPE_MACHINE, |
| NULL, CreateExternalDataFetcher("dummy").release()); |
| b.Set(key::kShowHomeButton, POLICY_LEVEL_MANDATORY, POLICY_SCOPE_MACHINE, |
| Value::CreateBooleanValue(false), NULL); |
| b.Set(key::kAlternateErrorPagesEnabled, |
| POLICY_LEVEL_MANDATORY, POLICY_SCOPE_MACHINE, |
| NULL, CreateExternalDataFetcher("b").release()); |
| b.Set(key::kBookmarkBarEnabled, POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, |
| Value::CreateBooleanValue(false), NULL); |
| b.Set(key::kDefaultSearchProviderSearchURL, POLICY_LEVEL_RECOMMENDED, |
| POLICY_SCOPE_USER, Value::CreateStringValue("google.com/q={x}"), NULL); |
| b.Set(key::kDisableSpdy, POLICY_LEVEL_RECOMMENDED, POLICY_SCOPE_USER, |
| Value::CreateBooleanValue(true), NULL); |
| |
| std::set<std::string> diff; |
| std::set<std::string> diff2; |
| a.GetDifferingKeys(b, &diff); |
| b.GetDifferingKeys(a, &diff2); |
| // Order shouldn't matter. |
| EXPECT_EQ(diff, diff2); |
| // No change. |
| EXPECT_TRUE(diff.find(key::kHomepageLocation) == diff.end()); |
| EXPECT_TRUE(diff.find(key::kSearchSuggestEnabled) == diff.end()); |
| // Different values. |
| EXPECT_TRUE(diff.find(key::kShowHomeButton) != diff.end()); |
| // Different external data references. |
| EXPECT_TRUE(diff.find(key::kAlternateErrorPagesEnabled) != diff.end()); |
| // Different levels. |
| EXPECT_TRUE(diff.find(key::kBookmarkBarEnabled) != diff.end()); |
| // Different scopes. |
| EXPECT_TRUE(diff.find(key::kDefaultSearchProviderSearchURL) != diff.end()); |
| // Not in |a|. |
| EXPECT_TRUE(diff.find(key::kDisableSpdy) != diff.end()); |
| // Not in |b|. |
| EXPECT_TRUE(diff.find(key::kDisable3DAPIs) != diff.end()); |
| // No surprises. |
| EXPECT_EQ(6u, diff.size()); |
| } |
| |
| } // namespace policy |