Moves FrameDecorations 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/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_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 {
  public:
   WindowManagerFactoryRegistryTestApi(WindowManagerFactoryRegistry* registry);
   ~WindowManagerFactoryRegistryTestApi();
 
   void AddService(const UserId& user_id, mojom::WindowManagerFactory* factory);
 
  private:
   WindowManagerFactoryRegistry* registry_;
 
   DISALLOW_COPY_AND_ASSIGN(WindowManagerFactoryRegistryTestApi);
 };
 
 // -----------------------------------------------------------------------------
 
+class UserDisplayManagerTestApi {
+ public:
+  explicit UserDisplayManagerTestApi(UserDisplayManager* udm) : udm_(udm) {}
+  ~UserDisplayManagerTestApi() {}
+
+  void SetTestObserver(mojom::DisplayManagerObserver* observer) {
+    udm_->test_observer_ = observer;
+    if (observer)
+      udm_->OnObserverAdded(observer);
+  }
+
+ private:
+  UserDisplayManager* udm_;
+
+  DISALLOW_COPY_AND_ASSIGN(UserDisplayManagerTestApi);
+};
+
+// -----------------------------------------------------------------------------
+
 class WindowTreeTestApi {
  public:
   WindowTreeTestApi(WindowTree* tree);
   ~WindowTreeTestApi();
 
   void set_window_manager_internal(mojom::WindowManager* wm_internal) {
     tree_->window_manager_internal_ = wm_internal;
   }
 
  private:
@@ skipping 181 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_