OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/extensions/extension_apitest.h" | 5 #include "chrome/browser/extensions/extension_apitest.h" |
6 #include "chrome/browser/extensions/extension_browser_event_router.h" | 6 #include "chrome/browser/extensions/extension_browser_event_router.h" |
7 #include "chrome/browser/extensions/extension_service.h" | 7 #include "chrome/browser/extensions/extension_service.h" |
8 #include "chrome/browser/extensions/extension_tab_helper.h" | 8 #include "chrome/browser/extensions/extension_tab_helper.h" |
9 #include "chrome/browser/profiles/profile.h" | 9 #include "chrome/browser/profiles/profile.h" |
10 #include "chrome/browser/ui/browser.h" | 10 #include "chrome/browser/ui/browser.h" |
| 11 #include "chrome/browser/ui/browser_tabstrip.h" |
11 #include "chrome/browser/ui/tab_contents/tab_contents.h" | 12 #include "chrome/browser/ui/tab_contents/tab_contents.h" |
12 #include "chrome/common/extensions/extension.h" | 13 #include "chrome/common/extensions/extension.h" |
13 #include "chrome/common/extensions/extension_action.h" | 14 #include "chrome/common/extensions/extension_action.h" |
14 #include "chrome/test/base/ui_test_utils.h" | 15 #include "chrome/test/base/ui_test_utils.h" |
15 #include "content/public/browser/web_contents.h" | 16 #include "content/public/browser/web_contents.h" |
16 | 17 |
17 using extensions::Extension; | 18 using extensions::Extension; |
18 | 19 |
19 IN_PROC_BROWSER_TEST_F(ExtensionApiTest, ScriptBadge) { | 20 IN_PROC_BROWSER_TEST_F(ExtensionApiTest, ScriptBadge) { |
20 ASSERT_TRUE(test_server()->Start()); | 21 ASSERT_TRUE(test_server()->Start()); |
21 ASSERT_TRUE(RunExtensionTest("script_badge/basics")) << message_; | 22 ASSERT_TRUE(RunExtensionTest("script_badge/basics")) << message_; |
22 const Extension* extension = GetSingleLoadedExtension(); | 23 const Extension* extension = GetSingleLoadedExtension(); |
23 ASSERT_TRUE(extension) << message_; | 24 ASSERT_TRUE(extension) << message_; |
24 const ExtensionAction* script_badge = extension->script_badge(); | 25 const ExtensionAction* script_badge = extension->script_badge(); |
25 ASSERT_TRUE(script_badge); | 26 ASSERT_TRUE(script_badge); |
26 | 27 |
27 const int tab_id = browser()->GetActiveTabContents()-> | 28 const int tab_id = chrome::GetActiveTabContents(browser())-> |
28 extension_tab_helper()->tab_id(); | 29 extension_tab_helper()->tab_id(); |
29 EXPECT_EQ(GURL(extension->GetResourceURL("default_popup.html")), | 30 EXPECT_EQ(GURL(extension->GetResourceURL("default_popup.html")), |
30 script_badge->GetPopupUrl(tab_id)); | 31 script_badge->GetPopupUrl(tab_id)); |
31 | 32 |
32 { | 33 { |
33 ResultCatcher catcher; | 34 ResultCatcher catcher; |
34 // Tell the extension to update the script badge state. | 35 // Tell the extension to update the script badge state. |
35 ui_test_utils::NavigateToURL( | 36 ui_test_utils::NavigateToURL( |
36 browser(), GURL(extension->GetResourceURL("update.html"))); | 37 browser(), GURL(extension->GetResourceURL("update.html"))); |
37 ASSERT_TRUE(catcher.GetNextResult()); | 38 ASSERT_TRUE(catcher.GetNextResult()); |
38 } | 39 } |
39 | 40 |
40 // Test that we received the changes. | 41 // Test that we received the changes. |
41 EXPECT_EQ(GURL(extension->GetResourceURL("set_popup.html")), | 42 EXPECT_EQ(GURL(extension->GetResourceURL("set_popup.html")), |
42 script_badge->GetPopupUrl(tab_id)); | 43 script_badge->GetPopupUrl(tab_id)); |
43 | 44 |
44 { | 45 { |
45 // Simulate the script badge being clicked. | 46 // Simulate the script badge being clicked. |
46 ResultCatcher catcher; | 47 ResultCatcher catcher; |
47 ExtensionService* service = browser()->profile()->GetExtensionService(); | 48 ExtensionService* service = browser()->profile()->GetExtensionService(); |
48 service->browser_event_router()->ScriptBadgeExecuted( | 49 service->browser_event_router()->ScriptBadgeExecuted( |
49 browser()->profile(), *script_badge, tab_id); | 50 browser()->profile(), *script_badge, tab_id); |
50 EXPECT_TRUE(catcher.GetNextResult()); | 51 EXPECT_TRUE(catcher.GetNextResult()); |
51 } | 52 } |
52 } | 53 } |
OLD | NEW |