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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
// Copyright 2018 The Chromium Authors
// 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 "base/values.h"
#include "content/public/browser/browser_context.h"
#include "content/public/browser/service_worker_context.h"
#include "content/public/browser/storage_partition.h"
#include "content/public/browser/web_contents.h"
#include "content/public/test/service_worker_test_helpers.h"
#include "extensions/browser/background_script_executor.h"
#include "extensions/browser/extension_host.h"
#include "extensions/browser/extension_registry.h"
#include "extensions/browser/process_manager.h"
#include "extensions/common/extension_id.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace extensions::browsertest_util {
namespace {
// Returns a log-friendly script string.
std::string GetScriptToLog(const std::string& script) {
// The maximum script size for which to print on failure.
static constexpr int kMaxFailingScriptSizeToLog = 1000;
return (script.size() < kMaxFailingScriptSizeToLog) ? script
: "<script too large>";
}
} // namespace
base::Value ExecuteScriptInBackgroundPage(
content::BrowserContext* context,
const ExtensionId& extension_id,
const std::string& script,
ScriptUserActivation script_user_activation) {
BackgroundScriptExecutor script_executor(context);
base::Value value = script_executor.ExecuteScript(
extension_id, script,
BackgroundScriptExecutor::ResultCapture::kSendScriptResult,
script_user_activation);
if (value.is_none()) {
ADD_FAILURE() << "Bad return value. Script: " << GetScriptToLog(script);
}
return value;
}
bool ExecuteScriptInBackgroundPageNoWait(
content::BrowserContext* context,
const ExtensionId& extension_id,
const std::string& script,
ScriptUserActivation script_user_activation) {
return BackgroundScriptExecutor::ExecuteScriptAsync(
context, extension_id, script, script_user_activation);
}
std::string ExecuteScriptInBackgroundPageDeprecated(
content::BrowserContext* context,
const ExtensionId& extension_id,
const std::string& script,
ScriptUserActivation script_user_activation) {
BackgroundScriptExecutor script_executor(context);
// Legacy scripts were written to pass the (string) result via
// window.domAutomationController.send().
base::Value value = script_executor.ExecuteScript(
extension_id, script,
BackgroundScriptExecutor::ResultCapture::kWindowDomAutomationController,
script_user_activation);
if (!value.is_string()) {
ADD_FAILURE() << "Bad return value: " << value.type()
<< "; script: " << GetScriptToLog(script);
return "";
}
return value.GetString();
}
void StopServiceWorkerForExtensionGlobalScope(content::BrowserContext* context,
const ExtensionId& extension_id) {
const Extension* extension =
ExtensionRegistry::Get(context)->GetExtensionById(
extension_id, ExtensionRegistry::ENABLED);
ASSERT_TRUE(extension) << "Unknown extension ID.";
base::RunLoop run_loop;
content::ServiceWorkerContext* service_worker_context =
context->GetDefaultStoragePartition()->GetServiceWorkerContext();
content::StopServiceWorkerForScope(service_worker_context, extension->url(),
run_loop.QuitClosure());
run_loop.Run();
}
bool DidChangeTitle(content::WebContents& web_contents,
const std::u16string& original_title,
const std::u16string& changed_title) {
const std::u16string& title = web_contents.GetTitle();
if (title == changed_title) {
return true;
}
if (title == original_title) {
return false;
}
ADD_FAILURE() << "Unexpected page title found: " << title;
return false;
}
} // namespace extensions::browsertest_util
|
