Moves EventDispatcher from Display to WindowManagerState

components/mus/ws/test_utils.h

 // Copyright 2016 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 COMPONENTS_MUS_WS_TEST_UTILS_H_
 #define COMPONENTS_MUS_WS_TEST_UTILS_H_
 
 #include <stdint.h>
 
 #include "components/mus/public/interfaces/display.mojom.h"
 #include "components/mus/public/interfaces/window_tree.mojom.h"
 #include "components/mus/ws/connection_manager_delegate.h"
 #include "components/mus/ws/display.h"
+#include "components/mus/ws/event_dispatcher.h"
 #include "components/mus/ws/platform_display.h"
 #include "components/mus/ws/platform_display_factory.h"
 #include "components/mus/ws/test_change_tracker.h"
 #include "components/mus/ws/user_display_manager.h"
 #include "components/mus/ws/window_manager_factory_registry.h"
+#include "components/mus/ws/window_manager_state.h"
 #include "components/mus/ws/window_tree.h"
 #include "components/mus/ws/window_tree_binding.h"
 
 namespace mus {
 namespace ws {
 namespace test {
 
 // Collection of utilities useful in creating mus tests.
 
 class WindowManagerFactoryRegistryTestApi {
@@ skipping 32 matching lines @@
 
 class WindowTreeTestApi {
  public:
   WindowTreeTestApi(WindowTree* tree);
   ~WindowTreeTestApi();
 
   void set_window_manager_internal(mojom::WindowManager* wm_internal) {
     tree_->window_manager_internal_ = wm_internal;
   }
 
+  void EnableCapture() { tree_->event_ack_id_ = 1u; }
+
  private:
   WindowTree* tree_;
 
   DISALLOW_COPY_AND_ASSIGN(WindowTreeTestApi);
 };
 
 // -----------------------------------------------------------------------------
 
 class DisplayTestApi {
  public:
   explicit DisplayTestApi(Display* display);
   ~DisplayTestApi();
 
   void OnEvent(const ui::Event& event) { display_->OnEvent(event); }
 
-  mojom::WindowTree* tree_awaiting_input_ack() {
-    return display_->tree_awaiting_input_ack_;
-  }
-
  private:
   Display* display_;
 
   DISALLOW_COPY_AND_ASSIGN(DisplayTestApi);
 };
 
 // -----------------------------------------------------------------------------
+
+class EventDispatcherTestApi {
+ public:
+  explicit EventDispatcherTestApi(EventDispatcher* ed) : ed_(ed) {}
+  ~EventDispatcherTestApi() {}
+
+  bool AreAnyPointersDown() const { return ed_->AreAnyPointersDown(); }
+  bool is_mouse_button_down() const { return ed_->mouse_button_down_; }
+  bool IsObservingWindow(ServerWindow* window) {
+    return ed_->IsObservingWindow(window);
+  }
+  int NumberPointerTargetsForWindow(ServerWindow* window);
+
+ private:
+  EventDispatcher* ed_;
+
+  DISALLOW_COPY_AND_ASSIGN(EventDispatcherTestApi);
+};
+
+// -----------------------------------------------------------------------------
+
+class WindowManagerStateTestApi {
+ public:
+  explicit WindowManagerStateTestApi(WindowManagerState* wms) : wms_(wms) {}
+  ~WindowManagerStateTestApi() {}
+
+  mojom::WindowTree* tree_awaiting_input_ack() {
+    return wms_->tree_awaiting_input_ack_;
+  }
+
+ private:
+  WindowManagerState* wms_;
+
+  DISALLOW_COPY_AND_ASSIGN(WindowManagerStateTestApi);
+};
+
+// -----------------------------------------------------------------------------
 
 // Factory that dispenses TestPlatformDisplays.
 class TestPlatformDisplayFactory : public PlatformDisplayFactory {
  public:
   explicit TestPlatformDisplayFactory(int32_t* cursor_id_storage);
   ~TestPlatformDisplayFactory();
 
   // PlatformDisplayFactory:
   PlatformDisplay* CreatePlatformDisplay(
       mojo::Connector* connector,
@@ skipping 146 matching lines @@
   bool got_on_no_more_displays_ = false;
 
   DISALLOW_COPY_AND_ASSIGN(TestConnectionManagerDelegate);
 };
 
 }  // namespace test
 }  // namespace ws
 }  // namespace mus
 
 #endif  // COMPONENTS_MUS_WS_TEST_UTILS_H_