Move instant support to SearchTabHelper.

chrome/browser/ui/search/search_model_unittest.cc

+// Copyright 2013 The Chromium Authors. All rights reserved.

[samarth, 2013/05/13 20:19:03]

Thanks for adding this!

+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "chrome/browser/ui/search/search_model.h"
+
+#include "base/command_line.h"
+#include "chrome/browser/ui/search/search_model_observer.h"
+#include "chrome/browser/ui/search/search_tab_helper.h"
+#include "chrome/common/chrome_switches.h"
+#include "chrome/common/search_types.h"
+#include "chrome/test/base/chrome_render_view_host_test_harness.h"
+
+namespace {
+
+class MockSearchModelObserver : public SearchModelObserver {
+ public:
+  MockSearchModelObserver();
+  virtual ~MockSearchModelObserver();
+
+  virtual void ModelChanged(const SearchModel::State& old_state,
+                            const SearchModel::State& new_state) OVERRIDE;
+
+  void VerifySearchModelStates(const SearchModel::State& expected_old_state,
+                               const SearchModel::State& expected_new_state);
+
+  void VerifyNotificationCount(int expected_count);
+
+ private:
+  // How many times we've seen search model changed notifications.
+  int modelchanged_notification_count_;
+
+  SearchModel::State actual_old_state_;
+  SearchModel::State actual_new_state_;
+
+  DISALLOW_COPY_AND_ASSIGN(MockSearchModelObserver);
+};
+
+MockSearchModelObserver::MockSearchModelObserver()
+    : modelchanged_notification_count_(0) {
+}
+
+MockSearchModelObserver::~MockSearchModelObserver() {
+}
+
+void MockSearchModelObserver::ModelChanged(
+    const SearchModel::State& old_state,
+    const SearchModel::State& new_state) {
+  actual_old_state_ = old_state;
+  actual_new_state_ = new_state;
+  modelchanged_notification_count_++;
+}
+
+void MockSearchModelObserver::VerifySearchModelStates(
+    const SearchModel::State& expected_old_state,
+    const SearchModel::State& expected_new_state) {
+  EXPECT_TRUE(actual_old_state_ == expected_old_state);
+  EXPECT_TRUE(actual_new_state_ == expected_new_state);
+}
+
+void MockSearchModelObserver::VerifyNotificationCount(int expected_count) {
+  EXPECT_EQ(modelchanged_notification_count_, expected_count);
+}
+
+}  // namespace
+
+class SearchModelTest : public ChromeRenderViewHostTestHarness {
+ public:
+  virtual void SetUp() OVERRIDE;
+  virtual void TearDown() OVERRIDE;
+
+  MockSearchModelObserver mock_observer;
+  SearchModel* model;
+};
+
+void SearchModelTest::SetUp() {
+  CommandLine::ForCurrentProcess()->AppendSwitch(
+      switches::kEnableInstantExtendedAPI);
+  ChromeRenderViewHostTestHarness::SetUp();
+  SearchTabHelper::CreateForWebContents(web_contents());
+  SearchTabHelper* search_tab_helper =
+      SearchTabHelper::FromWebContents(web_contents());
+  ASSERT_TRUE(search_tab_helper != NULL);
+  model = search_tab_helper->model();
+  model->AddObserver(&mock_observer);
+}
+
+void SearchModelTest::TearDown() {
+  model->RemoveObserver(&mock_observer);
+  ChromeRenderViewHostTestHarness::TearDown();
+}
+
+TEST_F(SearchModelTest, UpdateSearchModelInstantSupport) {
+  mock_observer.VerifyNotificationCount(0);
+  EXPECT_TRUE(model->instant_support() == INSTANT_SUPPORT_UNKNOWN);
+  SearchModel::State expected_old_state = model->state();
+  SearchModel::State expected_new_state(model->state());
+  expected_new_state.instant_support = INSTANT_SUPPORT_YES;
+
+  model->SetSupportsInstant(true);
+  mock_observer.VerifySearchModelStates(expected_old_state, expected_new_state);
+  mock_observer.VerifyNotificationCount(1);
+  EXPECT_TRUE(model->instant_support() == INSTANT_SUPPORT_YES);
+
+  expected_old_state = expected_new_state;
+  expected_new_state.instant_support = INSTANT_SUPPORT_NO;
+  model->SetSupportsInstant(false);
+  mock_observer.VerifySearchModelStates(expected_old_state, expected_new_state);
+  mock_observer.VerifyNotificationCount(2);
+
+  // Notify the observer only if the search model state is changed.
+  model->SetSupportsInstant(false);
+  EXPECT_TRUE(model->state() == expected_new_state);
+  EXPECT_TRUE(model->instant_support() == INSTANT_SUPPORT_NO);
+  mock_observer.VerifyNotificationCount(2);
+}
+
+TEST_F(SearchModelTest, UpdateSearchModelMode) {
+  mock_observer.VerifyNotificationCount(0);
+  SearchMode search_mode(SearchMode::MODE_NTP, SearchMode::ORIGIN_NTP);
+  SearchModel::State expected_old_state = model->state();
+  SearchModel::State expected_new_state(model->state());
+  expected_new_state.mode = search_mode;
+
+  model->SetMode(search_mode);
+  mock_observer.VerifySearchModelStates(expected_old_state, expected_new_state);
+  mock_observer.VerifyNotificationCount(1);
+
+  search_mode.mode = SearchMode::MODE_SEARCH_RESULTS;
+  expected_old_state = expected_new_state;
+  expected_new_state.mode = search_mode;
+  model->SetMode(search_mode);
+  mock_observer.VerifySearchModelStates(expected_old_state, expected_new_state);
+  mock_observer.VerifyNotificationCount(2);
+  EXPECT_TRUE(model->state() == expected_new_state);
+}
+
+TEST_F(SearchModelTest, UpdateTopBarVisibility) {
+  mock_observer.VerifyNotificationCount(0);
+  EXPECT_TRUE(model->top_bars_visible());
+
+  SearchModel::State expected_old_state = model->state();
+  SearchModel::State expected_new_state(model->state());
+  expected_new_state.top_bars_visible = false;
+
+  model->SetTopBarsVisible(false);
+  mock_observer.VerifySearchModelStates(expected_old_state, expected_new_state);
+  mock_observer.VerifyNotificationCount(1);
+  EXPECT_FALSE(model->top_bars_visible());
+}
+
+TEST_F(SearchModelTest, UpdateSearchModelState) {
+  SearchModel::State expected_new_state(model->state());
+  expected_new_state.top_bars_visible = false;
+  expected_new_state.instant_support = INSTANT_SUPPORT_NO;
+  EXPECT_FALSE(model->state() == expected_new_state);
+  model->SetState(expected_new_state);
+  mock_observer.VerifyNotificationCount(1);
+  EXPECT_TRUE(model->state() == expected_new_state);
+}