Changes how window manager obtains WindowTree from mus

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

 // 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.
 
 #include "components/mus/public/cpp/tests/window_tree_client_private.h"
 
 #include "components/mus/public/cpp/window.h"
 #include "components/mus/public/cpp/window_tree_client.h"
+#include "ui/display/display.h"
 
 namespace mus {
 
 WindowTreeClientPrivate::WindowTreeClientPrivate(
     WindowTreeClient* tree_client_impl)
     : tree_client_impl_(tree_client_impl) {}
 
 WindowTreeClientPrivate::WindowTreeClientPrivate(Window* window)
     : WindowTreeClientPrivate(window->window_tree()) {}
 
 WindowTreeClientPrivate::~WindowTreeClientPrivate() {}
 
 uint32_t WindowTreeClientPrivate::event_observer_id() {
   return tree_client_impl_->event_observer_id_;
 }
 
 void WindowTreeClientPrivate::OnEmbed(mojom::WindowTree* window_tree) {
   mojom::WindowDataPtr root_data(mojom::WindowData::New());
   root_data->parent_id = 0;
   root_data->window_id = 1;
   root_data->properties.SetToEmpty();
   root_data->visible = true;
   const int64_t display_id = 1;
   const Id focused_window_id = 0;
   tree_client_impl_->OnEmbedImpl(window_tree, 1, std::move(root_data),
                                  display_id, focused_window_id, true);
 }
 
+void WindowTreeClientPrivate::CallWmNewDisplayAdded(
+    const display::Display& display) {
+  mojom::WindowDataPtr root_data(mojom::WindowData::New());
+  root_data->parent_id = 0;
+  root_data->window_id = 1;
+  root_data->properties.SetToEmpty();
+  root_data->visible = true;
+  root_data->bounds = gfx::Rect(display.bounds().size());
+  const bool parent_drawn = true;
+  tree_client_impl_->WmNewDisplayAddedImpl(display, std::move(root_data),
+                                           parent_drawn);
+}
+
 void WindowTreeClientPrivate::CallOnWindowInputEvent(
     Window* window,
     std::unique_ptr<ui::Event> event) {
   const uint32_t event_id = 0u;
   const uint32_t observer_id = 0u;
   tree_client_impl_->OnWindowInputEvent(event_id, window->server_id(),
                                         std::move(event), observer_id);
 }
 
+void WindowTreeClientPrivate::SetTreeAndClientId(mojom::WindowTree* window_tree,
+                                                 ClientSpecificId client_id) {
+  tree_client_impl_->tree_ = window_tree;
+  tree_client_impl_->client_id_ = client_id;
+}
+
 }  // namespace mus