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
content / browser / webui / web_ui_impl_unittest.cc [blame]
// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "content/browser/webui/web_ui_impl.h"
#include "content/browser/webui/url_data_manager_backend.h"
#include "content/browser/webui/web_ui_data_source_impl.h"
#include "content/public/browser/url_data_source.h"
#include "content/public/test/browser_task_environment.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "ui/base/webui/web_ui_util.h"
namespace content {
class WebUIImplTest : public testing::Test {
protected:
WebUIDataSourceImpl* CreateDataSource(const std::string& source_name) {
return new WebUIDataSourceImpl(source_name);
}
BrowserTaskEnvironment task_environment;
};
// The URLDataManagerBackend starts with two data source, "chrome://resources"
// and "chrome-untrusted://resources". Currently, only "chrome://resources" is
// supported.
TEST_F(WebUIImplTest, LocalResourceLoaderConfigForDefaultDataSource) {
URLDataManagerBackend data_backend;
auto config =
WebUIImpl::GetLocalResourceLoaderConfigForTesting(&data_backend);
ASSERT_EQ(config->sources.size(), 1ul);
const auto& config_source = config->sources[0];
EXPECT_EQ(config_source->name, "resources");
EXPECT_TRUE(config_source->headers.starts_with("HTTP/1.1 200 OK"));
EXPECT_EQ(config_source->should_replace_i18n_in_js, false);
EXPECT_GT(config_source->path_to_resource_id_map.size(), 0ul);
EXPECT_GT(config_source->replacement_strings.size(), 0ul);
}
// This test adds a data source with extra string replacements and resource
// paths added to it and ensures this metadata is reflected in the resulting
// Mojo struct.
TEST_F(WebUIImplTest, LocalResourceLoaderConfigForCustomDataSource) {
URLDataManagerBackend data_backend;
auto* data_source = CreateDataSource("my-data-source");
data_source->AddString("a", "b");
data_source->AddResourcePath("path/to/resource", 42);
data_source->EnableReplaceI18nInJS();
data_backend.AddDataSource(data_source);
auto config =
WebUIImpl::GetLocalResourceLoaderConfigForTesting(&data_backend);
ASSERT_EQ(config->sources.size(), 2ul);
auto it =
std::find_if(config->sources.begin(), config->sources.end(),
[](const blink::mojom::LocalResourceSourcePtr& source) {
return source->name == "my-data-source";
});
ASSERT_NE(it, config->sources.end());
const auto& config_source = *it;
EXPECT_TRUE(config_source->headers.starts_with("HTTP/1.1 200 OK"));
EXPECT_EQ(config_source->should_replace_i18n_in_js, true);
EXPECT_EQ(config_source->path_to_resource_id_map["path/to/resource"], 42);
EXPECT_EQ(config_source->replacement_strings["a"], "b");
}
} // namespace content