chrome/browser/ui/cocoa/omnibox/omnibox_view_mac_unittest.mm

   1 // Copyright (c) 2012 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 #import "chrome/browser/ui/cocoa/omnibox/omnibox_view_mac.h"
   6
   7 #include "chrome/browser/ui/cocoa/cocoa_profile_test.h"
   8 #include "chrome/browser/ui/omnibox/chrome_omnibox_client.h"
   9 #include "chrome/browser/ui/omnibox/chrome_omnibox_edit_controller.h"
  10 #include "chrome/browser/ui/toolbar/toolbar_model_delegate.h"
  11 #include "chrome/browser/ui/toolbar/toolbar_model_impl.h"
  12 #include "chrome/test/base/testing_profile.h"
  13 #include "components/omnibox/browser/omnibox_popup_model.h"
  14 #include "components/omnibox/browser/omnibox_popup_view.h"
  15 #include "testing/platform_test.h"
  16 #include "ui/gfx/font.h"
  17 #include "ui/gfx/geometry/rect.h"
  18 #include "ui/gfx/image/image.h"
  19
  20 namespace {
  21
  22 class MockOmniboxEditModel : public OmniboxEditModel {
  23  public:
  24   MockOmniboxEditModel(OmniboxView* view,
  25                        OmniboxEditController* controller,
  26                        Profile* profile)
  27       : OmniboxEditModel(
  28             view,
  29             controller,
  30             make_scoped_ptr(new ChromeOmniboxClient(controller, profile))),
  31         up_or_down_count_(0) {}
  32
  33   void OnUpOrDownKeyPressed(int count) override { up_or_down_count_ = count; }
  34
  35   int up_or_down_count() const { return up_or_down_count_; }
  36
  37   void set_up_or_down_count(int count) {
  38     up_or_down_count_ = count;
  39   }
  40
  41  private:
  42   int up_or_down_count_;
  43
  44   DISALLOW_COPY_AND_ASSIGN(MockOmniboxEditModel);
  45 };
  46
  47 class MockOmniboxPopupView : public OmniboxPopupView {
  48  public:
  49   MockOmniboxPopupView() : is_open_(false) {}
  50   ~MockOmniboxPopupView() override {}
  51
  52   // Overridden from OmniboxPopupView:
  53   bool IsOpen() const override { return is_open_; }
  54   void InvalidateLine(size_t line) override {}
  55   void UpdatePopupAppearance() override {}
  56   gfx::Rect GetTargetBounds() override { return gfx::Rect(); }
  57   void PaintUpdatesNow() override {}
  58   void OnDragCanceled() override {}
  59
  60   void set_is_open(bool is_open) { is_open_ = is_open; }
  61
  62  private:
  63   bool is_open_;
  64
  65   DISALLOW_COPY_AND_ASSIGN(MockOmniboxPopupView);
  66 };
  67
  68 class TestingToolbarModelDelegate : public ToolbarModelDelegate {
  69  public:
  70   TestingToolbarModelDelegate() {}
  71   ~TestingToolbarModelDelegate() override {}
  72
  73   // Overridden from ToolbarModelDelegate:
  74   content::WebContents* GetActiveWebContents() const override { return NULL; }
  75
  76  private:
  77   DISALLOW_COPY_AND_ASSIGN(TestingToolbarModelDelegate);
  78 };
  79
  80 class TestingOmniboxEditController : public ChromeOmniboxEditController {
  81  public:
  82   explicit TestingOmniboxEditController(ToolbarModel* toolbar_model)
  83       : ChromeOmniboxEditController(NULL), toolbar_model_(toolbar_model) {}
  84   ~TestingOmniboxEditController() override {}
  85
  86  protected:
  87   // Overridden from ChromeOmniboxEditController:
  88   void UpdateWithoutTabRestore() override {}
  89   void OnChanged() override {}
  90   void OnSetFocus() override {}
  91   void ShowURL() override {}
  92   ToolbarModel* GetToolbarModel() override { return toolbar_model_; }
  93   const ToolbarModel* GetToolbarModel() const override {
  94     return toolbar_model_;
  95   }
  96   content::WebContents* GetWebContents() override { return nullptr; }
  97
  98  private:
  99   ToolbarModel* toolbar_model_;
 100
 101   DISALLOW_COPY_AND_ASSIGN(TestingOmniboxEditController);
 102 };
 103
 104 }  // namespace
 105
 106 class OmniboxViewMacTest : public CocoaProfileTest {
 107  public:
 108   void SetModel(OmniboxViewMac* view, OmniboxEditModel* model) {
 109     view->model_.reset(model);
 110   }
 111 };
 112
 113 TEST_F(OmniboxViewMacTest, GetFieldFont) {
 114   EXPECT_TRUE(OmniboxViewMac::GetFieldFont(gfx::Font::NORMAL));
 115 }
 116
 117 TEST_F(OmniboxViewMacTest, TabToAutocomplete) {
 118   OmniboxViewMac view(NULL, profile(), NULL, NULL);
 119
 120   // This is deleted by the omnibox view.
 121   MockOmniboxEditModel* model =
 122       new MockOmniboxEditModel(&view, NULL, profile());
 123   SetModel(&view, model);
 124
 125   MockOmniboxPopupView popup_view;
 126   OmniboxPopupModel popup_model(&popup_view, model);
 127
 128   // With popup closed verify that tab doesn't autocomplete.
 129   popup_view.set_is_open(false);
 130   view.OnDoCommandBySelector(@selector(insertTab:));
 131   EXPECT_EQ(0, model->up_or_down_count());
 132   view.OnDoCommandBySelector(@selector(insertBacktab:));
 133   EXPECT_EQ(0, model->up_or_down_count());
 134
 135   // With popup open verify that tab does autocomplete.
 136   popup_view.set_is_open(true);
 137   view.OnDoCommandBySelector(@selector(insertTab:));
 138   EXPECT_EQ(1, model->up_or_down_count());
 139   view.OnDoCommandBySelector(@selector(insertBacktab:));
 140   EXPECT_EQ(-1, model->up_or_down_count());
 141 }
 142
 143 TEST_F(OmniboxViewMacTest, SetGrayTextAutocompletion) {
 144   const NSRect frame = NSMakeRect(0, 0, 50, 30);
 145   base::scoped_nsobject<AutocompleteTextField> field(
 146       [[AutocompleteTextField alloc] initWithFrame:frame]);
 147
 148   TestingToolbarModelDelegate delegate;
 149   ToolbarModelImpl toolbar_model(&delegate);
 150   TestingOmniboxEditController controller(&toolbar_model);
 151   OmniboxViewMac view(&controller, profile(), NULL, field.get());
 152
 153   // This is deleted by the omnibox view.
 154   MockOmniboxEditModel* model =
 155       new MockOmniboxEditModel(&view, &controller, profile());
 156   SetModel(&view, model);
 157
 158   MockOmniboxPopupView popup_view;
 159   OmniboxPopupModel popup_model(&popup_view, model);
 160
 161   view.SetUserText(base::ASCIIToUTF16("Alfred"));
 162   EXPECT_EQ("Alfred", base::UTF16ToUTF8(view.GetText()));
 163   view.SetGrayTextAutocompletion(base::ASCIIToUTF16(" Hitchcock"));
 164   EXPECT_EQ("Alfred", base::UTF16ToUTF8(view.GetText()));
 165   EXPECT_EQ(" Hitchcock", base::UTF16ToUTF8(view.GetGrayTextAutocompletion()));
 166
 167   view.SetUserText(base::string16());
 168   EXPECT_EQ(base::string16(), view.GetText());
 169   EXPECT_EQ(base::string16(), view.GetGrayTextAutocompletion());
 170 }
 171
 172 TEST_F(OmniboxViewMacTest, UpDownArrow) {
 173   OmniboxViewMac view(NULL, profile(), NULL, NULL);
 174
 175   // This is deleted by the omnibox view.
 176   MockOmniboxEditModel* model =
 177       new MockOmniboxEditModel(&view, NULL, profile());
 178   SetModel(&view, model);
 179
 180   MockOmniboxPopupView popup_view;
 181   OmniboxPopupModel popup_model(&popup_view, model);
 182
 183   // With popup closed verify that pressing up and down arrow works.
 184   popup_view.set_is_open(false);
 185   model->set_up_or_down_count(0);
 186   view.OnDoCommandBySelector(@selector(moveDown:));
 187   EXPECT_EQ(1, model->up_or_down_count());
 188   model->set_up_or_down_count(0);
 189   view.OnDoCommandBySelector(@selector(moveUp:));
 190   EXPECT_EQ(-1, model->up_or_down_count());
 191
 192   // With popup open verify that pressing up and down arrow works.
 193   popup_view.set_is_open(true);
 194   model->set_up_or_down_count(0);
 195   view.OnDoCommandBySelector(@selector(moveDown:));
 196   EXPECT_EQ(1, model->up_or_down_count());
 197   model->set_up_or_down_count(0);
 198   view.OnDoCommandBySelector(@selector(moveUp:));
 199   EXPECT_EQ(-1, model->up_or_down_count());
 200 }