14e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)// Copyright 2013 The Chromium Authors. All rights reserved. 24e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 34e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)// found in the LICENSE file. 44e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) 54e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "base/command_line.h" 64e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "base/strings/stringprintf.h" 74e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "base/strings/utf_string_conversions.h" 8cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)#include "chrome/browser/download/download_prefs.h" 94e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "chrome/browser/extensions/extension_install_prompt.h" 104e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "chrome/browser/extensions/tab_helper.h" 114e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "chrome/browser/extensions/webstore_inline_installer.h" 124e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "chrome/browser/extensions/webstore_inline_installer_factory.h" 134e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "chrome/browser/extensions/webstore_installer_test.h" 144e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "chrome/browser/extensions/webstore_standalone_installer.h" 154e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "chrome/browser/profiles/profile.h" 164e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "chrome/browser/ui/browser.h" 174e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "chrome/browser/ui/tabs/tab_strip_model.h" 184e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "chrome/common/chrome_switches.h" 194e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "chrome/test/base/in_process_browser_test.h" 204e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "chrome/test/base/test_switches.h" 214e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "chrome/test/base/ui_test_utils.h" 224e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "content/public/browser/notification_registrar.h" 234e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "content/public/browser/notification_service.h" 244e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "content/public/browser/notification_types.h" 25a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)#include "content/public/browser/render_frame_host.h" 264e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "content/public/browser/web_contents.h" 274e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "content/public/test/browser_test_utils.h" 284e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "net/base/host_port_pair.h" 294e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "net/dns/mock_host_resolver.h" 304e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)#include "url/gurl.h" 314e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) 324e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)using content::WebContents; 334e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)using extensions::Extension; 344e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)using extensions::TabHelper; 354e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)using extensions::WebstoreInlineInstaller; 364e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)using extensions::WebstoreInlineInstallerFactory; 374e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)using extensions::WebstoreStandaloneInstaller; 384e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) 394e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)WebstoreInstallerTest::WebstoreInstallerTest( 404e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) const std::string& webstore_domain, 414e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) const std::string& test_data_path, 424e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) const std::string& crx_filename, 434e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) const std::string& verified_domain, 444e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) const std::string& unverified_domain) 454e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) : webstore_domain_(webstore_domain), 464e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) test_data_path_(test_data_path), 474e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) crx_filename_(crx_filename), 484e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) verified_domain_(verified_domain), 494e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) unverified_domain_(unverified_domain) { 504e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)} 514e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) 524e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)WebstoreInstallerTest::~WebstoreInstallerTest() {} 534e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) 544e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)void WebstoreInstallerTest::SetUpCommandLine(CommandLine* command_line) { 55116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch ExtensionBrowserTest::SetUpCommandLine(command_line); 564e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) // We start the test server now instead of in 574e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) // SetUpInProcessBrowserTestFixture so that we can get its port number. 584e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) ASSERT_TRUE(test_server()->Start()); 594e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) 604e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) net::HostPortPair host_port = test_server()->host_port_pair(); 614e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) test_gallery_url_ = base::StringPrintf( 624e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) "http://%s:%d/files/%s", 634e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) webstore_domain_.c_str(), host_port.port(), test_data_path_.c_str()); 644e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) command_line->AppendSwitchASCII( 654e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) switches::kAppsGalleryURL, test_gallery_url_); 664e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) 674e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) GURL crx_url = GenerateTestServerUrl(webstore_domain_, crx_filename_); 684e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) CommandLine::ForCurrentProcess()->AppendSwitchASCII( 694e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) switches::kAppsGalleryUpdateURL, crx_url.spec()); 704e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) 714e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) // Allow tests to call window.gc(), so that we can check that callback 724e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) // functions don't get collected prematurely. 734e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) command_line->AppendSwitchASCII(switches::kJavaScriptFlags, "--expose-gc"); 744e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)} 754e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) 764e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)void WebstoreInstallerTest::SetUpInProcessBrowserTestFixture() { 774e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) host_resolver()->AddRule(webstore_domain_, "127.0.0.1"); 784e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) host_resolver()->AddRule(verified_domain_, "127.0.0.1"); 794e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) host_resolver()->AddRule(unverified_domain_, "127.0.0.1"); 804e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)} 814e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) 82cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)void WebstoreInstallerTest::SetUpOnMainThread() { 83116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch ExtensionBrowserTest::SetUpOnMainThread(); 84cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) ASSERT_TRUE(download_directory_.CreateUniqueTempDir()); 85cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) DownloadPrefs* download_prefs = DownloadPrefs::FromBrowserContext( 86cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) browser()->profile()); 87cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) download_prefs->SetDownloadPath(download_directory_.path()); 88cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles)} 89cedac228d2dd51db4b79ea1e72c7f249408ee061Torne (Richard Coles) 904e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)GURL WebstoreInstallerTest::GenerateTestServerUrl( 914e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) const std::string& domain, 924e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) const std::string& page_filename) { 934e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) GURL page_url = test_server()->GetURL( 944e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) base::StringPrintf("files/%s/%s", 954e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) test_data_path_.c_str(), 964e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) page_filename.c_str())); 974e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) 984e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) GURL::Replacements replace_host; 994e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) replace_host.SetHostStr(domain); 1004e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) return page_url.ReplaceComponents(replace_host); 1014e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)} 1024e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) 1034e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)void WebstoreInstallerTest::RunTest(const std::string& test_function_name) { 1044e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) bool result = false; 1054e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) std::string script = base::StringPrintf( 1064e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) "%s('%s')", test_function_name.c_str(), 1074e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) test_gallery_url_.c_str()); 1084e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) ASSERT_TRUE(content::ExecuteScriptAndExtractBool( 1094e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) browser()->tab_strip_model()->GetActiveWebContents(), 1104e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) script, 1114e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) &result)); 1124e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) EXPECT_TRUE(result); 1134e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)} 1144e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) 1154e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)bool WebstoreInstallerTest::RunIndexedTest( 1164e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) const std::string& test_function_name, 1174e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) int i) { 1184e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) std::string result = "FAILED"; 1194e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) std::string script = base::StringPrintf("%s('%s', %d)", 1204e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) test_function_name.c_str(), test_gallery_url_.c_str(), i); 1214e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) EXPECT_TRUE(content::ExecuteScriptAndExtractString( 1224e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) browser()->tab_strip_model()->GetActiveWebContents(), 1234e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) script, 1244e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) &result)); 1254e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) EXPECT_TRUE(result != "FAILED"); 1264e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) return result == "KEEPGOING"; 1274e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)} 1284e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) 1294e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)void WebstoreInstallerTest::RunTestAsync( 1304e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) const std::string& test_function_name) { 1314e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) std::string script = base::StringPrintf( 1324e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles) "%s('%s')", test_function_name.c_str(), test_gallery_url_.c_str()); 133a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) browser()->tab_strip_model()->GetActiveWebContents()->GetMainFrame()-> 1341320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci ExecuteJavaScriptForTests(base::UTF8ToUTF16(script)); 1354e180b6a0b4720a9b8e9e959a882386f690f08ffTorne (Richard Coles)} 136116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch 137116680a4aac90f2aa7413d9095a592090648e557Ben Murdochvoid WebstoreInstallerTest::AutoAcceptInstall() { 1385f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) ExtensionInstallPrompt::g_auto_confirm_for_tests = 1395f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) ExtensionInstallPrompt::ACCEPT; 140116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch} 141116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch 142116680a4aac90f2aa7413d9095a592090648e557Ben Murdochvoid WebstoreInstallerTest::AutoCancelInstall() { 1435f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) ExtensionInstallPrompt::g_auto_confirm_for_tests = 1445f1c94371a64b3196d4be9466099bb892df9b88eTorne (Richard Coles) ExtensionInstallPrompt::CANCEL; 145116680a4aac90f2aa7413d9095a592090648e557Ben Murdoch} 146