DesktopAura: Track windows "owned" via the DesktopWindowTreeHost

chrome/test/base/view_event_test_platform_part_aura.cc

 // Copyright 2014 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 <memory>
 
 #include "base/macros.h"
+#include "base/memory/ptr_util.h"
 #include "chrome/test/base/view_event_test_platform_part.h"
 #include "ui/aura/env.h"
 #include "ui/display/screen.h"
+#include "ui/views/test/platform_test_helper.h"
 #include "ui/views/widget/desktop_aura/desktop_screen.h"
+#include "ui/wm/core/wm_state.h"
 
-#if !defined(OS_CHROMEOS) && defined(OS_LINUX)
+#if defined(USE_X11)
 #include "ui/views/test/test_desktop_screen_x11.h"
 #endif
 
 namespace {
 
-// ViewEventTestPlatformPart implementation for Views, but non-CrOS.
-class ViewEventTestPlatformPartDefault : public ViewEventTestPlatformPart {
+// ViewEventTestPlatformPart implementation for Aura Views, but non-CrOS.
+class ViewEventTestPlatformPartAura : public ViewEventTestPlatformPart {
  public:
-  ViewEventTestPlatformPartDefault(
+  ViewEventTestPlatformPartAura(
       ui::ContextFactory* context_factory,
       ui::ContextFactoryPrivate* context_factory_private) {
-#if defined(USE_AURA)
     DCHECK(!display::Screen::GetScreen());
-#if defined(USE_X11) && !defined(OS_CHROMEOS)
+#if defined(USE_X11)
     display::Screen::SetScreenInstance(
         views::test::TestDesktopScreenX11::GetInstance());
 #else
     screen_.reset(views::CreateDesktopScreen());
     display::Screen::SetScreenInstance(screen_.get());
 #endif
     env_ = aura::Env::CreateInstance();
     env_->set_context_factory(context_factory);
     env_->set_context_factory_private(context_factory_private);
-#endif
+
+    // See AuraTestHelper::SetUp().
+    if (!views::PlatformTestHelper::IsMus())
+      wm_state_ = base::MakeUnique<wm::WMState>();

[tapted, 2017/01/23 10:52:55]

Some menu-related interactive_ui_tests fail with `Check failed:
TransientWindowStackingClient::instance_` without this. (Usually AuraTestHelper
adds this, but ViewEventTestBase doesn't use that).

   }
 
-  ~ViewEventTestPlatformPartDefault() override {
-#if defined(USE_AURA)
+  ~ViewEventTestPlatformPartAura() override {
+    wm_state_.reset();
     env_.reset();
     display::Screen::SetScreenInstance(nullptr);
-#endif
   }
 
-  // Overridden from ViewEventTestPlatformPart:
-  gfx::NativeWindow GetContext() override { return NULL; }
+  // ViewEventTestPlatformPart:
+  gfx::NativeWindow GetContext() override { return nullptr; }
 
  private:
   std::unique_ptr<display::Screen> screen_;
   std::unique_ptr<aura::Env> env_;
+  std::unique_ptr<wm::WMState> wm_state_;
 
-  DISALLOW_COPY_AND_ASSIGN(ViewEventTestPlatformPartDefault);
+  DISALLOW_COPY_AND_ASSIGN(ViewEventTestPlatformPartAura);
 };
 
 }  // namespace
 
 // static
 ViewEventTestPlatformPart* ViewEventTestPlatformPart::Create(
     ui::ContextFactory* context_factory,
     ui::ContextFactoryPrivate* context_factory_private) {
-  return new ViewEventTestPlatformPartDefault(context_factory,
-                                              context_factory_private);
+  return new ViewEventTestPlatformPartAura(context_factory,
+                                           context_factory_private);
 }