src/chrome/browser/chromeos/policy/proxy_policy_provider_unittest.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/basictypes.h"
   6 #include "base/callback.h"
   7 #include "chrome/browser/chromeos/policy/proxy_policy_provider.h"
   8 #include "chrome/browser/policy/external_data_fetcher.h"
   9 #include "chrome/browser/policy/mock_configuration_policy_provider.h"
  10 #include "testing/gmock/include/gmock/gmock.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12
  13 using testing::Mock;
  14
  15 namespace policy {
  16
  17 class ProxyPolicyProviderTest : public testing::Test {
  18  protected:
  19   ProxyPolicyProviderTest() {
  20     mock_provider_.Init();
  21     proxy_provider_.Init();
  22     proxy_provider_.AddObserver(&observer_);
  23   }
  24
  25   virtual ~ProxyPolicyProviderTest() {
  26     proxy_provider_.RemoveObserver(&observer_);
  27     proxy_provider_.Shutdown();
  28     mock_provider_.Shutdown();
  29   }
  30
  31   MockConfigurationPolicyObserver observer_;
  32   MockConfigurationPolicyProvider mock_provider_;
  33   ProxyPolicyProvider proxy_provider_;
  34
  35   static scoped_ptr<PolicyBundle> CopyBundle(const PolicyBundle& bundle) {
  36     scoped_ptr<PolicyBundle> copy(new PolicyBundle());
  37     copy->CopyFrom(bundle);
  38     return copy.Pass();
  39   }
  40
  41   DISALLOW_COPY_AND_ASSIGN(ProxyPolicyProviderTest);
  42 };
  43
  44 TEST_F(ProxyPolicyProviderTest, Init) {
  45   EXPECT_TRUE(proxy_provider_.IsInitializationComplete(POLICY_DOMAIN_CHROME));
  46   EXPECT_TRUE(PolicyBundle().Equals(proxy_provider_.policies()));
  47 }
  48
  49 TEST_F(ProxyPolicyProviderTest, Delegate) {
  50   PolicyBundle bundle;
  51   bundle.Get(PolicyNamespace(POLICY_DOMAIN_CHROME, std::string()))
  52       .Set("policy",
  53            POLICY_LEVEL_MANDATORY,
  54            POLICY_SCOPE_USER,
  55            Value::CreateStringValue("value"),
  56            NULL);
  57   mock_provider_.UpdatePolicy(CopyBundle(bundle));
  58
  59   EXPECT_CALL(observer_, OnUpdatePolicy(&proxy_provider_));
  60   proxy_provider_.SetDelegate(&mock_provider_);
  61   Mock::VerifyAndClearExpectations(&observer_);
  62   EXPECT_TRUE(bundle.Equals(proxy_provider_.policies()));
  63
  64   EXPECT_CALL(observer_, OnUpdatePolicy(&proxy_provider_));
  65   bundle.Get(PolicyNamespace(POLICY_DOMAIN_CHROME, std::string()))
  66       .Set("policy",
  67            POLICY_LEVEL_MANDATORY,
  68            POLICY_SCOPE_USER,
  69            Value::CreateStringValue("new value"),
  70            NULL);
  71   mock_provider_.UpdatePolicy(CopyBundle(bundle));
  72   Mock::VerifyAndClearExpectations(&observer_);
  73   EXPECT_TRUE(bundle.Equals(proxy_provider_.policies()));
  74
  75   EXPECT_CALL(observer_, OnUpdatePolicy(&proxy_provider_));
  76   proxy_provider_.SetDelegate(NULL);
  77   EXPECT_TRUE(PolicyBundle().Equals(proxy_provider_.policies()));
  78 }
  79
  80 TEST_F(ProxyPolicyProviderTest, RefreshPolicies) {
  81   EXPECT_CALL(observer_, OnUpdatePolicy(&proxy_provider_));
  82   proxy_provider_.RefreshPolicies();
  83   Mock::VerifyAndClearExpectations(&observer_);
  84
  85   EXPECT_CALL(observer_, OnUpdatePolicy(&proxy_provider_));
  86   proxy_provider_.SetDelegate(&mock_provider_);
  87   Mock::VerifyAndClearExpectations(&observer_);
  88
  89   EXPECT_CALL(observer_, OnUpdatePolicy(&proxy_provider_)).Times(0);
  90   EXPECT_CALL(mock_provider_, RefreshPolicies());
  91   proxy_provider_.RefreshPolicies();
  92   Mock::VerifyAndClearExpectations(&observer_);
  93   Mock::VerifyAndClearExpectations(&mock_provider_);
  94
  95   EXPECT_CALL(observer_, OnUpdatePolicy(&proxy_provider_));
  96   mock_provider_.UpdatePolicy(scoped_ptr<PolicyBundle>(new PolicyBundle()));
  97   Mock::VerifyAndClearExpectations(&observer_);
  98 }
  99
 100 }  // namespace policy