| // 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 UI_AURA_TEST_ENV_TEST_HELPER_H_ |
| #define UI_AURA_TEST_ENV_TEST_HELPER_H_ |
| |
| #include <utility> |
| |
| #include "base/macros.h" |
| #include "ui/aura/env.h" |
| #include "ui/aura/env_input_state_controller.h" |
| #include "ui/aura/input_state_lookup.h" |
| #include "ui/events/gestures/gesture_recognizer.h" |
| |
| namespace aura { |
| namespace test { |
| |
| // Used to set the WindowTreeClient of Env. The constructor installs the |
| // supplied WindowTreeClient and the destructor restores the WindowTreeClient |
| // to what it previously was. |
| class EnvWindowTreeClientSetter { |
| public: |
| explicit EnvWindowTreeClientSetter(WindowTreeClient* client); |
| ~EnvWindowTreeClientSetter(); |
| |
| private: |
| void SetWindowTreeClient(WindowTreeClient* client); |
| |
| WindowTreeClient* previous_client_; |
| |
| DISALLOW_COPY_AND_ASSIGN(EnvWindowTreeClientSetter); |
| }; |
| |
| class EnvTestHelper { |
| public: |
| EnvTestHelper() : EnvTestHelper(Env::GetInstance()) {} |
| explicit EnvTestHelper(Env* env) : env_(env) {} |
| ~EnvTestHelper() {} |
| |
| void SetInputStateLookup( |
| std::unique_ptr<InputStateLookup> input_state_lookup) { |
| env_->input_state_lookup_ = std::move(input_state_lookup); |
| } |
| |
| void ResetEventState() { |
| env_->mouse_button_flags_ = 0; |
| env_->is_touch_down_ = false; |
| env_->last_mouse_location_ = gfx::Point(); |
| env_->env_controller_->touch_ids_down_ = 0; |
| } |
| |
| // Changes Env's Mode to |mode|, returning the old value. |
| Env::Mode SetMode(Env::Mode mode) { |
| const Env::Mode old_mode = env_->mode_; |
| env_->mode_ = mode; |
| if (mode == Env::Mode::MUS) |
| env_->EnableMusOSExchangeDataProvider(); |
| env_->in_mus_shutdown_ = false; |
| return old_mode; |
| } |
| |
| void SetInMusShutdown(bool value) { env_->in_mus_shutdown_ = value; } |
| |
| // Use to force Env::last_mouse_location() to return the value last set. |
| // This matters for MUS, which may not return the last explicitly set |
| // location. |
| void SetAlwaysUseLastMouseLocation(bool value) { |
| env_->always_use_last_mouse_location_ = value; |
| } |
| |
| // Reset aura::Env to eliminate potential test dependency. |
| // (https://crbug.com/586514) |
| void ResetEnvForTesting() { env_->is_touch_down_ = false; } |
| |
| void SetGestureRecognizer( |
| std::unique_ptr<ui::GestureRecognizer> gesture_recognizer) { |
| env_->gesture_recognizer_ = std::move(gesture_recognizer); |
| } |
| |
| WindowTreeClient* GetWindowTreeClient() { return env_->window_tree_client_; } |
| |
| private: |
| Env* env_; |
| |
| DISALLOW_COPY_AND_ASSIGN(EnvTestHelper); |
| }; |
| |
| } // namespace test |
| } // namespace aura |
| |
| #endif // UI_AURA_TEST_ENV_TEST_HELPER_H_ |