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
|
// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "ui/webui/untrusted_web_ui_browsertest_util.h"
#include <string_view>
#include "content/public/browser/web_contents.h"
#include "content/public/common/url_constants.h"
#include "ui/webui/untrusted_web_ui_controller.h"
namespace ui {
namespace {
class TestUntrustedWebUIController : public ui::UntrustedWebUIController {
public:
TestUntrustedWebUIController(
content::WebUI* web_ui,
const std::string& host,
const content::TestUntrustedDataSourceHeaders& headers)
: ui::UntrustedWebUIController(web_ui) {
content::AddUntrustedDataSource(
web_ui->GetWebContents()->GetBrowserContext(), host, headers);
}
~TestUntrustedWebUIController() override = default;
};
} // namespace
TestUntrustedWebUIConfig::TestUntrustedWebUIConfig(std::string_view host)
: WebUIConfig(content::kChromeUIUntrustedScheme, host) {}
TestUntrustedWebUIConfig::TestUntrustedWebUIConfig(
std::string_view host,
const content::TestUntrustedDataSourceHeaders& headers)
: WebUIConfig(content::kChromeUIUntrustedScheme, host), headers_(headers) {}
TestUntrustedWebUIConfig::~TestUntrustedWebUIConfig() = default;
std::unique_ptr<content::WebUIController>
TestUntrustedWebUIConfig::CreateWebUIController(content::WebUI* web_ui,
const GURL& url) {
return std::make_unique<TestUntrustedWebUIController>(web_ui, host(),
headers_);
}
} // namespace ui
|
