Reload Instant NTP and Instant-process tabs on search url change

chrome/browser/search/instant_service_unittest.cc

+// Copyright 2013 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 <string>
+#include <vector>
+
+#include "base/memory/scoped_ptr.h"
+#include "chrome/browser/search/instant_service.h"
+#include "chrome/browser/search/instant_service_observer.h"
+#include "chrome/browser/search/instant_unittest_base.h"
+#include "chrome/browser/search/search.h"
+#include "content/public/browser/web_contents.h"
+#include "content/public/browser/web_contents_observer.h"
+#include "testing/gmock/include/gmock/gmock.h"
+
+using testing::_;
+
+namespace chrome {
+
+namespace {
+
+class MockInstantServiceObserver : public InstantServiceObserver {
+ public:
+  MOCK_METHOD0(DefaultSearchProviderChanged, void());
+  MOCK_METHOD0(GoogleURLUpdated, void());
+  // TODO(skanuj@chromium.org) : Add more tests.

[samarth, 2013/08/12 19:07:28]

That's like a TODO: make things better :)

I'm definitely all for it, but I don't think the TODO is buying us much.

[Anuj, 2013/08/13 00:18:09]

Done.

+  // MOCK_METHOD1(ThemeInfoChanged, void(const ThemeBackgroundInfo&));
+  // MOCK_METHOD1(MostVisitedItemsChanged,
+  // void(const std::vector<InstantMostVisitedItem>&));
+};
+
+class MockWebContentsObserver : public content::WebContentsObserver {
+ public:
+  MOCK_METHOD1(WebContentsDestroyed, void(content::WebContents*));
+
+ protected:
+  friend class InstantServiceTest;
+};
+
+class InstantServiceTest : public InstantUnitTestBase {
+ protected:
+  virtual void SetUp() OVERRIDE {
+    InstantUnitTestBase::SetUp();
+
+    instant_service_observer_.reset(new MockInstantServiceObserver());
+    instant_service_->AddObserver(instant_service_observer());
+
+    instant_ntp_contents_observer_.reset(new MockWebContentsObserver());
+    instant_ntp_contents_observer()->Observe(
+        instant_service_->GetNTPContents());
+  }
+
+  MockInstantServiceObserver* instant_service_observer() {
+    return instant_service_observer_.get();
+  }
+
+  MockWebContentsObserver* instant_ntp_contents_observer() {
+    return instant_ntp_contents_observer_.get();
+  }
+
+  scoped_ptr<MockInstantServiceObserver> instant_service_observer_;
+  scoped_ptr<MockWebContentsObserver> instant_ntp_contents_observer_;
+};
+
+TEST_F(InstantServiceTest, DispatchDefaultSearchProviderChanged) {
+  EXPECT_CALL(*instant_service_observer(), DefaultSearchProviderChanged())
+      .Times(1);
+  EXPECT_CALL(*instant_ntp_contents_observer(),
+              WebContentsDestroyed(instant_service_->GetNTPContents()))
+      .Times(1);
+
+  GURL ntp_url = instant_service_->GetNTPContents()->GetURL();
+  const std::string& new_base_url = "https://bar.com/";
+  SetSearchProvider(new_base_url);
+  GURL new_ntp_url = instant_service_->GetNTPContents()->GetURL();
+  EXPECT_NE(ntp_url, new_ntp_url);
+  EXPECT_TRUE(new_ntp_url.spec().find(new_base_url) == 0);
+}
+
+TEST_F(InstantServiceTest, DispatchGoogleURLUpdated) {
+  EXPECT_CALL(*instant_service_observer(), GoogleURLUpdated()).Times(1);
+  EXPECT_CALL(*instant_ntp_contents_observer(),
+              WebContentsDestroyed(instant_service_->GetNTPContents()))
+      .Times(1);
+
+  GURL ntp_url = instant_service_->GetNTPContents()->GetURL();
+  const std::string& new_base_url = "https://www.google.es/";
+  NotifyGoogleBaseURLUpdate(new_base_url);
+  GURL new_ntp_url = instant_service_->GetNTPContents()->GetURL();
+  EXPECT_NE(ntp_url, new_ntp_url);
+  EXPECT_TRUE(new_ntp_url.spec().find(new_base_url) == 0);
+}
+
+}  // namespace
+
+}  // namespace chrome