1 // Copyright 2013 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #include "chrome/browser/extensions/browsertest_util.h" 6 7 #include "chrome/browser/profiles/profile.h" 8 #include "content/public/test/browser_test_utils.h" 9 #include "extensions/browser/extension_host.h" 10 #include "extensions/browser/extension_system.h" 11 #include "extensions/browser/process_manager.h" 12 #include "testing/gtest/include/gtest/gtest.h" 13 14 namespace extensions { 15 namespace browsertest_util { 16 17 std::string ExecuteScriptInBackgroundPage(Profile* profile, 18 const std::string& extension_id, 19 const std::string& script) { 20 extensions::ProcessManager* manager = 21 extensions::ExtensionSystem::Get(profile)->process_manager(); 22 extensions::ExtensionHost* host = 23 manager->GetBackgroundHostForExtension(extension_id); 24 if (host == NULL) { 25 ADD_FAILURE() << "Extension " << extension_id << " has no background page."; 26 return ""; 27 } 28 std::string result; 29 if (!content::ExecuteScriptAndExtractString( 30 host->host_contents(), script, &result)) { 31 ADD_FAILURE() << "Executing script failed: " << script; 32 result.clear(); 33 } 34 return result; 35 } 36 37 bool ExecuteScriptInBackgroundPageNoWait(Profile* profile, 38 const std::string& extension_id, 39 const std::string& script) { 40 extensions::ProcessManager* manager = 41 extensions::ExtensionSystem::Get(profile)->process_manager(); 42 extensions::ExtensionHost* host = 43 manager->GetBackgroundHostForExtension(extension_id); 44 if (host == NULL) { 45 ADD_FAILURE() << "Extension " << extension_id << " has no background page."; 46 return false; 47 } 48 return content::ExecuteScript(host->host_contents(), script); 49 } 50 51 } // namespace browsertest_util 52 } // namespace extensions 53
