Changes how window manager obtains WindowTree from mus

components/mus/public/cpp/tests/window_tree_client_private.h

Index: components/mus/public/cpp/tests/window_tree_client_private.h
diff --git a/components/mus/public/cpp/tests/window_tree_client_private.h b/components/mus/public/cpp/tests/window_tree_client_private.h
index 7cec901b36071e80286c757c64ebedd12f2393cb..0612bbf721af6120967abc12679c838c4f68c570 100644
--- a/components/mus/public/cpp/tests/window_tree_client_private.h
+++ b/components/mus/public/cpp/tests/window_tree_client_private.h
@@ -10,6 +10,11 @@
 #include <memory>
 
 #include "base/macros.h"
+#include "components/mus/common/types.h"
+
+namespace display {
+class Display;
+}
 
 namespace ui {
 class Event;
@@ -35,9 +40,15 @@ class WindowTreeClientPrivate {
   // Calls OnEmbed() on the WindowTreeClient.
   void OnEmbed(mojom::WindowTree* window_tree);
 
+  void CallWmNewDisplayAdded(const display::Display& display);
+
   // Pretends that |event| has been received from the window server.
   void CallOnWindowInputEvent(Window* window, std::unique_ptr<ui::Event> event);
 
+  // Sets the WindowTree and client id.
+  void SetTreeAndClientId(mojom::WindowTree* window_tree,
+                          ClientSpecificId client_id);
+
  private:
    WindowTreeClient* tree_client_impl_;