| Index: chrome/browser/policy/asynchronous_policy_test_base.h
|
| diff --git a/chrome/browser/policy/asynchronous_policy_test_base.h b/chrome/browser/policy/asynchronous_policy_test_base.h
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..ffab5e86940ed9ee788aca258c438528964c8ab5
|
| --- /dev/null
|
| +++ b/chrome/browser/policy/asynchronous_policy_test_base.h
|
| @@ -0,0 +1,83 @@
|
| +// Copyright (c) 2010 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.
|
| +
|
| +#ifndef CHROME_BROWSER_POLICY_ASYNCHRONOUS_POLICY_TEST_BASE_H_
|
| +#define CHROME_BROWSER_POLICY_ASYNCHRONOUS_POLICY_TEST_BASE_H_
|
| +#pragma once
|
| +
|
| +#include "base/message_loop.h"
|
| +#include "base/time.h"
|
| +#include "chrome/browser/browser_thread.h"
|
| +#include "chrome/browser/policy/asynchronous_policy_provider.h"
|
| +#include "chrome/browser/policy/mock_configuration_policy_store.h"
|
| +#include "testing/gmock/include/gmock/gmock.h"
|
| +#include "testing/gtest/include/gtest/gtest.h"
|
| +
|
| +namespace policy {
|
| +
|
| +// A delegate for testing that can feed arbitrary information to the loader.
|
| +class ProviderDelegateMock : public AsynchronousPolicyProvider::Delegate {
|
| + public:
|
| + ProviderDelegateMock() : AsynchronousPolicyProvider::Delegate() {}
|
| + virtual ~ProviderDelegateMock() {}
|
| +
|
| + MOCK_METHOD1(Init, void(AsynchronousPolicyProvider::PolicyChangeObserver*));
|
| + MOCK_METHOD0(Stop, void());
|
| + MOCK_METHOD0(Load, DictionaryValue*());
|
| + MOCK_METHOD2(IsSafeToReloadPolicy, bool(const base::Time&, base::TimeDelta*));
|
| +
|
| + private:
|
| + DISALLOW_COPY_AND_ASSIGN(ProviderDelegateMock);
|
| +};
|
| +
|
| +class AsynchronousPolicyTestBase : public testing::Test {
|
| + public:
|
| + AsynchronousPolicyTestBase()
|
| + : run_pending_during_tear_down_(true),
|
| + ui_thread_(BrowserThread::UI, &loop_),
|
| + file_thread_(BrowserThread::FILE, &loop_) {}
|
| +
|
| + virtual ~AsynchronousPolicyTestBase() {}
|
| +
|
| + virtual void SetUp() {
|
| + delegate_.reset(new ProviderDelegateMock());
|
| + store_.reset(new MockConfigurationPolicyStore);
|
| + }
|
| +
|
| + virtual void TearDown() {
|
| + provider_.reset(NULL);
|
| + if (run_pending_during_tear_down_)
|
| + loop_.RunAllPending();
|
| + }
|
| +
|
| + void DontRunAllPendingDuringTearDown() {
|
| + run_pending_during_tear_down_ = false;
|
| + }
|
| +
|
| + protected:
|
| + MessageLoop loop_;
|
| + scoped_ptr<AsynchronousPolicyProvider> provider_;
|
| +
|
| + // The mocks that are used in the test must outlive the scope of the test
|
| + // because they still get accessed in the RunAllPending of the TearDown.
|
| + scoped_ptr<MockConfigurationPolicyStore> store_;
|
| + scoped_ptr<ProviderDelegateMock> delegate_;
|
| +
|
| + private:
|
| + // True if a test guarantees that it's handled all tasks pending in the
|
| + // message loop and doesn't want RunAllPending run during TearDown.
|
| + bool run_pending_during_tear_down_;
|
| + BrowserThread ui_thread_;
|
| + BrowserThread file_thread_;
|
| +
|
| + DISALLOW_COPY_AND_ASSIGN(AsynchronousPolicyTestBase);
|
| +};
|
| +
|
| +inline static void MessageLoopQuitNow() {
|
| + MessageLoop::current()->QuitNow();
|
| +}
|
| +
|
| +} // namespace policy
|
| +
|
| +#endif // CHROME_BROWSER_POLICY_ASYNCHRONOUS_POLICY_TEST_BASE_H_
|
|
|