Merge Prevent nested Menu Cancelling

ui/views/controls/menu/menu_runner_unittest.cc

 // Copyright 2015 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/views/controls/menu/menu_runner.h"
 
 #include <stdint.h>
 
 #include <memory>
 
 #include "base/macros.h"
 #include "base/memory/ptr_util.h"
 #include "base/strings/utf_string_conversions.h"
 #include "ui/base/ui_base_types.h"
 #include "ui/events/test/event_generator.h"
 #include "ui/views/controls/menu/menu_controller.h"
 #include "ui/views/controls/menu/menu_delegate.h"
 #include "ui/views/controls/menu/menu_item_view.h"
 #include "ui/views/controls/menu/menu_runner_impl.h"
 #include "ui/views/controls/menu/menu_types.h"
 #include "ui/views/controls/menu/submenu_view.h"
 #include "ui/views/test/menu_test_utils.h"
 #include "ui/views/test/test_views.h"
+#include "ui/views/test/test_views_delegate.h"
 #include "ui/views/test/views_test_base.h"
 #include "ui/views/widget/native_widget_private.h"
 #include "ui/views/widget/widget.h"
 
+namespace {
+
+// Accepts a MenuRunnerImpl to release when this is. Simulates shutdown
+// occurring immediately during the release of ViewsDelegate.
+class DeletingTestViewsDelegate : public views::TestViewsDelegate {
+ public:
+  DeletingTestViewsDelegate() : menu_runner_(nullptr) {}
+  ~DeletingTestViewsDelegate() override{};
+
+  void set_menu_runner(views::internal::MenuRunnerImpl* menu_runner) {
+    menu_runner_ = menu_runner;
+  }
+
+  // views::ViewsDelegate:
+  void ReleaseRef() override;
+
+ private:
+  // Not owned, deletes itself.
+  views::internal::MenuRunnerImpl* menu_runner_;
+
+  DISALLOW_COPY_AND_ASSIGN(DeletingTestViewsDelegate);
+};
+
+void DeletingTestViewsDelegate::ReleaseRef() {
+  if (menu_runner_)
+    menu_runner_->Release();
+}
+
+}  // namespace
+
 namespace views {
 namespace test {
 
 class MenuRunnerTest : public ViewsTestBase {
  public:
   MenuRunnerTest();
   ~MenuRunnerTest() override;
 
   // Initializes a MenuRunner with |run_types|. It takes ownership of
   // |menu_item_view_|.
@@ skipping 375 matching lines @@
   // controller that it created.
   menu_runner->Release();
 
   // This is not expected to run, however this is from the origin ASAN stack
   // traces. So regressions will be caught with the same stack trace.
   if (menu_controller.controller())
     menu_controller.controller()->CancelAll();
   EXPECT_EQ(nullptr, menu_controller.controller());
 }
 
+// Test class which overrides the ViewsDelegate. Allowing to simulate shutdown
+// during its release.
+class MenuRunnerDestructionTest : public MenuRunnerTest {
+ public:
+  MenuRunnerDestructionTest() {}
+  ~MenuRunnerDestructionTest() override {}
+
+  DeletingTestViewsDelegate* views_delegate() { return views_delegate_; }
+
+  base::WeakPtr<internal::MenuRunnerImpl> MenuRunnerAsWeakPtr(
+      internal::MenuRunnerImpl* menu_runner);
+
+  // ViewsTestBase:
+  void SetUp() override;
+
+ private:
+  // Not owned
+  DeletingTestViewsDelegate* views_delegate_;
+
+  DISALLOW_COPY_AND_ASSIGN(MenuRunnerDestructionTest);
+};
+
+base::WeakPtr<internal::MenuRunnerImpl>
+MenuRunnerDestructionTest::MenuRunnerAsWeakPtr(
+    internal::MenuRunnerImpl* menu_runner) {
+  return menu_runner->weak_factory_.GetWeakPtr();
+}
+
+void MenuRunnerDestructionTest::SetUp() {
+  std::unique_ptr<DeletingTestViewsDelegate> views_delegate(
+      new DeletingTestViewsDelegate);
+  views_delegate_ = views_delegate.get();
+  set_views_delegate(std::move(views_delegate));
+  MenuRunnerTest::SetUp();
+}
+
+// Tests that when ViewsDelegate is released that a nested Cancel of the
+// MenuRunner does not occur.
+TEST_F(MenuRunnerDestructionTest, MenuRunnerDestroyedDuringReleaseRef) {
+  internal::MenuRunnerImpl* menu_runner =
+      new internal::MenuRunnerImpl(menu_item_view());
+  EXPECT_EQ(MenuRunner::NORMAL_EXIT,
+            menu_runner->RunMenuAt(owner(), nullptr, gfx::Rect(),
+                                   MENU_ANCHOR_TOPLEFT, MenuRunner::ASYNC));
+
+  views_delegate()->set_menu_runner(menu_runner);
+
+  base::WeakPtr<internal::MenuRunnerImpl> ref(MenuRunnerAsWeakPtr(menu_runner));
+  MenuControllerTestApi menu_controller;
+  // This will release the ref on ViewsDelegate. The test version will release
+  // |menu_runner| simulating device shutdown.
+  menu_controller.controller()->CancelAll();
+  // Both the |menu_runner| and |menu_controller| should have been deleted.
+  EXPECT_EQ(nullptr, ref);
+  EXPECT_EQ(nullptr, menu_controller.controller());
+}
+
 }  // namespace test
 }  // namespace views