Index: chrome/browser/ui/search/instant_page_unittest.cc |
diff --git a/chrome/browser/ui/search/instant_page_unittest.cc b/chrome/browser/ui/search/instant_page_unittest.cc |
index 797d9d2d96e6e4fde63e32e430ea4227a0570467..592c510c7e012719367211e24dd1ad191098f82d 100644 |
--- a/chrome/browser/ui/search/instant_page_unittest.cc |
+++ b/chrome/browser/ui/search/instant_page_unittest.cc |
@@ -49,10 +49,8 @@ class FakePageDelegate : public InstantPage::Delegate { |
content::PageTransition transition, |
WindowOpenDisposition disposition, |
bool is_search_type)); |
- MOCK_METHOD1(DeleteMostVisitedItem, |
- void(InstantRestrictedID most_visited_item_id)); |
- MOCK_METHOD1(UndoMostVisitedDeletion, |
- void(InstantRestrictedID most_visited_item_id)); |
+ MOCK_METHOD1(DeleteMostVisitedItem, void(const GURL& url)); |
+ MOCK_METHOD1(UndoMostVisitedDeletion, void(const GURL& url)); |
MOCK_METHOD0(UndoAllMostVisitedDeletions, void()); |
MOCK_METHOD1(InstantPageLoadFailed, void(content::WebContents* contents)); |
}; |
@@ -113,3 +111,32 @@ TEST_F(InstantPageTest, DetermineIfPageSupportsInstant_NonLocal) { |
ASSERT_TRUE(message != NULL); |
EXPECT_EQ(web_contents()->GetRoutingID(), message->routing_id()); |
} |
+ |
+TEST_F(InstantPageTest, DispatchRequestToDeleteMostVisitedItem) { |
+ page.reset(new FakePage(&delegate, "")); |
+ page->SetContents(web_contents()); |
+ GURL item_url("www.foo.com"); |
+ EXPECT_CALL(delegate, DeleteMostVisitedItem(item_url)).Times(1); |
+ EXPECT_TRUE(page->OnMessageReceived( |
+ ChromeViewHostMsg_SearchBoxDeleteMostVisitedItem(rvh()->GetRoutingID(), |
+ item_url))); |
+} |
+ |
+TEST_F(InstantPageTest, DispatchRequestToUndoMostVisitedDeletion) { |
+ page.reset(new FakePage(&delegate, "")); |
+ page->SetContents(web_contents()); |
+ GURL item_url("www.foo.com"); |
+ EXPECT_CALL(delegate, UndoMostVisitedDeletion(item_url)).Times(1); |
+ EXPECT_TRUE(page->OnMessageReceived( |
+ ChromeViewHostMsg_SearchBoxUndoMostVisitedDeletion(rvh()->GetRoutingID(), |
+ item_url))); |
+} |
+ |
+TEST_F(InstantPageTest, DispatchRequestToUndoAllMostVisitedDeletions) { |
+ page.reset(new FakePage(&delegate, "")); |
+ page->SetContents(web_contents()); |
+ EXPECT_CALL(delegate, UndoAllMostVisitedDeletions()).Times(1); |
+ EXPECT_TRUE(page->OnMessageReceived( |
+ ChromeViewHostMsg_SearchBoxUndoAllMostVisitedDeletions( |
+ rvh()->GetRoutingID()))); |
+} |