| OLD | NEW |
| 1 // Copyright (c) 2006-2008 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2006-2008 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 "base/ref_counted.h" | 5 #include "base/ref_counted.h" |
| 6 #include "chrome/browser/browser.h" | 6 #include "chrome/browser/browser.h" |
| 7 #include "chrome/browser/renderer_host/render_view_host.h" | 7 #include "chrome/browser/renderer_host/render_view_host.h" |
| 8 #include "chrome/browser/extensions/extension_error_reporter.h" | 8 #include "chrome/browser/extensions/extension_error_reporter.h" |
| 9 #include "chrome/browser/extensions/extension_view.h" | 9 #include "chrome/browser/extensions/extension_host.h" |
| 10 #include "chrome/browser/extensions/extensions_service.h" | 10 #include "chrome/browser/extensions/extensions_service.h" |
| 11 #include "chrome/browser/extensions/test_extension_loader.h" | 11 #include "chrome/browser/extensions/test_extension_loader.h" |
| 12 #include "chrome/browser/tab_contents/site_instance.h" | 12 #include "chrome/browser/tab_contents/site_instance.h" |
| 13 #include "chrome/common/chrome_paths.h" | 13 #include "chrome/common/chrome_paths.h" |
| 14 #include "chrome/common/chrome_switches.h" | 14 #include "chrome/common/chrome_switches.h" |
| 15 #include "chrome/test/in_process_browser_test.h" | 15 #include "chrome/test/in_process_browser_test.h" |
| 16 #include "chrome/test/ui_test_utils.h" | 16 #include "chrome/test/ui_test_utils.h" |
| 17 | 17 |
| 18 namespace { | 18 namespace { |
| 19 | 19 |
| 20 // How long to wait for the extension to put up a javascript alert before giving | 20 // How long to wait for the extension to put up a javascript alert before giving |
| 21 // up. | 21 // up. |
| 22 const int kAlertTimeoutMs = 20000; | 22 const int kAlertTimeoutMs = 20000; |
| 23 | 23 |
| 24 // The extension we're using as our test case. | 24 // The extension we're using as our test case. |
| 25 const char* kExtensionId = "00123456789abcdef0123456789abcdef0123456"; | 25 const char* kExtensionId = "00123456789abcdef0123456789abcdef0123456"; |
| 26 | 26 |
| 27 }; // namespace | 27 }; // namespace |
| 28 | 28 |
| 29 // This class starts up an extension process and waits until it tries to put | 29 // This class starts up an extension process and waits until it tries to put |
| 30 // up a javascript alert. | 30 // up a javascript alert. |
| 31 class MockExtensionView : public ExtensionView { | 31 class MockExtensionHost : public ExtensionHost { |
| 32 public: | 32 public: |
| 33 MockExtensionView(Extension* extension, const GURL& url, | 33 MockExtensionHost(Extension* extension, const GURL& url, |
| 34 SiteInstance* instance, Browser* browser) | 34 SiteInstance* instance) |
| 35 : ExtensionView(extension, url, instance, browser), got_message_(false) { | 35 : ExtensionHost(extension, instance), |
| 36 InitHidden(); | 36 got_message_(false) { |
| 37 CreateRenderView(url, NULL); |
| 37 MessageLoop::current()->PostDelayedTask(FROM_HERE, | 38 MessageLoop::current()->PostDelayedTask(FROM_HERE, |
| 38 new MessageLoop::QuitTask, kAlertTimeoutMs); | 39 new MessageLoop::QuitTask, kAlertTimeoutMs); |
| 39 ui_test_utils::RunMessageLoop(); | 40 ui_test_utils::RunMessageLoop(); |
| 40 } | 41 } |
| 41 | 42 |
| 42 virtual ExtensionFunctionDispatcher* CreateExtensionFunctionDispatcher( | 43 virtual ExtensionFunctionDispatcher* CreateExtensionFunctionDispatcher( |
| 43 RenderViewHost* render_view_host) { | 44 RenderViewHost* render_view_host) { |
| 44 NOTREACHED(); | 45 NOTREACHED(); |
| 45 return NULL; | 46 return NULL; |
| 46 } | 47 } |
| 47 | 48 |
| 48 bool got_message() { return got_message_; } | 49 bool got_message() { return got_message_; } |
| 49 private: | 50 private: |
| 50 virtual void RunJavaScriptMessage( | 51 virtual void RunJavaScriptMessage( |
| 51 const std::wstring& message, | 52 const std::wstring& message, |
| 52 const std::wstring& default_prompt, | 53 const std::wstring& default_prompt, |
| 53 const GURL& frame_url, | 54 const GURL& frame_url, |
| 54 const int flags, | 55 const int flags, |
| 55 IPC::Message* reply_msg, | 56 IPC::Message* reply_msg, |
| 56 bool* did_suppress_message) { | 57 bool* did_suppress_message) { |
| 57 got_message_ = true; | 58 got_message_ = true; |
| 58 MessageLoopForUI::current()->Quit(); | 59 MessageLoopForUI::current()->Quit(); |
| 59 | 60 |
| 60 // Call super, otherwise we'll leak reply_msg. | 61 // Call super, otherwise we'll leak reply_msg. |
| 61 ExtensionView::RunJavaScriptMessage( | 62 ExtensionHost::RunJavaScriptMessage( |
| 62 message, default_prompt, frame_url, flags, | 63 message, default_prompt, frame_url, flags, |
| 63 reply_msg, did_suppress_message); | 64 reply_msg, did_suppress_message); |
| 64 } | 65 } |
| 65 | 66 |
| 66 bool got_message_; | 67 bool got_message_; |
| 67 }; | 68 }; |
| 68 | 69 |
| 69 class ExtensionViewTest : public InProcessBrowserTest { | 70 class ExtensionViewTest : public InProcessBrowserTest { |
| 70 public: | 71 public: |
| 71 virtual void SetUp() { | 72 virtual void SetUp() { |
| (...skipping 21 matching lines...) Expand all Loading... |
| 93 AppendASCII("good").AppendASCII("extension1").AppendASCII("1"); | 94 AppendASCII("good").AppendASCII("extension1").AppendASCII("1"); |
| 94 ASSERT_TRUE(file_util::DirectoryExists(path)); // sanity check | 95 ASSERT_TRUE(file_util::DirectoryExists(path)); // sanity check |
| 95 | 96 |
| 96 // Wait for the extension to load and grab a pointer to it. | 97 // Wait for the extension to load and grab a pointer to it. |
| 97 TestExtensionLoader loader(browser()->profile()); | 98 TestExtensionLoader loader(browser()->profile()); |
| 98 Extension* extension = loader.Load(kExtensionId, path); | 99 Extension* extension = loader.Load(kExtensionId, path); |
| 99 ASSERT_TRUE(extension); | 100 ASSERT_TRUE(extension); |
| 100 GURL url = Extension::GetResourceURL(extension->url(), "toolstrip1.html"); | 101 GURL url = Extension::GetResourceURL(extension->url(), "toolstrip1.html"); |
| 101 | 102 |
| 102 // Start the extension process and wait for it to show a javascript alert. | 103 // Start the extension process and wait for it to show a javascript alert. |
| 103 MockExtensionView view(extension, url, | 104 MockExtensionHost host(extension, url, |
| 104 browser()->profile()->GetExtensionsService()->GetSiteInstanceForURL(url), | 105 browser()->profile()->GetExtensionsService()->GetSiteInstanceForURL(url)); |
| 105 browser()); | 106 EXPECT_TRUE(host.got_message()); |
| 106 EXPECT_TRUE(view.got_message()); | |
| 107 } | 107 } |
| OLD | NEW |