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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
|
// Copyright 2012 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/ui/webui/theme_source.h"
#include <stddef.h>
#include <vector>
#include "base/functional/bind.h"
#include "base/memory/ref_counted_memory.h"
#include "base/run_loop.h"
#include "base/strings/strcat.h"
#include "base/test/bind.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/webui/webui_util_desktop.h"
#include "chrome/common/url_constants.h"
#include "chrome/grit/theme_resources.h"
#include "chrome/test/base/test_theme_provider.h"
#include "chrome/test/base/testing_profile.h"
#include "content/public/test/browser_task_environment.h"
#include "content/public/test/test_renderer_host.h"
#include "content/public/test/web_contents_tester.h"
#include "testing/gtest/include/gtest/gtest.h"
class WebUISourcesTest : public testing::Test {
public:
WebUISourcesTest() = default;
TestingProfile* profile() const { return profile_.get(); }
ThemeSource* theme_source() const { return theme_source_.get(); }
size_t result_data_size() const { return result_data_size_; }
void StartDataRequest(const std::string& source) {
theme_source()->StartDataRequest(
GURL(base::StrCat({content::kChromeUIScheme, "://",
chrome::kChromeUIThemeHost, "/", source})),
test_web_contents_getter_,
base::BindOnce(&WebUISourcesTest::SendResponse,
base::Unretained(this)));
}
size_t result_data_size_ = 0;
private:
void SetUp() override {
webui::SetThemeProviderForTestingDeprecated(&test_theme_provider_);
profile_ = std::make_unique<TestingProfile>();
theme_source_ = std::make_unique<ThemeSource>(profile_.get());
test_web_contents_ = content::WebContentsTester::CreateTestWebContents(
profile_.get(), nullptr);
test_web_contents_getter_ =
base::BindLambdaForTesting([&] { return test_web_contents_.get(); });
}
void TearDown() override {
theme_source_.reset();
test_web_contents_.reset();
test_web_contents_getter_ = content::WebContents::Getter();
profile_.reset();
webui::SetThemeProviderForTestingDeprecated(nullptr);
}
void SendResponse(scoped_refptr<base::RefCountedMemory> data) {
result_data_size_ = data ? data->size() : 0;
}
content::BrowserTaskEnvironment task_environment_;
content::RenderViewHostTestEnabler render_view_host_test_enabler_;
TestThemeProvider test_theme_provider_;
std::unique_ptr<TestingProfile> profile_;
std::unique_ptr<ThemeSource> theme_source_;
std::unique_ptr<content::WebContents> test_web_contents_;
content::WebContents::Getter test_web_contents_getter_;
};
TEST_F(WebUISourcesTest, ThemeSourceMimeTypes) {
std::vector<std::string> css_urls{
"chrome://theme/css/new_tab_theme.css",
"chrome://theme/css/new_tab_theme.css?foo",
"chrome://theme/colors.css?sets.css",
"chrome://theme/colors.css?sets.css=ui",
};
for (auto url : css_urls) {
EXPECT_EQ(theme_source()->GetMimeType(GURL(url)), "text/css");
}
std::vector<std::string> png_urls{
"chrome://theme/current-channel-logo",
"chrome://theme/other.png",
"chrome://theme/WRONGURL",
};
for (auto url : png_urls) {
EXPECT_EQ(theme_source()->GetMimeType(GURL(url)), "image/png");
}
}
TEST_F(WebUISourcesTest, ThemeSourceImages) {
// We used to PNGEncode the images ourselves, but encoder differences
// invalidated that. We now just check that the image exists.
StartDataRequest("IDR_THEME_FRAME_INCOGNITO");
size_t min = 0;
EXPECT_GT(result_data_size_, min);
StartDataRequest("IDR_THEME_TOOLBAR");
EXPECT_GT(result_data_size_, min);
}
TEST_F(WebUISourcesTest, ThemeSourceCSS) {
// Generating the test data for the NTP CSS would just involve copying the
// method, or being super brittle and hard-coding the result (requiring
// an update to the unittest every time the CSS template changes), so we
// just check for a successful request and data that is non-null.
size_t empty_size = 0;
StartDataRequest("css/new_tab_theme.css");
base::RunLoop().RunUntilIdle();
EXPECT_NE(result_data_size_, empty_size);
StartDataRequest("css/new_tab_theme.css?pie");
base::RunLoop().RunUntilIdle();
EXPECT_NE(result_data_size_, empty_size);
#if !DCHECK_IS_ON()
// Check that we send NULL back when we can't find what we're looking for.
StartDataRequest("css/WRONGURL");
EXPECT_EQ(result_data_size_, empty_size);
#endif
}
TEST_F(WebUISourcesTest, ThemeSourceColorsCSS) {
// Check for a successful request and that the data is non-null. The actual
// conversion of color provider colors to css colors is tested in helper
// functions.
size_t empty_size = 0;
StartDataRequest("colors.css?sets=ui");
base::RunLoop().RunUntilIdle();
EXPECT_NE(result_data_size_, empty_size);
}
TEST_F(WebUISourcesTest, ThemeAllowedOrigin) {
EXPECT_EQ(
theme_source()->GetAccessControlAllowOriginForOrigin("chrome://settings"),
"chrome://settings");
EXPECT_EQ(theme_source()->GetAccessControlAllowOriginForOrigin(
"chrome-extensions://some-id"),
"");
EXPECT_EQ(
theme_source()->GetAccessControlAllowOriginForOrigin("http://google.com"),
"");
}
|
