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
14namespace extensions {
15namespace browsertest_util {
16
17std::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
37bool 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