Make aura work with mus

ui/aura/test/test_screen.cc

 // Copyright (c) 2012 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 "ui/aura/test/test_screen.h"
 
 #include <stdint.h>
 
 #include "base/logging.h"
 #include "ui/aura/env.h"
+#include "ui/aura/mus/window_tree_client.h"
+#include "ui/aura/test/mus/window_tree_client_private.h"
 #include "ui/aura/window.h"
 #include "ui/aura/window_event_dispatcher.h"
 #include "ui/aura/window_tree_host.h"
 #include "ui/base/ime/input_method.h"
 #include "ui/display/screen.h"
 #include "ui/gfx/geometry/rect_conversions.h"
 #include "ui/gfx/geometry/size_conversions.h"
 #include "ui/gfx/native_widget_types.h"
 
 namespace aura {
 
 namespace {
 
 bool IsRotationPortrait(display::Display::Rotation rotation) {
   return rotation == display::Display::ROTATE_90 ||
          rotation == display::Display::ROTATE_270;
 }
 
 }  // namespace
 
 // static
-TestScreen* TestScreen::Create(const gfx::Size& size) {
+TestScreen* TestScreen::Create(const gfx::Size& size,
+                               WindowTreeClient* window_tree_client) {
   const gfx::Size kDefaultSize(800, 600);
   // Use (0,0) because the desktop aura tests are executed in
   // native environment where the display's origin is (0,0).
-  return new TestScreen(gfx::Rect(size.IsEmpty() ? kDefaultSize : size));
+  return new TestScreen(gfx::Rect(size.IsEmpty() ? kDefaultSize : size),
+                        window_tree_client);
 }
 
 TestScreen::~TestScreen() {
 }
 
 WindowTreeHost* TestScreen::CreateHostForPrimaryDisplay() {
   DCHECK(!host_);
-  host_ = WindowTreeHost::Create(gfx::Rect(display_.GetSizeInPixel()));
+  if (window_tree_client_) {
+    host_ = WindowTreeClientPrivate(window_tree_client_)
+                .CallWmNewDisplayAdded(display_);
+  } else {
+    host_ = WindowTreeHost::Create(gfx::Rect(display_.GetSizeInPixel()));
+  }
   // Some tests don't correctly manage window focus/activation states.
   // Makes sure InputMethod is default focused so that IME basics can work.
   host_->GetInputMethod()->OnFocus();
   host_->window()->AddObserver(this);
   host_->InitHost();
   return host_;
 }
 
 void TestScreen::SetDeviceScaleFactor(float device_scale_factor) {
   gfx::Rect bounds_in_pixel(display_.GetSizeInPixel());
@@ skipping 105 matching lines @@
 
 display::Display TestScreen::GetDisplayMatching(
     const gfx::Rect& match_rect) const {
   return display_;
 }
 
 void TestScreen::AddObserver(display::DisplayObserver* observer) {}
 
 void TestScreen::RemoveObserver(display::DisplayObserver* observer) {}
 
-TestScreen::TestScreen(const gfx::Rect& screen_bounds)
-    : host_(NULL),
-      ui_scale_(1.0f) {
+TestScreen::TestScreen(const gfx::Rect& screen_bounds,
+                       WindowTreeClient* window_tree_client)
+    : host_(nullptr), ui_scale_(1.0f), window_tree_client_(window_tree_client) {
   static int64_t synthesized_display_id = 2000;
   display_.set_id(synthesized_display_id++);
   display_.SetScaleAndBounds(1.0f, screen_bounds);
 }
 
 }  // namespace aura