1 // Copyright 2014 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
5 #include "athena/test/athena_test_helper.h"
7 #include "athena/env/public/athena_env.h"
8 #include "athena/extensions/public/extensions_delegate.h"
9 #include "athena/main/public/athena_launcher.h"
10 #include "athena/test/sample_activity_factory.h"
11 #include "athena/test/test_app_model_builder.h"
12 #include "base/message_loop/message_loop.h"
13 #include "base/run_loop.h"
14 #include "base/threading/thread.h"
15 #include "chromeos/dbus/dbus_thread_manager.h"
16 #include "chromeos/network/network_handler.h"
17 #include "ui/aura/env.h"
18 #include "ui/aura/input_state_lookup.h"
19 #include "ui/aura/test/env_test_helper.h"
20 #include "ui/base/ime/input_method_initializer.h"
21 #include "ui/compositor/scoped_animation_duration_scale_mode.h"
22 #include "ui/wm/core/focus_controller.h"
23 #include "ui/wm/core/input_method_event_filter.h"
26 #include "ui/base/x/x11_util.h"
32 AthenaTestHelper::AthenaTestHelper(base::MessageLoopForUI* message_loop)
33 : setup_called_(false), teardown_called_(false) {
35 message_loop_ = message_loop;
36 // Disable animations during tests.
37 zero_duration_mode_.reset(new ui::ScopedAnimationDurationScaleMode(
38 ui::ScopedAnimationDurationScaleMode::ZERO_DURATION));
41 AthenaTestHelper::~AthenaTestHelper() {
42 CHECK(setup_called_) << "AthenaTestHelper::SetUp() never called.";
43 CHECK(teardown_called_) << "AthenaTestHelper::TearDown() never called.";
46 void AthenaTestHelper::SetUp(ui::ContextFactory* context_factory) {
48 file_thread_.reset(new base::Thread("FileThread"));
49 base::Thread::Options options(base::MessageLoop::TYPE_IO, 0);
50 file_thread_->StartWithOptions(options);
52 chromeos::DBusThreadManager::Initialize();
53 chromeos::NetworkHandler::Initialize();
54 ui::InitializeInputMethodForTesting();
55 aura::Env::CreateInstance(true);
56 aura::Env::GetInstance()->set_context_factory(context_factory);
58 // Unit tests generally don't want to query the system, rather use the state
60 aura::test::EnvTestHelper(aura::Env::GetInstance())
61 .SetInputStateLookup(scoped_ptr<aura::InputStateLookup>());
63 // TODO(oshima): Use a BlockingPool task runner.
64 athena::StartAthenaEnv(file_thread_->message_loop_proxy());
65 athena::ExtensionsDelegate::CreateExtensionsDelegateForTest();
66 athena::StartAthenaSession(new SampleActivityFactory(),
67 new TestAppModelBuilder());
70 void AthenaTestHelper::TearDown() {
71 teardown_called_ = true;
73 athena::ShutdownAthena();
74 aura::Env::DeleteInstance();
77 ui::test::ResetXCursorCache();
80 ui::ShutdownInputMethodForTesting();
81 chromeos::NetworkHandler::Shutdown();
82 chromeos::DBusThreadManager::Shutdown();
85 aura::Window* AthenaTestHelper::GetRootWindow() {
86 return GetHost()->window();
89 aura::WindowTreeHost* AthenaTestHelper::GetHost() {
90 return AthenaEnv::Get()->GetHost();
93 void AthenaTestHelper::RunAllPendingInMessageLoop() {
94 // TODO(jbates) crbug.com/134753 Find quitters of this RunLoop and have them
95 // use run_loop.QuitClosure().
96 base::RunLoop run_loop;
97 run_loop.RunUntilIdle();
101 } // namespace athena