| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/policy/cloud/cloud_policy_core.h" | 5 #include "chrome/browser/policy/cloud/cloud_policy_core.h" |
| 6 | 6 |
| 7 #include "base/basictypes.h" | 7 #include "base/basictypes.h" |
| 8 #include "base/message_loop/message_loop.h" | 8 #include "base/message_loop/message_loop.h" |
| 9 #include "base/prefs/pref_registry_simple.h" |
| 9 #include "base/prefs/testing_pref_service.h" | 10 #include "base/prefs/testing_pref_service.h" |
| 10 #include "chrome/browser/policy/cloud/cloud_policy_constants.h" | 11 #include "chrome/browser/policy/cloud/cloud_policy_constants.h" |
| 11 #include "chrome/browser/policy/cloud/cloud_policy_refresh_scheduler.h" | 12 #include "chrome/browser/policy/cloud/cloud_policy_refresh_scheduler.h" |
| 12 #include "chrome/browser/policy/cloud/mock_cloud_policy_client.h" | 13 #include "chrome/browser/policy/cloud/mock_cloud_policy_client.h" |
| 13 #include "chrome/browser/policy/cloud/mock_cloud_policy_store.h" | 14 #include "chrome/browser/policy/cloud/mock_cloud_policy_store.h" |
| 14 #include "chrome/browser/prefs/browser_prefs.h" | 15 #include "components/policy/core/common/policy_pref_names.h" |
| 15 #include "chrome/common/pref_names.h" | |
| 16 #include "testing/gtest/include/gtest/gtest.h" | 16 #include "testing/gtest/include/gtest/gtest.h" |
| 17 | 17 |
| 18 namespace policy { | 18 namespace policy { |
| 19 | 19 |
| 20 class CloudPolicyCoreTest : public testing::Test, | 20 class CloudPolicyCoreTest : public testing::Test, |
| 21 public CloudPolicyCore::Observer { | 21 public CloudPolicyCore::Observer { |
| 22 protected: | 22 protected: |
| 23 CloudPolicyCoreTest() | 23 CloudPolicyCoreTest() |
| 24 : core_(PolicyNamespaceKey(dm_protocol::kChromeUserPolicyType, | 24 : core_(PolicyNamespaceKey(dm_protocol::kChromeUserPolicyType, |
| 25 std::string()), | 25 std::string()), |
| 26 &store_, | 26 &store_, |
| 27 loop_.message_loop_proxy()), | 27 loop_.message_loop_proxy()), |
| 28 core_connected_callback_count_(0), | 28 core_connected_callback_count_(0), |
| 29 refresh_scheduler_started_callback_count_(0), | 29 refresh_scheduler_started_callback_count_(0), |
| 30 core_disconnecting_callback_count_(0), | 30 core_disconnecting_callback_count_(0), |
| 31 bad_callback_count_(0) { | 31 bad_callback_count_(0) { |
| 32 chrome::RegisterLocalState(prefs_.registry()); | 32 prefs_.registry()->RegisterIntegerPref( |
| 33 policy_prefs::kUserPolicyRefreshRate, |
| 34 CloudPolicyRefreshScheduler::kDefaultRefreshDelayMs); |
| 33 core_.AddObserver(this); | 35 core_.AddObserver(this); |
| 34 } | 36 } |
| 35 | 37 |
| 36 virtual ~CloudPolicyCoreTest() { | 38 virtual ~CloudPolicyCoreTest() { |
| 37 core_.RemoveObserver(this); | 39 core_.RemoveObserver(this); |
| 38 } | 40 } |
| 39 | 41 |
| 40 virtual void OnCoreConnected(CloudPolicyCore* core) OVERRIDE { | 42 virtual void OnCoreConnected(CloudPolicyCore* core) OVERRIDE { |
| 41 // Make sure core is connected at callback time. | 43 // Make sure core is connected at callback time. |
| 42 if (core_.client()) | 44 if (core_.client()) |
| (...skipping 70 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 113 | 115 |
| 114 TEST_F(CloudPolicyCoreTest, RefreshScheduler) { | 116 TEST_F(CloudPolicyCoreTest, RefreshScheduler) { |
| 115 EXPECT_FALSE(core_.refresh_scheduler()); | 117 EXPECT_FALSE(core_.refresh_scheduler()); |
| 116 core_.Connect(scoped_ptr<CloudPolicyClient>(new MockCloudPolicyClient())); | 118 core_.Connect(scoped_ptr<CloudPolicyClient>(new MockCloudPolicyClient())); |
| 117 core_.StartRefreshScheduler(); | 119 core_.StartRefreshScheduler(); |
| 118 ASSERT_TRUE(core_.refresh_scheduler()); | 120 ASSERT_TRUE(core_.refresh_scheduler()); |
| 119 | 121 |
| 120 int default_refresh_delay = core_.refresh_scheduler()->refresh_delay(); | 122 int default_refresh_delay = core_.refresh_scheduler()->refresh_delay(); |
| 121 | 123 |
| 122 const int kRefreshRate = 1000 * 60 * 60; | 124 const int kRefreshRate = 1000 * 60 * 60; |
| 123 prefs_.SetInteger(prefs::kUserPolicyRefreshRate, kRefreshRate); | 125 prefs_.SetInteger(policy_prefs::kUserPolicyRefreshRate, kRefreshRate); |
| 124 core_.TrackRefreshDelayPref(&prefs_, prefs::kUserPolicyRefreshRate); | 126 core_.TrackRefreshDelayPref(&prefs_, policy_prefs::kUserPolicyRefreshRate); |
| 125 EXPECT_EQ(kRefreshRate, core_.refresh_scheduler()->refresh_delay()); | 127 EXPECT_EQ(kRefreshRate, core_.refresh_scheduler()->refresh_delay()); |
| 126 | 128 |
| 127 prefs_.ClearPref(prefs::kUserPolicyRefreshRate); | 129 prefs_.ClearPref(policy_prefs::kUserPolicyRefreshRate); |
| 128 EXPECT_EQ(default_refresh_delay, core_.refresh_scheduler()->refresh_delay()); | 130 EXPECT_EQ(default_refresh_delay, core_.refresh_scheduler()->refresh_delay()); |
| 129 | 131 |
| 130 EXPECT_EQ(1, core_connected_callback_count_); | 132 EXPECT_EQ(1, core_connected_callback_count_); |
| 131 EXPECT_EQ(1, refresh_scheduler_started_callback_count_); | 133 EXPECT_EQ(1, refresh_scheduler_started_callback_count_); |
| 132 EXPECT_EQ(0, core_disconnecting_callback_count_); | 134 EXPECT_EQ(0, core_disconnecting_callback_count_); |
| 133 EXPECT_EQ(0, bad_callback_count_); | 135 EXPECT_EQ(0, bad_callback_count_); |
| 134 } | 136 } |
| 135 | 137 |
| 136 } // namespace policy | 138 } // namespace policy |
| OLD | NEW |