DesktopAura: Track windows "owned" via the DesktopWindowTreeHost

ui/views/widget/widget_unittest.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 <algorithm>
 #include <memory>
 #include <set>
 
 #include "base/bind.h"
 #include "base/macros.h"
@@ skipping 2972 matching lines @@
   EXPECT_EQ(expected.size(), child_widgets.size());
   EXPECT_TRUE(
       std::equal(expected.begin(), expected.end(), child_widgets.begin()));
 
   // Check GetAllOwnedWidgets(). On Aura, this includes "transient" children.
   // Otherwise (on all platforms), it should be the same as GetAllChildWidgets()
   // except the root Widget is not included.
   EXPECT_EQ(1u, expected.erase(toplevel.get()));
 
   std::set<Widget*> owned_widgets;
-  Widget::GetAllOwnedWidgets(toplevel->GetNativeView(), &owned_widgets);
+  Widget::GetAllOwnedWidgets(toplevel->GetNativeView(), &owned_widgets, true);
 
   EXPECT_EQ(expected.size(), owned_widgets.size());
   EXPECT_TRUE(
       std::equal(expected.begin(), expected.end(), owned_widgets.begin()));
 }
 
 // Used by DestroyChildWidgetsInOrder. On destruction adds the supplied name to
 // a vector.
 class DestroyedTrackingView : public View {
  public:
@@ skipping 79 matching lines @@
        DestroyChildWidgetsInOrderWithDesktopNativeWidgetForBoth) {
   RunDestroyChildWidgetsTest(true, true);
 }
 #endif  // !defined(OS_CHROMEOS)
 
 // See description of RunDestroyChildWidgetsTest().
 TEST_F(WidgetChildDestructionTest, DestroyChildWidgetsInOrder) {
   RunDestroyChildWidgetsTest(false, false);
 }
 
+Widget* TrackedTopLevelWidgetWithParent(Widget::InitParams params,
+                                        bool* destroyed,
+                                        Widget* parent) {
+  Widget* widget = new Widget();  // Owned by Native Widget.
+
+  params.bounds = gfx::Rect(100, 100, 100, 100);
+  params.parent = parent ? parent->GetNativeView() : nullptr;
+  params.native_widget =
+      CreatePlatformDesktopNativeWidgetImpl(params, widget, destroyed);
+  widget->Init(params);
+  widget->Show();
+  return widget;
+}
+
+// Test Widget::GetAllOwnedWidgets() for top-level child windows.
+// Also ensure that if a Widget is created with parent and is created using a
+// top-level Widget, then it is destroyed when the parent Widget goes away.
+TEST_F(WidgetTest, OwnedTopLevelChildWindows) {
+  Widget::InitParams params = CreateParams(Widget::InitParams::TYPE_WINDOW);
+
+  bool parent_destroyed = false;
+  bool child_destroyed = false;
+  bool grandchild_destroyed = false;
+
+  Widget* parent =
+      TrackedTopLevelWidgetWithParent(params, &parent_destroyed, nullptr);
+  Widget* child =
+      TrackedTopLevelWidgetWithParent(params, &child_destroyed, parent);
+  Widget* grandchild =
+      TrackedTopLevelWidgetWithParent(params, &grandchild_destroyed, child);
+
+  // |child| and |grandchild| are destroyed when their parents go away, so they
+  // should be considered "owned".
+  std::set<Widget*> owned_widgets;
+  Widget::GetAllOwnedWidgets(parent->GetNativeView(), &owned_widgets, true);
+  EXPECT_EQ(0u, owned_widgets.count(parent));
+  EXPECT_EQ(1u, owned_widgets.count(child));
+  EXPECT_EQ(1u, owned_widgets.count(grandchild));
+
+  owned_widgets.clear();
+  Widget::GetAllOwnedWidgets(child->GetNativeView(), &owned_widgets, true);
+  EXPECT_EQ(0u, owned_widgets.count(parent));
+  EXPECT_EQ(0u, owned_widgets.count(child));
+  EXPECT_EQ(1u, owned_widgets.count(grandchild));
+
+  // Trigger child destruction via a call to one of the following on the parent
+  // window: DesktopWindowTreeHostX11::CloseNow(), ::DestroyWindow(),
+  // DesktopWindowTreeHostMus::CloseNow(), or BridgedNativeWidget::
+  // OnWindowWillClose().
+  parent->CloseNow();
+  EXPECT_TRUE(parent_destroyed);
+  EXPECT_TRUE(child_destroyed);
+  EXPECT_TRUE(grandchild_destroyed);
+}
+
 // Verifies nativeview visbility matches that of Widget visibility when
 // SetFullscreen is invoked.
 TEST_F(WidgetTest, FullscreenStatePropagated) {
   Widget::InitParams init_params =
       CreateParams(Widget::InitParams::TYPE_WINDOW);
   init_params.show_state = ui::SHOW_STATE_NORMAL;
   init_params.bounds = gfx::Rect(0, 0, 500, 500);
   init_params.ownership = Widget::InitParams::WIDGET_OWNS_NATIVE_WIDGET;
 
   Widget top_level_widget;
@@ skipping 769 matching lines @@
 }
 
 TEST_F(CompositingWidgetTest, Transparency_DesktopWidgetTranslucent) {
   CheckAllWidgetsForOpacity(Widget::InitParams::TRANSLUCENT_WINDOW);
 }
 
 #endif  // !defined(OS_CHROMEOS)
 
 }  // namespace test
 }  // namespace views