Added ink drop hover/ripple to menu hosting bookmark buttons.

ui/views/controls/button/custom_button_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 "ui/views/controls/button/custom_button.h"
 
 #include "base/macros.h"
 #include "build/build_config.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "ui/base/layout.h"
 #include "ui/base/material_design/material_design_controller.h"
 #include "ui/events/event_utils.h"
 #include "ui/events/test/event_generator.h"
 #include "ui/gfx/screen.h"
 #include "ui/views/animation/ink_drop_delegate.h"
 #include "ui/views/animation/ink_drop_host.h"
+#include "ui/views/animation/test/test_ink_drop_delegate.h"
 #include "ui/views/animation/test/test_ink_drop_host.h"
+#include "ui/views/context_menu_controller.h"
 #include "ui/views/controls/button/checkbox.h"
 #include "ui/views/controls/button/image_button.h"
 #include "ui/views/controls/button/label_button.h"
 #include "ui/views/controls/button/menu_button.h"
 #include "ui/views/controls/button/radio_button.h"
 #include "ui/views/controls/link.h"
 #include "ui/views/controls/textfield/textfield.h"
 #include "ui/views/test/views_test_base.h"
 
 #if defined(USE_AURA)
 #include "ui/aura/test/test_cursor_client.h"
 #include "ui/aura/window.h"
 #include "ui/aura/window_event_dispatcher.h"
 #endif
 
 namespace views {
 
+using test::TestInkDropDelegate;
+
 namespace {
 
+// No-op test double of a ContextMenuController.
+class TestContextMenuController : public ContextMenuController {
+ public:
+  TestContextMenuController() {}
+  ~TestContextMenuController() override {}
+
+  // ContextMenuController:
+  void ShowContextMenuForView(View* source,
+                              const gfx::Point& point,
+                              ui::MenuSourceType source_type) override {}
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(TestContextMenuController);
+};
+
 class TestCustomButton : public CustomButton, public ButtonListener {
  public:
   explicit TestCustomButton()
       : CustomButton(this) {
   }
 
   ~TestCustomButton() override {}
 
   void ButtonPressed(Button* sender, const ui::Event& event) override {
     pressed_ = true;
@@ skipping 12 matching lines @@
   }
 
  private:
   bool pressed_ = false;
   bool canceled_ = false;
 
   DISALLOW_COPY_AND_ASSIGN(TestCustomButton);
 };
 
 // An InkDropDelegate that keeps track of ink drop visibility.
-class TestInkDropDelegate : public InkDropDelegate {
+class TestInkDropDelegateThatTracksVisibilty : public InkDropDelegate {
  public:
-  TestInkDropDelegate(InkDropHost* ink_drop_host,
-                      bool* ink_shown,
-                      bool* ink_hidden)
-      : ink_drop_host_(ink_drop_host),
-        ink_shown_(ink_shown),
-        ink_hidden_(ink_hidden) {}
-  ~TestInkDropDelegate() override {}
+  TestInkDropDelegateThatTracksVisibilty(bool* ink_shown, bool* ink_hidden)
+      : ink_shown_(ink_shown), ink_hidden_(ink_hidden) {}
+  ~TestInkDropDelegateThatTracksVisibilty() override {}
 
   // InkDropDelegate:
   void OnAction(InkDropState state) override {
     switch (state) {
       case InkDropState::ACTION_PENDING:
       case InkDropState::SLOW_ACTION_PENDING:
       case InkDropState::ACTIVATED:
         *ink_shown_ = true;
         break;
       case InkDropState::HIDDEN:
         *ink_hidden_ = true;
         break;
       case InkDropState::QUICK_ACTION:
       case InkDropState::SLOW_ACTION:
       case InkDropState::DEACTIVATED:
         break;
     }
   }
 
+  void SnapToActivated() override { *ink_shown_ = true; }
+
   void SetHovered(bool is_hovered) override {}
 
  private:
-  InkDropHost* ink_drop_host_;
   bool* ink_shown_;
   bool* ink_hidden_;
 
-  DISALLOW_COPY_AND_ASSIGN(TestInkDropDelegate);
+  DISALLOW_COPY_AND_ASSIGN(TestInkDropDelegateThatTracksVisibilty);
 };
 
 // A test Button class that owns a TestInkDropDelegate.
 class TestButtonWithInkDrop : public TestCustomButton {
  public:
-  TestButtonWithInkDrop(bool* ink_shown, bool* ink_hidden)
-      : TestCustomButton(),
-        ink_drop_delegate_(
-            new TestInkDropDelegate(this, ink_shown, ink_hidden)) {
+  TestButtonWithInkDrop(scoped_ptr<InkDropDelegate> ink_drop_delegate)
+      : TestCustomButton(), ink_drop_delegate_(std::move(ink_drop_delegate)) {
     set_ink_drop_delegate(ink_drop_delegate_.get());
   }
   ~TestButtonWithInkDrop() override {}
 
  private:
   scoped_ptr<views::InkDropDelegate> ink_drop_delegate_;
 
   DISALLOW_COPY_AND_ASSIGN(TestButtonWithInkDrop);
 };
 
@@ skipping 18 matching lines @@
 
     button_ = new TestCustomButton();
     widget_->SetContentsView(button_);
   }
 
   void TearDown() override {
     widget_.reset();
     ViewsTestBase::TearDown();
   }
 
-  void CreateButtonWithInkDrop() {
+  void CreateButtonWithInkDrop(scoped_ptr<InkDropDelegate> ink_drop_delegate) {
     delete button_;
-    ink_shown_ = false;
-    ink_hidden_ = false;
-    button_ = new TestButtonWithInkDrop(&ink_shown_, &ink_hidden_);
+    button_ = new TestButtonWithInkDrop(std::move(ink_drop_delegate));
     widget_->SetContentsView(button_);
   }
 
  protected:
   Widget* widget() { return widget_.get(); }
   TestCustomButton* button() { return button_; }
-  bool ink_shown() const { return ink_shown_; }
-  bool ink_hidden() const { return ink_hidden_; }
   void SetDraggedView(View* dragged_view) {
     widget_->dragged_view_ = dragged_view;
   }
 
  private:
   scoped_ptr<Widget> widget_;
   TestCustomButton* button_;
-  bool ink_shown_ = false;
-  bool ink_hidden_ = false;
 
   DISALLOW_COPY_AND_ASSIGN(CustomButtonTest);
 };
 
 // Tests that hover state changes correctly when visiblity/enableness changes.
 TEST_F(CustomButtonTest, HoverStateOnVisibilityChange) {
   ui::test::EventGenerator generator(GetContext(), widget()->GetNativeWindow());
 
   generator.PressLeftButton();
   EXPECT_EQ(CustomButton::STATE_PRESSED, button()->state());
@@ skipping 192 matching lines @@
   Textfield textfield;
   EXPECT_FALSE(CustomButton::AsCustomButton(&textfield));
 }
 
 // Tests that pressing a button shows the ink drop and releasing the button
 // does not hide the ink drop.
 // Note: Ink drop is not hidden upon release because CustomButton descendants
 // may enter a different ink drop state.
 TEST_F(CustomButtonTest, ButtonClickTogglesInkDrop) {
   gfx::Point old_cursor = gfx::Screen::GetScreen()->GetCursorScreenPoint();
-  CreateButtonWithInkDrop();
+  bool ink_shown = false;
+  bool ink_hidden = false;
+  CreateButtonWithInkDrop(make_scoped_ptr(
+      new TestInkDropDelegateThatTracksVisibilty(&ink_shown, &ink_hidden)));
 
   ui::test::EventGenerator generator(GetContext(), widget()->GetNativeWindow());
   generator.set_current_location(gfx::Point(50, 50));
   generator.PressLeftButton();
-  EXPECT_TRUE(ink_shown());
-  EXPECT_FALSE(ink_hidden());
+  EXPECT_TRUE(ink_shown);
+  EXPECT_FALSE(ink_hidden);
 
   generator.ReleaseLeftButton();
-  EXPECT_FALSE(ink_hidden());
+  EXPECT_FALSE(ink_hidden);
 }
 
 // Tests that pressing a button shows and releasing capture hides ink drop.
 // Releasing capture should also reset PRESSED button state to NORMAL.
 TEST_F(CustomButtonTest, CaptureLossHidesInkDrop) {
   gfx::Point old_cursor = gfx::Screen::GetScreen()->GetCursorScreenPoint();
-  CreateButtonWithInkDrop();
+  bool ink_shown = false;
+  bool ink_hidden = false;
+  CreateButtonWithInkDrop(make_scoped_ptr(
+      new TestInkDropDelegateThatTracksVisibilty(&ink_shown, &ink_hidden)));
 
   ui::test::EventGenerator generator(GetContext(), widget()->GetNativeWindow());
   generator.set_current_location(gfx::Point(50, 50));
   generator.PressLeftButton();
-  EXPECT_TRUE(ink_shown());
-  EXPECT_FALSE(ink_hidden());
+  EXPECT_TRUE(ink_shown);
+  EXPECT_FALSE(ink_hidden);
 
   EXPECT_EQ(Button::ButtonState::STATE_PRESSED, button()->state());
   SetDraggedView(button());
   widget()->SetCapture(button());
   widget()->ReleaseCapture();
   SetDraggedView(nullptr);
-  EXPECT_TRUE(ink_hidden());
+  EXPECT_TRUE(ink_hidden);
   EXPECT_EQ(ui::MaterialDesignController::IsModeMaterial()
                 ? Button::ButtonState::STATE_NORMAL
                 : Button::ButtonState::STATE_PRESSED,
             button()->state());
 }
 
+TEST_F(CustomButtonTest, InkDropAfterShowingContextMenu) {
+  TestInkDropDelegate* ink_drop_delegate = new TestInkDropDelegate();
+  CreateButtonWithInkDrop(make_scoped_ptr(ink_drop_delegate));
+  TestContextMenuController context_menu_controller;
+  button()->set_context_menu_controller(&context_menu_controller);
+
+  ink_drop_delegate->SetHovered(true);
+  ink_drop_delegate->OnAction(InkDropState::ACTION_PENDING);
+
+  button()->ShowContextMenu(gfx::Point(), ui::MENU_SOURCE_MOUSE);
+
+  EXPECT_FALSE(ink_drop_delegate->is_hovered());
+  EXPECT_EQ(InkDropState::HIDDEN, ink_drop_delegate->state());
+}
+
+TEST_F(CustomButtonTest, InkDropAfterTryingToShowContextMenu) {
+  TestInkDropDelegate* ink_drop_delegate = new TestInkDropDelegate();
+  CreateButtonWithInkDrop(make_scoped_ptr(ink_drop_delegate));
+  button()->set_context_menu_controller(nullptr);
+
+  ink_drop_delegate->SetHovered(true);
+  ink_drop_delegate->OnAction(InkDropState::ACTION_PENDING);
+
+  button()->ShowContextMenu(gfx::Point(), ui::MENU_SOURCE_MOUSE);
+
+  EXPECT_TRUE(ink_drop_delegate->is_hovered());
+  EXPECT_EQ(InkDropState::ACTION_PENDING, ink_drop_delegate->state());
+}
+
 }  // namespace views