// 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/extensions/extension_install_prompt.h" #include "chrome/browser/extensions/tab_helper.h" #include "chrome/browser/extensions/webstore_inline_installer.h" #include "chrome/browser/extensions/webstore_inline_installer_factory.h" #include "chrome/browser/extensions/webstore_installer_test.h" #include "chrome/browser/extensions/webstore_standalone_installer.h" #include "chrome/browser/ui/browser.h" #include "chrome/browser/ui/tabs/tab_strip_model.h" #include "chrome/test/base/ui_test_utils.h" #include "content/public/browser/render_view_host.h" #include "content/public/browser/web_contents.h" #include "url/gurl.h" using content::WebContents; using extensions::Extension; using extensions::TabHelper; using extensions::WebstoreInlineInstaller; using extensions::WebstoreInlineInstallerFactory; using extensions::WebstoreStandaloneInstaller; const char kWebstoreDomain[] = "cws.com"; const char kAppDomain[] = "app.com"; const char kNonAppDomain[] = "nonapp.com"; const char kTestExtensionId[] = "ecglahbcnmdpdciemllbhojghbkagdje"; const char kTestDataPath[] = "extensions/api_test/webstore_inline_install"; const char kCrxFilename[] = "extension.crx"; class WebstoreInlineInstallerTest : public WebstoreInstallerTest { public: WebstoreInlineInstallerTest() : WebstoreInstallerTest( kWebstoreDomain, kTestDataPath, kCrxFilename, kAppDomain, kNonAppDomain) {} }; class ProgrammableInstallPrompt : public ExtensionInstallPrompt { public: explicit ProgrammableInstallPrompt(WebContents* contents) : ExtensionInstallPrompt(contents) {} virtual ~ProgrammableInstallPrompt() {} virtual void ConfirmStandaloneInstall( Delegate* delegate, const Extension* extension, SkBitmap* icon, const ExtensionInstallPrompt::Prompt& prompt) OVERRIDE { delegate_ = delegate; } static bool Ready() { return delegate_ != NULL; } static void Accept() { delegate_->InstallUIProceed(); } static void Reject() { delegate_->InstallUIAbort(true); } private: static Delegate* delegate_; }; ExtensionInstallPrompt::Delegate* ProgrammableInstallPrompt::delegate_; // Fake inline installer which creates a programmable prompt in place of // the normal dialog UI. class WebstoreInlineInstallerForTest : public WebstoreInlineInstaller { public: WebstoreInlineInstallerForTest(WebContents* contents, const std::string& extension_id, const GURL& requestor_url, const Callback& callback) : WebstoreInlineInstaller( contents, kTestExtensionId, requestor_url, base::Bind(DummyCallback)), programmable_prompt_(NULL) { } virtual scoped_ptr CreateInstallUI() OVERRIDE { programmable_prompt_ = new ProgrammableInstallPrompt(web_contents()); return make_scoped_ptr(programmable_prompt_). PassAs(); } private: virtual ~WebstoreInlineInstallerForTest() {} friend class base::RefCountedThreadSafe; static void DummyCallback(bool /*success*/, const std::string& /*error*/) { } ProgrammableInstallPrompt* programmable_prompt_; }; class WebstoreInlineInstallerForTestFactory : public WebstoreInlineInstallerFactory { virtual ~WebstoreInlineInstallerForTestFactory() {} virtual WebstoreInlineInstaller* CreateInstaller( WebContents* contents, const std::string& webstore_item_id, const GURL& requestor_url, const WebstoreStandaloneInstaller::Callback& callback) OVERRIDE { return new WebstoreInlineInstallerForTest( contents, webstore_item_id, requestor_url, callback); } }; IN_PROC_BROWSER_TEST_F(WebstoreInlineInstallerTest, CloseTabBeforeInstallConfirmation) { GURL install_url = GenerateTestServerUrl(kAppDomain, "install.html"); ui_test_utils::NavigateToURL(browser(), install_url); WebContents* web_contents = browser()->tab_strip_model()->GetActiveWebContents(); TabHelper* tab_helper = TabHelper::FromWebContents(web_contents); tab_helper->SetWebstoreInlineInstallerFactoryForTests( new WebstoreInlineInstallerForTestFactory()); RunTestAsync("runTest"); while (!ProgrammableInstallPrompt::Ready()) base::RunLoop().RunUntilIdle(); web_contents->Close(); ProgrammableInstallPrompt::Accept(); }