forked from chromium/chromium
-
Notifications
You must be signed in to change notification settings - Fork 0
/
all_urls_apitest.cc
108 lines (91 loc) · 4.04 KB
/
all_urls_apitest.cc
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 (c) 2011 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 "base/command_line.h"
#include "base/macros.h"
#include "build/build_config.h"
#include "chrome/browser/extensions/extension_apitest.h"
#include "chrome/browser/extensions/extension_service.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/browser/ui/search/local_ntp_test_utils.h"
#include "chrome/common/url_constants.h"
#include "chrome/test/base/ui_test_utils.h"
#include "components/crx_file/id_util.h"
#include "extensions/browser/extension_registry.h"
#include "extensions/common/extension.h"
#include "extensions/common/extensions_client.h"
#include "extensions/test/extension_test_message_listener.h"
#include "net/test/embedded_test_server/embedded_test_server.h"
namespace extensions {
namespace {
const std::string kAllUrlsTarget = "/extensions/api_test/all_urls/index.html";
}
class AllUrlsApiTest : public ExtensionApiTest {
protected:
AllUrlsApiTest() {}
~AllUrlsApiTest() override {}
const Extension* content_script() const { return content_script_.get(); }
const Extension* execute_script() const { return execute_script_.get(); }
void WhitelistExtensions() {
ExtensionsClient::ScriptingWhitelist whitelist;
whitelist.push_back(content_script_->id());
whitelist.push_back(execute_script_->id());
ExtensionsClient::Get()->SetScriptingWhitelist(whitelist);
// Extensions will have certain permissions withheld at initialization if
// they aren't whitelisted, so we need to reload them.
ExtensionTestMessageListener listener("execute: ready", false);
extension_service()->ReloadExtension(content_script_->id());
extension_service()->ReloadExtension(execute_script_->id());
ASSERT_TRUE(listener.WaitUntilSatisfied());
}
void NavigateAndWait(const std::string& url) {
std::string expected_url = url;
if (url == chrome::kChromeUINewTabURL) {
expected_url =
local_ntp_test_utils::GetFinalNtpUrl(browser()->profile()).spec();
}
ExtensionTestMessageListener listener_a("content script: " + expected_url,
false);
ExtensionTestMessageListener listener_b("execute: " + expected_url, false);
ui_test_utils::NavigateToURL(browser(), GURL(url));
ASSERT_TRUE(listener_a.WaitUntilSatisfied());
ASSERT_TRUE(listener_b.WaitUntilSatisfied());
}
private:
void SetUpOnMainThread() override {
ExtensionApiTest::SetUpOnMainThread();
base::FilePath data_dir = test_data_dir_.AppendASCII("all_urls");
content_script_ = LoadExtension(data_dir.AppendASCII("content_script"));
ASSERT_TRUE(content_script_);
execute_script_ = LoadExtension(data_dir.AppendASCII("execute_script"));
ASSERT_TRUE(execute_script_);
}
scoped_refptr<const Extension> content_script_;
scoped_refptr<const Extension> execute_script_;
DISALLOW_COPY_AND_ASSIGN(AllUrlsApiTest);
};
IN_PROC_BROWSER_TEST_F(AllUrlsApiTest, WhitelistedExtension) {
WhitelistExtensions();
auto* bystander = LoadExtension(
test_data_dir_.AppendASCII("all_urls").AppendASCII("bystander"));
ASSERT_TRUE(bystander);
ASSERT_TRUE(StartEmbeddedTestServer());
// Now verify that we run content scripts on different URLs, including
// data URLs, regular HTTP pages, and resource URLs from extensions.
const std::string test_urls[] = {
"data:text/html;charset=utf-8,<html>asdf</html>",
embedded_test_server()->GetURL(kAllUrlsTarget).spec(),
bystander->GetResourceURL("page.html").spec()
};
for (const auto& test_url : test_urls)
NavigateAndWait(test_url);
}
// Test that an extension NOT whitelisted for scripting can ask for <all_urls>
// and run scripts on non-restricted all pages.
IN_PROC_BROWSER_TEST_F(AllUrlsApiTest, RegularExtensions) {
// Now verify we can script a regular http page.
ASSERT_TRUE(StartEmbeddedTestServer());
NavigateAndWait(embedded_test_server()->GetURL(kAllUrlsTarget).spec());
}
} // namespace extensions