ui/wm/test/wm_test_helper.h

// 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_WM_TEST_WM_TEST_HELPER_H_
#define UI_WM_TEST_WM_TEST_HELPER_H_

#include <memory>

#include "base/compiler_specific.h"
#include "base/macros.h"
#include "base/run_loop.h"
#include "services/ws/test_ws/test_ws.mojom.h"
#include "ui/aura/client/window_parenting_client.h"
#include "ui/aura/mus/window_tree_client_delegate.h"
#include "ui/aura/window_tree_host.h"

namespace aura {
class PropertyConverter;
class TestScreen;
class Window;
class WindowTreeClient;
class WindowTreeHost;
namespace client {
class DefaultCaptureClient;
class FocusClient;
}
}

namespace gfx {
class Rect;
class Size;
}

namespace service_manager {
class Connector;
}

namespace ui {
class ContextFactory;
}

namespace ws {
class InputDeviceClient;
}  // namespace ws

namespace wm {

class CompoundEventFilter;
class WMState;

// Creates a minimal environment for running the shell. We can't pull in all of
// ash here, but we can create and attach several of the same things we'd find
// in ash.
class WMTestHelper : public aura::client::WindowParentingClient,
                     public aura::WindowTreeClientDelegate {
 public:
  WMTestHelper(const gfx::Size& default_window_size,
               service_manager::Connector* coonnector,
               ui::ContextFactory* context_factory);
  ~WMTestHelper() override;

  aura::WindowTreeHost* host() { return host_.get(); }

  // Overridden from client::WindowParentingClient:
  aura::Window* GetDefaultParent(aura::Window* window,
                                 const gfx::Rect& bounds) override;

 private:
  // Used when aura is running in Mode::LOCAL.
  void InitLocalHost(const gfx::Size& default_window_size);

  // Used when aura is running in Mode::MUS.
  void InitMusHost(service_manager::Connector* connector,
                   const gfx::Size& default_window_size);

  // aura::WindowTreeClientDelegate:
  void OnEmbed(
      std::unique_ptr<aura::WindowTreeHostMus> window_tree_host) override;
  void OnUnembed(aura::Window* root) override;
  void OnEmbedRootDestroyed(aura::WindowTreeHostMus* window_tree_host) override;
  void OnLostConnection(aura::WindowTreeClient* client) override;
  aura::PropertyConverter* GetPropertyConverter() override;
  void OnDisplaysChanged(std::vector<ws::mojom::WsDisplayPtr> ws_displays,
                         int64_t primary_display_id,
                         int64_t internal_display_id,
                         int64_t display_id_for_new_windows) override;

  std::unique_ptr<WMState> wm_state_;
  std::unique_ptr<aura::TestScreen> test_screen_;
  std::unique_ptr<ws::InputDeviceClient> input_device_client_;
  std::unique_ptr<aura::PropertyConverter> property_converter_;
  std::unique_ptr<aura::WindowTreeClient> window_tree_client_;
  std::unique_ptr<aura::WindowTreeHost> host_;
  std::unique_ptr<wm::CompoundEventFilter> root_window_event_filter_;
  std::unique_ptr<aura::client::DefaultCaptureClient> capture_client_;
  std::unique_ptr<aura::client::FocusClient> focus_client_;

  // Loop to wait for |host_| gets embedded under mus.
  std::unique_ptr<base::RunLoop> display_wait_loop_;

  test_ws::mojom::TestWsPtr test_ws_;

  DISALLOW_COPY_AND_ASSIGN(WMTestHelper);
};

}  // namespace wm

#endif  // UI_WM_TEST_WM_TEST_HELPER_H_