Chromium Code Reviews| 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..8eb7f04f414da0f3920c923bde88eac1162026a7 |
| --- /dev/null |
| +++ b/chrome/browser/policy/asynchronous_policy_test_base.h |
| @@ -0,0 +1,77 @@ |
| +// 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" |
|
Mattias Nissler (ping if slow)
2010/12/06 10:26:20
do you need this header?
danno
2010/12/06 14:05:12
Done.
|
| +#include "chrome/browser/browser_thread.h" |
| +#include "chrome/browser/policy/asynchronous_policy_provider.h" |
| +#include "chrome/browser/policy/configuration_policy_provider.h" |
|
Mattias Nissler (ping if slow)
2010/12/06 10:26:20
isn't this one included in asynchronous_policy_pro
danno
2010/12/06 14:05:12
Done.
|
| +#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_METHOD0(Load, DictionaryValue*()); |
| + |
| + 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(); |
| + } |
| + |
| + protected: |
| + MessageLoop loop_; |
| + scoped_ptr<ConfigurationPolicyProvider> provider_; |
|
Mattias Nissler (ping if slow)
2010/12/06 10:26:20
Seems like the provider_ is not really needed here
danno
2010/12/06 14:05:12
Done.
|
| + |
| + // 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_; |
|
Mattias Nissler (ping if slow)
2010/12/06 10:26:20
But there's no harm in always running it? So why n
danno
2010/12/06 14:05:12
This is dead code now.
On 2010/12/06 10:26:20, Ma
|
| + BrowserThread ui_thread_; |
| + BrowserThread file_thread_; |
| + |
| + DISALLOW_COPY_AND_ASSIGN(AsynchronousPolicyTestBase); |
| +}; |
| + |
| +inline static void MessageLoopQuitNow() { |
| + MessageLoop::current()->QuitNow(); |
|
Mattias Nissler (ping if slow)
2010/12/06 10:26:20
Do you really want to call QuitNow() instead of Qu
danno
2010/12/06 14:05:12
Yes. But this functionality isn't used anymore aft
|
| +} |
| + |
| +} // namespace policy |
| + |
| +#endif // CHROME_BROWSER_POLICY_ASYNCHRONOUS_POLICY_TEST_BASE_H_ |