[views-mus] Prevent creating a native OzonePlatform in mus tests.

ui/aura/test/aura_test_helper.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/aura_test_helper.h"
 
+#include "base/command_line.h"
 #include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "ui/aura/client/default_capture_client.h"
 #include "ui/aura/client/focus_client.h"
 #include "ui/aura/env.h"
 #include "ui/aura/input_state_lookup.h"
 #include "ui/aura/mus/capture_synchronizer.h"
 #include "ui/aura/mus/focus_synchronizer.h"
 #include "ui/aura/mus/window_port_mus.h"
@@ skipping 21 matching lines @@
 #if defined(USE_X11)
 #include "ui/base/x/x11_util.h"  // nogncheck
 #endif
 
 namespace aura {
 namespace test {
 namespace {
 
 AuraTestHelper* g_instance = nullptr;
 
+// Returns true if running inside of mus. Checks for mojo specific flag.
+bool RunningInsideMus() {
+  return base::CommandLine::ForCurrentProcess()->HasSwitch(
+      "primordial-pipe-token");

[sky, 2017/04/10 23:53:26]

Use kPrimordialPipeToken?

[Elliot Glaysher, 2017/04/12 18:09:16]

Removed.

+}
+
 }  // namespace
 
 AuraTestHelper::AuraTestHelper(base::MessageLoopForUI* message_loop)
     : setup_called_(false), teardown_called_(false) {
   DCHECK(message_loop);
   // Disable animations during tests.
   zero_duration_mode_.reset(new ui::ScopedAnimationDurationScaleMode(
       ui::ScopedAnimationDurationScaleMode::ZERO_DURATION));
   ui::test::EnableTestConfigForPlatformWindows();
   InitializeAuraEventGeneratorDelegate();
@@ skipping 57 matching lines @@
   env_helper.SetAlwaysUseLastMouseLocation(true);
   Env::GetInstance()->set_context_factory(context_factory);
   Env::GetInstance()->set_context_factory_private(context_factory_private);
   // Unit tests generally don't want to query the system, rather use the state
   // from RootWindow.
   env_helper.SetInputStateLookup(nullptr);
   env_helper.ResetEventState();
 
   ui::InitializeInputMethodForTesting();
 
-  display::Screen* screen = display::Screen::GetScreen();
-  gfx::Size host_size(screen ? screen->GetPrimaryDisplay().GetSizeInPixel()
-                             : gfx::Size(800, 600));
-  // TODO(sky): creating the screen and host should not happen for mus.
-  test_screen_.reset(TestScreen::Create(host_size));
-  if (!screen)
-    display::Screen::SetScreenInstance(test_screen_.get());
-  host_.reset(test_screen_->CreateHostForPrimaryDisplay());
-  host_->window()->SetEventTargeter(
-      std::unique_ptr<ui::EventTargeter>(new WindowTargeter()));
+  if (!RunningInsideMus()) {
+    display::Screen* screen = display::Screen::GetScreen();
+    gfx::Size host_size(screen ? screen->GetPrimaryDisplay().GetSizeInPixel()
+                               : gfx::Size(800, 600));
+    // TODO(sky): creating the screen and host should not happen for mus.

[sky, 2017/04/10 23:53:26]

You can remove this todo now.

[Elliot Glaysher, 2017/04/12 18:09:16]

Done.

+    test_screen_.reset(TestScreen::Create(host_size));
+    if (!screen)
+      display::Screen::SetScreenInstance(test_screen_.get());
+    host_.reset(test_screen_->CreateHostForPrimaryDisplay());
+    host_->window()->SetEventTargeter(
+        std::unique_ptr<ui::EventTargeter>(new WindowTargeter()));
 
-  client::SetFocusClient(root_window(), focus_client_.get());
-  client::SetCaptureClient(root_window(), capture_client());
-  parenting_client_.reset(new TestWindowParentingClient(root_window()));
+    client::SetFocusClient(root_window(), focus_client_.get());
+    client::SetCaptureClient(root_window(), capture_client());
+    parenting_client_.reset(new TestWindowParentingClient(root_window()));
 
-  root_window()->Show();
-  // Ensure width != height so tests won't confuse them.
-  host()->SetBoundsInPixels(gfx::Rect(host_size));
+    root_window()->Show();
+    // Ensure width != height so tests won't confuse them.
+    host()->SetBoundsInPixels(gfx::Rect(host_size));
+  }
 
   if (mode_ == Mode::MUS_CREATE_WINDOW_TREE_CLIENT) {
     window_tree_client_->focus_synchronizer()->SetActiveFocusClient(
         focus_client_.get(), root_window());
     window_tree()->AckAllChanges();
   }
 
   g_instance = this;
 }
 
 void AuraTestHelper::TearDown() {
   g_instance = nullptr;
   teardown_called_ = true;
   parenting_client_.reset();
-  client::SetFocusClient(root_window(), nullptr);
-  client::SetCaptureClient(root_window(), nullptr);
-  host_.reset();
+  if (!RunningInsideMus()) {
+    client::SetFocusClient(root_window(), nullptr);
+    client::SetCaptureClient(root_window(), nullptr);
+    host_.reset();
+
+    if (display::Screen::GetScreen() == test_screen_.get())
+      display::Screen::SetScreenInstance(nullptr);
+    test_screen_.reset();
+
+    window_tree_client_setup_.reset();
+    focus_client_.reset();
+    capture_client_.reset();
+  }
   ui::GestureRecognizer::Reset();
-  if (display::Screen::GetScreen() == test_screen_.get())
-    display::Screen::SetScreenInstance(nullptr);
-  test_screen_.reset();
-
-  window_tree_client_setup_.reset();
-  focus_client_.reset();
-  capture_client_.reset();
-
   ui::ShutdownInputMethodForTesting();
 
   if (env_)
     env_.reset();
   else
     EnvTestHelper().SetWindowTreeClient(nullptr);
   wm_state_.reset();
 }
 
 void AuraTestHelper::RunAllPendingInMessageLoop() {
@@ skipping 19 matching lines @@
   window_tree_client_setup_ = base::MakeUnique<TestWindowTreeClientSetup>();
   window_tree_client_setup_->InitForWindowManager(window_tree_delegate_,
                                                   window_manager_delegate_);
   window_tree_client_ = window_tree_client_setup_->window_tree_client();
   window_tree_client_->capture_synchronizer()->AttachToCaptureClient(
       capture_client_.get());
 }
 
 }  // namespace test
 }  // namespace aura