Chromium Code Reviews| Index: chrome/browser/policy/cloud_policy_refresh_scheduler_unittest.cc |
| diff --git a/chrome/browser/policy/cloud_policy_refresh_scheduler_unittest.cc b/chrome/browser/policy/cloud_policy_refresh_scheduler_unittest.cc |
| index d133be7335f394ca9d3864c3631533b6a704d3c3..0aa8c7c4008bde36ed3324cd406d07ed5ec08325 100644 |
| --- a/chrome/browser/policy/cloud_policy_refresh_scheduler_unittest.cc |
| +++ b/chrome/browser/policy/cloud_policy_refresh_scheduler_unittest.cc |
| @@ -7,13 +7,13 @@ |
| #include "base/memory/ref_counted.h" |
| #include "base/memory/scoped_ptr.h" |
| #include "base/message_loop.h" |
| +#include "base/values.h" |
|
pastarmovj
2012/12/04 12:20:38
Do you need this include?
Mattias Nissler (ping if slow)
2012/12/04 13:20:40
No, removed.
|
| #include "chrome/browser/policy/cloud_policy_refresh_scheduler.h" |
| #include "chrome/browser/policy/mock_cloud_policy_client.h" |
| #include "chrome/browser/policy/mock_cloud_policy_store.h" |
| #include "chrome/browser/policy/test_task_runner.h" |
| #include "chrome/browser/prefs/browser_prefs.h" |
| -#include "chrome/common/pref_names.h" |
| -#include "chrome/test/base/testing_pref_service.h" |
| +#include "policy/policy_constants.h" |
| #include "testing/gmock/include/gmock/gmock.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| @@ -34,10 +34,7 @@ class CloudPolicyRefreshSchedulerTest : public testing::Test { |
| protected: |
| CloudPolicyRefreshSchedulerTest() |
| : task_runner_(new TestTaskRunner()), |
| - network_change_notifier_(net::NetworkChangeNotifier::CreateMock()) { |
| - chrome::RegisterLocalState(&prefs_); |
| - prefs_.SetInteger(prefs::kUserPolicyRefreshRate, kPolicyRefreshRate); |
| - } |
| + network_change_notifier_(net::NetworkChangeNotifier::CreateMock()) {} |
| virtual void SetUp() OVERRIDE { |
| client_.SetDMToken("token"); |
| @@ -61,9 +58,10 @@ class CloudPolicyRefreshSchedulerTest : public testing::Test { |
| } |
| CloudPolicyRefreshScheduler* CreateRefreshScheduler() { |
| - return new CloudPolicyRefreshScheduler(&client_, &store_, &prefs_, |
| - prefs::kUserPolicyRefreshRate, |
| - task_runner_); |
| + CloudPolicyRefreshScheduler* scheduler = |
| + new CloudPolicyRefreshScheduler(&client_, &store_, task_runner_); |
| + scheduler->SetRefreshDelay(kPolicyRefreshRate); |
| + return scheduler; |
| } |
| void NotifyIPAddressChanged() { |
| @@ -82,7 +80,6 @@ class CloudPolicyRefreshSchedulerTest : public testing::Test { |
| MessageLoop loop_; |
| MockCloudPolicyClient client_; |
| MockCloudPolicyStore store_; |
| - TestingPrefService prefs_; |
| scoped_refptr<TestTaskRunner> task_runner_; |
| scoped_ptr<net::NetworkChangeNotifier> network_change_notifier_; |
| @@ -136,19 +133,19 @@ TEST_F(CloudPolicyRefreshSchedulerTest, Unregistered) { |
| client_.NotifyPolicyFetched(); |
| client_.NotifyRegistrationStateChanged(); |
| client_.NotifyClientError(); |
| + scheduler->SetRefreshDelay(12 * 60 * 60 * 1000); |
| store_.NotifyStoreLoaded(); |
| store_.NotifyStoreError(); |
| - prefs_.SetInteger(prefs::kUserPolicyRefreshRate, 12 * 60 * 60 * 1000); |
| } |
| class CloudPolicyRefreshSchedulerSteadyStateTest |
| : public CloudPolicyRefreshSchedulerTest { |
| protected: |
| CloudPolicyRefreshSchedulerSteadyStateTest() |
| - : refresh_scheduler_(&client_, &store_, &prefs_, |
| - prefs::kUserPolicyRefreshRate, task_runner_) {} |
| + : refresh_scheduler_(&client_, &store_, task_runner_) {} |
| virtual void SetUp() OVERRIDE { |
| + refresh_scheduler_.SetRefreshDelay(kPolicyRefreshRate); |
| CloudPolicyRefreshSchedulerTest::SetUp(); |
| last_refresh_ = base::Time::NowFromSystemTime(); |
| client_.NotifyPolicyFetched(); |
| @@ -185,15 +182,15 @@ TEST_F(CloudPolicyRefreshSchedulerSteadyStateTest, OnStoreError) { |
| TEST_F(CloudPolicyRefreshSchedulerSteadyStateTest, RefreshDelayChange) { |
| const int delay_short_ms = 5 * 60 * 1000; |
| - prefs_.SetInteger(prefs::kUserPolicyRefreshRate, delay_short_ms); |
| + refresh_scheduler_.SetRefreshDelay(delay_short_ms); |
| CheckTiming(CloudPolicyRefreshScheduler::kRefreshDelayMinMs); |
| const int delay_ms = 12 * 60 * 60 * 1000; |
| - prefs_.SetInteger(prefs::kUserPolicyRefreshRate, delay_ms); |
| + refresh_scheduler_.SetRefreshDelay(delay_ms); |
| CheckTiming(delay_ms); |
| const int delay_long_ms = 2 * 24 * 60 * 60 * 1000; |
| - prefs_.SetInteger(prefs::kUserPolicyRefreshRate, delay_long_ms); |
| + refresh_scheduler_.SetRefreshDelay(delay_long_ms); |
| CheckTiming(CloudPolicyRefreshScheduler::kRefreshDelayMaxMs); |
| } |