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 |