Send ChromeViewMsg_SearchBoxSetSuggestionToPrefetch IPC message from SearchIPCRouter.

chrome/browser/ui/search/search_ipc_router_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 "chrome/browser/ui/search/search_ipc_router.h"
 
 #include <vector>
 
 #include "base/command_line.h"
 #include "base/memory/scoped_ptr.h"
@@ skipping 23 matching lines @@
   MOCK_METHOD1(OnInstantSupportDetermined, void(bool supports_instant));
   MOCK_METHOD1(OnSetVoiceSearchSupport, void(bool supports_voice_search));
 };
 
 class MockSearchIPCRouterPolicy : public SearchIPCRouter::Policy {
  public:
   virtual ~MockSearchIPCRouterPolicy() {}
 
   MOCK_METHOD0(ShouldProcessSetVoiceSearchSupport, bool());
   MOCK_METHOD0(ShouldSendSetDisplayInstantResults, bool());
+  MOCK_METHOD0(ShouldSendSetSuggestionToPrefetch, bool());
   MOCK_METHOD0(ShouldSendMostVisitedItems, bool());
   MOCK_METHOD0(ShouldSendThemeBackgroundInfo, bool());
 };
 
 }  // namespace
 
 class SearchIPCRouterTest : public ChromeRenderViewHostTestHarness {
  public:
   virtual void SetUp() {
     CommandLine::ForCurrentProcess()->AppendSwitch(
@@ skipping 81 matching lines @@
   NavigateAndCommit(GURL("chrome-search://foo/bar"));
   process()->sink().ClearMessages();
 
   SetupMockDelegateAndPolicy(web_contents());
   MockSearchIPCRouterPolicy* policy =
       GetSearchIPCRouterPolicy(web_contents());
   EXPECT_CALL(*policy, ShouldSendSetDisplayInstantResults()).Times(1)
       .WillOnce(testing::Return(true));
 
   GetSearchTabHelper(web_contents())->ipc_router().SetDisplayInstantResults();
-  ASSERT_TRUE(MessageWasSent(
+  EXPECT_TRUE(MessageWasSent(
       ChromeViewMsg_SearchBoxSetDisplayInstantResults::ID));
 }
 
 TEST_F(SearchIPCRouterTest, DoNotSendSetDisplayInstantResultsMsg) {
   NavigateAndCommit(GURL("chrome-search://foo/bar"));
   process()->sink().ClearMessages();
 
   SetupMockDelegateAndPolicy(web_contents());
   MockSearchIPCRouterPolicy* policy =
       GetSearchIPCRouterPolicy(web_contents());
   EXPECT_CALL(*policy, ShouldSendSetDisplayInstantResults()).Times(1)
       .WillOnce(testing::Return(false));
 
   GetSearchTabHelper(web_contents())->ipc_router().SetDisplayInstantResults();
-  ASSERT_FALSE(MessageWasSent(
+  EXPECT_FALSE(MessageWasSent(
       ChromeViewMsg_SearchBoxSetDisplayInstantResults::ID));
 }
 
+TEST_F(SearchIPCRouterTest, SendSetSuggestionToPrefetch) {
+  NavigateAndCommit(GURL("chrome-search://foo/bar"));
+  process()->sink().ClearMessages();
+
+  SetupMockDelegateAndPolicy(web_contents());
+  MockSearchIPCRouterPolicy* policy =
+      GetSearchIPCRouterPolicy(web_contents());
+  EXPECT_CALL(*policy, ShouldSendSetSuggestionToPrefetch()).Times(1)
+      .WillOnce(testing::Return(true));
+
+  GetSearchTabHelper(web_contents())->SetSuggestionToPrefetch(
+      InstantSuggestion());
+  EXPECT_TRUE(MessageWasSent(
+      ChromeViewMsg_SearchBoxSetSuggestionToPrefetch::ID));
+}
+
+TEST_F(SearchIPCRouterTest, DoNotSendSetSuggestionToPrefetch) {
+  NavigateAndCommit(GURL("chrome-search://foo/bar"));
+  process()->sink().ClearMessages();
+
+  SetupMockDelegateAndPolicy(web_contents());
+  MockSearchIPCRouterPolicy* policy =
+      GetSearchIPCRouterPolicy(web_contents());
+  EXPECT_CALL(*policy, ShouldSendSetSuggestionToPrefetch()).Times(1)
+      .WillOnce(testing::Return(false));
+
+  GetSearchTabHelper(web_contents())->SetSuggestionToPrefetch(
+      InstantSuggestion());
+  EXPECT_FALSE(MessageWasSent(
+      ChromeViewMsg_SearchBoxSetSuggestionToPrefetch::ID));
+}
+
 TEST_F(SearchIPCRouterTest, SendMostVisitedItemsMsg) {
   NavigateAndCommit(GURL(chrome::kChromeSearchLocalNtpUrl));
   process()->sink().ClearMessages();
 
   SetupMockDelegateAndPolicy(web_contents());
   MockSearchIPCRouterPolicy* policy =
       GetSearchIPCRouterPolicy(web_contents());
   EXPECT_CALL(*policy, ShouldSendMostVisitedItems()).Times(1)
       .WillOnce(testing::Return(true));
 
   GetSearchTabHelper(web_contents())->ipc_router().SendMostVisitedItems(
       std::vector<InstantMostVisitedItem>());
-  ASSERT_TRUE(MessageWasSent(
+  EXPECT_TRUE(MessageWasSent(
       ChromeViewMsg_SearchBoxMostVisitedItemsChanged::ID));
 }
 
 TEST_F(SearchIPCRouterTest, DoNotSendMostVisitedItemsMsg) {
   NavigateAndCommit(GURL(chrome::kChromeSearchLocalNtpUrl));
   process()->sink().ClearMessages();
 
   SetupMockDelegateAndPolicy(web_contents());
   MockSearchIPCRouterPolicy* policy =
       GetSearchIPCRouterPolicy(web_contents());
   EXPECT_CALL(*policy, ShouldSendMostVisitedItems()).Times(1)
       .WillOnce(testing::Return(false));
 
   GetSearchTabHelper(web_contents())->ipc_router().SendMostVisitedItems(
       std::vector<InstantMostVisitedItem>());
-  ASSERT_FALSE(MessageWasSent(
+  EXPECT_FALSE(MessageWasSent(
       ChromeViewMsg_SearchBoxMostVisitedItemsChanged::ID));
 }
 
 TEST_F(SearchIPCRouterTest, SendThemeBackgroundInfoMsg) {
   NavigateAndCommit(GURL(chrome::kChromeSearchLocalNtpUrl));
   process()->sink().ClearMessages();
 
   SetupMockDelegateAndPolicy(web_contents());
   MockSearchIPCRouterPolicy* policy =
       GetSearchIPCRouterPolicy(web_contents());
   EXPECT_CALL(*policy, ShouldSendThemeBackgroundInfo()).Times(1)
       .WillOnce(testing::Return(true));
 
   GetSearchTabHelper(web_contents())->ipc_router().SendThemeBackgroundInfo(
       ThemeBackgroundInfo());
-  ASSERT_TRUE(MessageWasSent(ChromeViewMsg_SearchBoxThemeChanged::ID));
+  EXPECT_TRUE(MessageWasSent(ChromeViewMsg_SearchBoxThemeChanged::ID));
 }
 
 TEST_F(SearchIPCRouterTest, DoNotSendThemeBackgroundInfoMsg) {
   NavigateAndCommit(GURL(chrome::kChromeSearchLocalNtpUrl));
   process()->sink().ClearMessages();
 
   SetupMockDelegateAndPolicy(web_contents());
   MockSearchIPCRouterPolicy* policy =
       GetSearchIPCRouterPolicy(web_contents());
   EXPECT_CALL(*policy, ShouldSendThemeBackgroundInfo()).Times(1)
       .WillOnce(testing::Return(false));
 
   GetSearchTabHelper(web_contents())->ipc_router().SendThemeBackgroundInfo(
       ThemeBackgroundInfo());
-  ASSERT_FALSE(MessageWasSent(ChromeViewMsg_SearchBoxThemeChanged::ID));
+  EXPECT_FALSE(MessageWasSent(ChromeViewMsg_SearchBoxThemeChanged::ID));
 }