chrome/browser/ui/cocoa/extensions/browser_action_test_util_mac.mm

   1 // Copyright (c) 2009 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/extensions/browser_action_test_util.h"
   6
   7 #include "base/mac/foundation_util.h"
   8 #include "base/strings/sys_string_conversions.h"
   9 #include "chrome/browser/ui/browser.h"
  10 #import "chrome/browser/ui/cocoa/browser_window_cocoa.h"
  11 #import "chrome/browser/ui/cocoa/browser_window_controller.h"
  12 #import "chrome/browser/ui/cocoa/extensions/browser_action_button.h"
  13 #import "chrome/browser/ui/cocoa/extensions/browser_actions_container_view.h"
  14 #import "chrome/browser/ui/cocoa/extensions/browser_actions_controller.h"
  15 #import "chrome/browser/ui/cocoa/extensions/extension_popup_controller.h"
  16 #import "chrome/browser/ui/cocoa/toolbar/toolbar_controller.h"
  17 #import "chrome/browser/ui/cocoa/toolbar/wrench_toolbar_button_cell.h"
  18 #import "chrome/browser/ui/cocoa/info_bubble_window.h"
  19 #import "chrome/browser/ui/cocoa/themed_window.h"
  20 #import "chrome/browser/ui/cocoa/toolbar/toolbar_controller.h"
  21 #include "chrome/browser/ui/toolbar/toolbar_action_view_controller.h"
  22 #include "grit/theme_resources.h"
  23 #include "ui/base/theme_provider.h"
  24 #include "ui/gfx/geometry/rect.h"
  25 #include "ui/gfx/geometry/size.h"
  26
  27 namespace {
  28
  29 BrowserActionsController* GetController(
  30     Browser* browser,
  31     ToolbarActionsBarDelegate* barDelegate) {
  32   if (barDelegate)
  33     return [BrowserActionsController fromToolbarActionsBarDelegate:barDelegate];
  34
  35   BrowserWindowCocoa* window =
  36       static_cast<BrowserWindowCocoa*>(browser->window());
  37   return [[window->cocoa_controller() toolbarController]
  38            browserActionsController];
  39 }
  40
  41 BrowserActionButton* GetButton(
  42     Browser* browser,
  43     ToolbarActionsBarDelegate* barDelegate,
  44     int index) {
  45   return [GetController(browser, barDelegate) buttonWithIndex:index];
  46 }
  47
  48 }  // namespace
  49
  50 int BrowserActionTestUtil::NumberOfBrowserActions() {
  51   return [GetController(browser_, bar_delegate_) buttonCount];
  52 }
  53
  54 int BrowserActionTestUtil::VisibleBrowserActions() {
  55   return [GetController(browser_, bar_delegate_) visibleButtonCount];
  56 }
  57
  58 bool BrowserActionTestUtil::IsChevronShowing() {
  59   BrowserActionsController* controller = GetController(browser_, bar_delegate_);
  60   // The magic "18" comes from kChevronWidth in browser_actions_controller.mm.
  61   return ![controller chevronIsHidden] &&
  62          NSWidth([[controller containerView] animationEndFrame]) >= 18;
  63 }
  64
  65 void BrowserActionTestUtil::InspectPopup(int index) {
  66   NOTREACHED();
  67 }
  68
  69 bool BrowserActionTestUtil::HasIcon(int index) {
  70   return [GetButton(browser_, bar_delegate_, index) image] != nil;
  71 }
  72
  73 gfx::Image BrowserActionTestUtil::GetIcon(int index) {
  74   NSImage* ns_image = [GetButton(browser_, bar_delegate_, index) image];
  75   // gfx::Image takes ownership of the |ns_image| reference. We have to increase
  76   // the ref count so |ns_image| stays around when the image object is
  77   // destroyed.
  78   base::mac::NSObjectRetain(ns_image);
  79   return gfx::Image(ns_image);
  80 }
  81
  82 void BrowserActionTestUtil::Press(int index) {
  83   NSButton* button = GetButton(browser_, bar_delegate_, index);
  84   [button performClick:nil];
  85 }
  86
  87 std::string BrowserActionTestUtil::GetExtensionId(int index) {
  88   return [GetButton(browser_, bar_delegate_, index) viewController]->GetId();
  89 }
  90
  91 std::string BrowserActionTestUtil::GetTooltip(int index) {
  92   NSString* tooltip = [GetButton(browser_, bar_delegate_, index) toolTip];
  93   return base::SysNSStringToUTF8(tooltip);
  94 }
  95
  96 gfx::NativeView BrowserActionTestUtil::GetPopupNativeView() {
  97   return [[ExtensionPopupController popup] view];
  98 }
  99
 100 bool BrowserActionTestUtil::HasPopup() {
 101   return [ExtensionPopupController popup] != nil;
 102 }
 103
 104 gfx::Size BrowserActionTestUtil::GetPopupSize() {
 105   NSRect bounds = [[[ExtensionPopupController popup] view] bounds];
 106   return gfx::Size(NSSizeToCGSize(bounds.size));
 107 }
 108
 109 bool BrowserActionTestUtil::HidePopup() {
 110   ExtensionPopupController* controller = [ExtensionPopupController popup];
 111   // The window must be gone or we'll fail a unit test with windows left open.
 112   [static_cast<InfoBubbleWindow*>([controller window])
 113       setAllowedAnimations:info_bubble::kAnimateNone];
 114   [controller close];
 115   return !HasPopup();
 116 }
 117
 118 bool BrowserActionTestUtil::ActionButtonWantsToRun(size_t index) {
 119   BrowserActionsController* controller = GetController(browser_, bar_delegate_);
 120   ui::ThemeProvider* themeProvider =
 121       [[[controller containerView] window] themeProvider];
 122   DCHECK(themeProvider);
 123   NSImage* wantsToRunImage =
 124       themeProvider->GetNSImageNamed(IDR_BROWSER_ACTION_R);
 125   BrowserActionButton* button = [controller buttonWithIndex:index];
 126   BrowserActionCell* cell =
 127       base::mac::ObjCCastStrict<BrowserActionCell>([button cell]);
 128   NSImage* actualImage = [cell imageForState:image_button_cell::kDefaultState
 129                                         view:button];
 130
 131   return wantsToRunImage == actualImage;
 132 }
 133
 134 bool BrowserActionTestUtil::OverflowedActionButtonWantsToRun() {
 135   NSView* wrench = [[[BrowserWindowController browserWindowControllerForWindow:
 136       browser_->window()->GetNativeWindow()] toolbarController] wrenchButton];
 137   NSButton* wrenchButton = base::mac::ObjCCastStrict<NSButton>(wrench);
 138   WrenchToolbarButtonCell* cell =
 139       base::mac::ObjCCastStrict<WrenchToolbarButtonCell>([wrenchButton cell]);
 140   return [cell overflowedToolbarActionWantsToRun];
 141 }
 142
 143 ToolbarActionsBar* BrowserActionTestUtil::GetToolbarActionsBar() {
 144   return [GetController(browser_, bar_delegate_) toolbarActionsBar];
 145 }
 146
 147 // static
 148 gfx::Size BrowserActionTestUtil::GetMinPopupSize() {
 149   return gfx::Size(NSSizeToCGSize([ExtensionPopupController minPopupSize]));
 150 }
 151
 152 // static
 153 gfx::Size BrowserActionTestUtil::GetMaxPopupSize() {
 154   return gfx::Size(NSSizeToCGSize([ExtensionPopupController maxPopupSize]));
 155 }