Index: chrome/browser/extensions/api/extension_action/page_action_apitest.cc |
diff --git a/chrome/browser/extensions/api/extension_action/page_action_apitest.cc b/chrome/browser/extensions/api/extension_action/page_action_apitest.cc |
index b6f1cc6ed2d084e3a698d4fc17a5007f5d64a3dc..1ce764303ac8814f88591138fb0444a6edec4963 100644 |
--- a/chrome/browser/extensions/api/extension_action/page_action_apitest.cc |
+++ b/chrome/browser/extensions/api/extension_action/page_action_apitest.cc |
@@ -17,8 +17,22 @@ |
#include "chrome/common/extensions/extension_action.h" |
#include "chrome/test/base/ui_test_utils.h" |
#include "content/public/browser/web_contents.h" |
+#include "testing/gmock/include/gmock/gmock.h" |
using extensions::Extension; |
+using ::testing::_; |
+ |
+namespace { |
+ |
+class MockActionIconFactory : public ExtensionAction::IconFactory { |
+ public: |
+ virtual ~MockActionIconFactory() {} |
+ |
+ MOCK_METHOD2(GetIcon, gfx::ImageSkia(const ExtensionIconSet* icon_set, |
+ ExtensionAction::Type type)); |
+}; |
+ |
+} // namespace |
IN_PROC_BROWSER_TEST_F(ExtensionApiTest, PageAction) { |
ASSERT_TRUE(test_server()->Start()); |
@@ -59,10 +73,13 @@ IN_PROC_BROWSER_TEST_F(ExtensionApiTest, PageAction) { |
ASSERT_TRUE(catcher.GetNextResult()); |
} |
+ MockActionIconFactory mock_icon_factory; |
+ EXPECT_CALL(mock_icon_factory, GetIcon(_, _)).Times(0); |
+ |
// Test that we received the changes. |
tab_id = chrome::GetActiveTabContents(browser())->restore_tab_helper()-> |
session_id().id(); |
- EXPECT_FALSE(action->GetIcon(tab_id).IsEmpty()); |
+ EXPECT_FALSE(action->GetIcon(tab_id, &mock_icon_factory).IsEmpty()); |
} |
// Test that calling chrome.pageAction.setPopup() can enable a popup. |