blob: efd004a779841db94b7f00335c486df45fdc10b9 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
// Copyright 2018 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 "extensions/browser/browsertest_util.h"
#include "content/public/test/browser_test_utils.h"
#include "extensions/browser/extension_host.h"
#include "extensions/browser/process_manager.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace extensions {
namespace browsertest_util {
std::string ExecuteScriptInBackgroundPage(
content::BrowserContext* context,
const std::string& extension_id,
const std::string& script,
ScriptUserActivation script_user_activation) {
ExtensionHost* host =
ProcessManager::Get(context)->GetBackgroundHostForExtension(extension_id);
if (!host) {
ADD_FAILURE() << "Extension " << extension_id << " has no background page.";
return "";
}
std::string result;
bool success;
if (script_user_activation == ScriptUserActivation::kActivate) {
success = content::ExecuteScriptAndExtractString(host->host_contents(),
script, &result);
} else {
DCHECK_EQ(script_user_activation, ScriptUserActivation::kDontActivate);
success = content::ExecuteScriptWithoutUserGestureAndExtractString(
host->host_contents(), script, &result);
}
// The maximum script size for which to print on failure.
constexpr int kMaxFailingScriptSizeToLog = 1000;
if (!success) {
std::string message_detail = script.length() < kMaxFailingScriptSizeToLog
? script
: "<script too large>";
ADD_FAILURE() << "Executing script failed: " << message_detail;
result.clear();
}
return result;
}
bool ExecuteScriptInBackgroundPageNoWait(content::BrowserContext* context,
const std::string& extension_id,
const std::string& script) {
ExtensionHost* host =
ProcessManager::Get(context)->GetBackgroundHostForExtension(extension_id);
if (!host) {
ADD_FAILURE() << "Extension " << extension_id << " has no background page.";
return false;
}
content::ExecuteScriptAsync(host->host_contents(), script);
return true;
}
} // namespace browsertest_util
} // namespace extensions
|
