chrome/browser/ui/search/search_model_unittest.cc

   1 // Copyright 2013 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/ui/search/search_model.h"
   6
   7 #include "base/command_line.h"
   8 #include "chrome/browser/ui/search/search_model_observer.h"
   9 #include "chrome/browser/ui/search/search_tab_helper.h"
  10 #include "chrome/common/chrome_switches.h"
  11 #include "chrome/common/search_types.h"
  12 #include "chrome/test/base/chrome_render_view_host_test_harness.h"
  13
  14 namespace {
  15
  16 class MockSearchModelObserver : public SearchModelObserver {
  17  public:
  18   MockSearchModelObserver();
  19   ~MockSearchModelObserver() override;
  20
  21   void ModelChanged(const SearchModel::State& old_state,
  22                     const SearchModel::State& new_state) override;
  23
  24   void VerifySearchModelStates(const SearchModel::State& expected_old_state,
  25                                const SearchModel::State& expected_new_state);
  26
  27   void VerifyNotificationCount(int expected_count);
  28
  29  private:
  30   // How many times we've seen search model changed notifications.
  31   int modelchanged_notification_count_;
  32
  33   SearchModel::State actual_old_state_;
  34   SearchModel::State actual_new_state_;
  35
  36   DISALLOW_COPY_AND_ASSIGN(MockSearchModelObserver);
  37 };
  38
  39 MockSearchModelObserver::MockSearchModelObserver()
  40     : modelchanged_notification_count_(0) {
  41 }
  42
  43 MockSearchModelObserver::~MockSearchModelObserver() {
  44 }
  45
  46 void MockSearchModelObserver::ModelChanged(
  47     const SearchModel::State& old_state,
  48     const SearchModel::State& new_state) {
  49   actual_old_state_ = old_state;
  50   actual_new_state_ = new_state;
  51   modelchanged_notification_count_++;
  52 }
  53
  54 void MockSearchModelObserver::VerifySearchModelStates(
  55     const SearchModel::State& expected_old_state,
  56     const SearchModel::State& expected_new_state) {
  57   EXPECT_TRUE(actual_old_state_ == expected_old_state);
  58   EXPECT_TRUE(actual_new_state_ == expected_new_state);
  59 }
  60
  61 void MockSearchModelObserver::VerifyNotificationCount(int expected_count) {
  62   EXPECT_EQ(modelchanged_notification_count_, expected_count);
  63 }
  64
  65 }  // namespace
  66
  67 class SearchModelTest : public ChromeRenderViewHostTestHarness {
  68  public:
  69   void SetUp() override;
  70   void TearDown() override;
  71
  72   MockSearchModelObserver mock_observer;
  73   SearchModel* model;
  74 };
  75
  76 void SearchModelTest::SetUp() {
  77   ChromeRenderViewHostTestHarness::SetUp();
  78   SearchTabHelper::CreateForWebContents(web_contents());
  79   SearchTabHelper* search_tab_helper =
  80       SearchTabHelper::FromWebContents(web_contents());
  81   ASSERT_TRUE(search_tab_helper != NULL);
  82   model = search_tab_helper->model();
  83   model->AddObserver(&mock_observer);
  84 }
  85
  86 void SearchModelTest::TearDown() {
  87   model->RemoveObserver(&mock_observer);
  88   ChromeRenderViewHostTestHarness::TearDown();
  89 }
  90
  91 TEST_F(SearchModelTest, UpdateSearchModelInstantSupport) {
  92   mock_observer.VerifyNotificationCount(0);
  93   EXPECT_TRUE(model->instant_support() == INSTANT_SUPPORT_UNKNOWN);
  94   SearchModel::State expected_old_state = model->state();
  95   SearchModel::State expected_new_state(model->state());
  96   expected_new_state.instant_support = INSTANT_SUPPORT_YES;
  97
  98   model->SetInstantSupportState(INSTANT_SUPPORT_YES);
  99   mock_observer.VerifySearchModelStates(expected_old_state, expected_new_state);
 100   mock_observer.VerifyNotificationCount(1);
 101   EXPECT_TRUE(model->instant_support() == INSTANT_SUPPORT_YES);
 102
 103   expected_old_state = expected_new_state;
 104   expected_new_state.instant_support = INSTANT_SUPPORT_NO;
 105   model->SetInstantSupportState(INSTANT_SUPPORT_NO);
 106   mock_observer.VerifySearchModelStates(expected_old_state, expected_new_state);
 107   mock_observer.VerifyNotificationCount(2);
 108
 109   // Notify the observer only if the search model state is changed.
 110   model->SetInstantSupportState(INSTANT_SUPPORT_NO);
 111   EXPECT_TRUE(model->state() == expected_new_state);
 112   EXPECT_TRUE(model->instant_support() == INSTANT_SUPPORT_NO);
 113   mock_observer.VerifyNotificationCount(2);
 114 }
 115
 116 TEST_F(SearchModelTest, UpdateSearchModelMode) {
 117   mock_observer.VerifyNotificationCount(0);
 118   SearchMode search_mode(SearchMode::MODE_NTP, SearchMode::ORIGIN_NTP);
 119   SearchModel::State expected_old_state = model->state();
 120   SearchModel::State expected_new_state(model->state());
 121   expected_new_state.mode = search_mode;
 122
 123   model->SetMode(search_mode);
 124   mock_observer.VerifySearchModelStates(expected_old_state, expected_new_state);
 125   mock_observer.VerifyNotificationCount(1);
 126
 127   search_mode.mode = SearchMode::MODE_SEARCH_RESULTS;
 128   expected_old_state = expected_new_state;
 129   expected_new_state.mode = search_mode;
 130   model->SetMode(search_mode);
 131   mock_observer.VerifySearchModelStates(expected_old_state, expected_new_state);
 132   mock_observer.VerifyNotificationCount(2);
 133   EXPECT_TRUE(model->state() == expected_new_state);
 134 }
 135
 136 TEST_F(SearchModelTest, UpdateSearchModelState) {
 137   SearchModel::State expected_new_state(model->state());
 138   expected_new_state.instant_support = INSTANT_SUPPORT_NO;
 139   EXPECT_FALSE(model->state() == expected_new_state);
 140   model->SetState(expected_new_state);
 141   mock_observer.VerifyNotificationCount(1);
 142   EXPECT_TRUE(model->state() == expected_new_state);
 143 }
 144
 145 TEST_F(SearchModelTest, UpdateVoiceSearchSupported) {
 146   mock_observer.VerifyNotificationCount(0);
 147   EXPECT_FALSE(model->voice_search_supported());
 148
 149   SearchModel::State expected_old_state = model->state();
 150   SearchModel::State expected_new_state(model->state());
 151   expected_new_state.voice_search_supported = true;
 152
 153   model->SetVoiceSearchSupported(true);
 154   mock_observer.VerifySearchModelStates(expected_old_state, expected_new_state);
 155   mock_observer.VerifyNotificationCount(1);
 156   EXPECT_TRUE(model->voice_search_supported());
 157 }