| // Copyright 2013 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 ASH_TEST_ASH_TEST_HELPER_H_ |
| #define ASH_TEST_ASH_TEST_HELPER_H_ |
| |
| #include <stdint.h> |
| |
| #include <memory> |
| #include <utility> |
| |
| #include "ash/assistant/test/test_assistant_service.h" |
| #include "ash/session/test_session_controller_client.h" |
| #include "ash/shell_init_params.h" |
| #include "base/macros.h" |
| #include "base/optional.h" |
| #include "base/test/scoped_command_line.h" |
| |
| class PrefService; |
| |
| namespace aura { |
| class Window; |
| } |
| |
| namespace chromeos { |
| namespace system { |
| class ScopedFakeStatisticsProvider; |
| } // namespace system |
| } // namespace chromeos |
| |
| namespace display { |
| class Display; |
| } |
| |
| namespace ui { |
| class ScopedAnimationDurationScaleMode; |
| class TestContextFactories; |
| } |
| |
| namespace wm { |
| class WMState; |
| } |
| |
| namespace ash { |
| |
| class AppListTestHelper; |
| class AshTestViewsDelegate; |
| class TestKeyboardControllerObserver; |
| class TestNewWindowDelegate; |
| class TestNotifierSettingsController; |
| class TestPrefServiceProvider; |
| class TestShellDelegate; |
| class TestSystemTrayClient; |
| |
| // A helper class that does common initialization required for Ash. Creates a |
| // root window and an ash::Shell instance with a test delegate. |
| class AshTestHelper { |
| public: |
| // Instantiates/destroys an AshTestHelper. This can happen in a |
| // single-threaded phase without a backing task environment. As such, the vast |
| // majority of initialization/tear down will be done in SetUp()/TearDown(). |
| AshTestHelper(); |
| ~AshTestHelper(); |
| |
| enum ConfigType { |
| // The configuration for shell executable. |
| kShell, |
| // The configuration for unit tests. |
| kUnitTest, |
| // The configuration for perf tests. Unlike kUnitTest, this |
| // does not disable animations. |
| kPerfTest, |
| }; |
| |
| struct InitParams { |
| // True if the user should log in. |
| bool start_session = true; |
| // True to inject local-state PrefService into the Shell. |
| bool provide_local_state = true; |
| ConfigType config_type = kUnitTest; |
| }; |
| |
| // Creates the ash::Shell and performs associated initialization according |
| // to |init_params|. |shell_init_params| is used to initialize ash::Shell, |
| // or it uses test settings if omitted. |
| void SetUp(const InitParams& init_params, |
| base::Optional<ShellInitParams> shell_init_params = base::nullopt); |
| |
| // Destroys the ash::Shell and performs associated cleanup. |
| void TearDown(); |
| |
| // Returns a root Window. Usually this is the active root Window, but that |
| // method can return NULL sometimes, and in those cases, we fall back on the |
| // primary root Window. |
| aura::Window* CurrentContext(); |
| |
| PrefService* GetLocalStatePrefService(); |
| |
| TestShellDelegate* test_shell_delegate() { return test_shell_delegate_; } |
| void set_test_shell_delegate(TestShellDelegate* test_shell_delegate) { |
| test_shell_delegate_ = test_shell_delegate; |
| } |
| AshTestViewsDelegate* test_views_delegate() { |
| return test_views_delegate_.get(); |
| } |
| |
| display::Display GetSecondaryDisplay() const; |
| |
| TestSessionControllerClient* test_session_controller_client() { |
| return session_controller_client_.get(); |
| } |
| void set_test_session_controller_client( |
| std::unique_ptr<TestSessionControllerClient> session_controller_client) { |
| session_controller_client_ = std::move(session_controller_client); |
| } |
| TestNotifierSettingsController* notifier_settings_controller() { |
| return notifier_settings_controller_.get(); |
| } |
| TestSystemTrayClient* system_tray_client() { |
| return system_tray_client_.get(); |
| } |
| TestPrefServiceProvider* prefs_provider() { return prefs_provider_.get(); } |
| |
| AppListTestHelper* app_list_test_helper() { |
| return app_list_test_helper_.get(); |
| } |
| |
| TestKeyboardControllerObserver* test_keyboard_controller_observer() { |
| return test_keyboard_controller_observer_.get(); |
| } |
| |
| void reset_commandline() { command_line_.reset(); } |
| |
| private: |
| // Called when running in ash to create Shell. |
| void CreateShell(bool provide_local_state, |
| base::Optional<ShellInitParams> init_params); |
| |
| std::unique_ptr<chromeos::system::ScopedFakeStatisticsProvider> |
| statistics_provider_; |
| |
| TestShellDelegate* test_shell_delegate_ = nullptr; // Owned by ash::Shell. |
| std::unique_ptr<ui::ScopedAnimationDurationScaleMode> zero_duration_mode_; |
| |
| std::unique_ptr<::wm::WMState> wm_state_; |
| std::unique_ptr<AshTestViewsDelegate> test_views_delegate_; |
| |
| // Flags for whether various services were initialized here. |
| bool bluez_dbus_manager_initialized_ = false; |
| bool power_policy_controller_initialized_ = false; |
| |
| std::unique_ptr<TestSessionControllerClient> session_controller_client_; |
| std::unique_ptr<TestNotifierSettingsController> notifier_settings_controller_; |
| std::unique_ptr<TestSystemTrayClient> system_tray_client_; |
| std::unique_ptr<TestPrefServiceProvider> prefs_provider_; |
| std::unique_ptr<TestAssistantService> assistant_service_; |
| std::unique_ptr<ui::TestContextFactories> context_factories_; |
| |
| std::unique_ptr<base::test::ScopedCommandLine> command_line_; |
| |
| std::unique_ptr<AppListTestHelper> app_list_test_helper_; |
| |
| std::unique_ptr<TestNewWindowDelegate> new_window_delegate_; |
| |
| std::unique_ptr<TestKeyboardControllerObserver> |
| test_keyboard_controller_observer_; |
| |
| std::unique_ptr<PrefService> local_state_; |
| |
| DISALLOW_COPY_AND_ASSIGN(AshTestHelper); |
| }; |
| |
| } // namespace ash |
| |
| #endif // ASH_TEST_ASH_TEST_HELPER_H_ |