1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "base/basictypes.h"
6 #include "googleurl/src/gurl.h"
7 #include "net/base/data_url.h"
8 #include "testing/gtest/include/gtest/gtest.h"
9
10 namespace {
11
12 struct ParseTestData {
13 const char* url;
14 bool is_valid;
15 const char* mime_type;
16 const char* charset;
17 const char* data;
18 };
19
20 }
21
TEST(DataURLTest,Parse)22 TEST(DataURLTest, Parse) {
23 const ParseTestData tests[] = {
24 { "data:",
25 false,
26 "",
27 "",
28 "" },
29
30 { "data:,",
31 true,
32 "text/plain",
33 "US-ASCII",
34 "" },
35
36 { "data:;base64,",
37 true,
38 "text/plain",
39 "US-ASCII",
40 "" },
41
42 { "data:;charset=,test",
43 true,
44 "text/plain",
45 "US-ASCII",
46 "test" },
47
48 { "data:TeXt/HtMl,<b>x</b>",
49 true,
50 "text/html",
51 "US-ASCII",
52 "<b>x</b>" },
53
54 { "data:,foo",
55 true,
56 "text/plain",
57 "US-ASCII",
58 "foo" },
59
60 { "data:;base64,aGVsbG8gd29ybGQ=",
61 true,
62 "text/plain",
63 "US-ASCII",
64 "hello world" },
65
66 { "data:foo/bar;baz=1;charset=kk,boo",
67 true,
68 "foo/bar",
69 "kk",
70 "boo" },
71
72 { "data:text/html,%3Chtml%3E%3Cbody%3E%3Cb%3Ehello%20world"
73 "%3C%2Fb%3E%3C%2Fbody%3E%3C%2Fhtml%3E",
74 true,
75 "text/html",
76 "US-ASCII",
77 "<html><body><b>hello world</b></body></html>" },
78
79 { "data:text/html,<html><body><b>hello world</b></body></html>",
80 true,
81 "text/html",
82 "US-ASCII",
83 "<html><body><b>hello world</b></body></html>" },
84
85 // the comma cannot be url-escaped!
86 { "data:%2Cblah",
87 false,
88 "",
89 "",
90 "" },
91
92 // invalid base64 content
93 { "data:;base64,aGVs_-_-",
94 false,
95 "",
96 "",
97 "" },
98
99 // Spaces should be removed from non-text data URLs (we already tested
100 // spaces above).
101 { "data:image/fractal,a b c d e f g",
102 true,
103 "image/fractal",
104 "US-ASCII",
105 "abcdefg" },
106
107 // Spaces should also be removed from anything base-64 encoded
108 { "data:;base64,aGVs bG8gd2 9ybGQ=",
109 true,
110 "text/plain",
111 "US-ASCII",
112 "hello world" },
113
114 // Other whitespace should also be removed from anything base-64 encoded.
115 { "data:;base64,aGVs bG8gd2 \n9ybGQ=",
116 true,
117 "text/plain",
118 "US-ASCII",
119 "hello world" },
120
121 // In base64 encoding, escaped whitespace should be stripped.
122 // (This test was taken from acid3)
123 // http://b/1054495
124 { "data:text/javascript;base64,%20ZD%20Qg%0D%0APS%20An%20Zm91cic%0D%0A%207"
125 "%20",
126 true,
127 "text/javascript",
128 "US-ASCII",
129 "d4 = 'four';" },
130
131 // Only unescaped whitespace should be stripped in non-base64.
132 // http://b/1157796
133 { "data:img/png,A B %20 %0A C",
134 true,
135 "img/png",
136 "US-ASCII",
137 "AB \nC" },
138
139 // TODO(darin): add more interesting tests
140 };
141
142 for (size_t i = 0; i < arraysize(tests); ++i) {
143 std::string mime_type;
144 std::string charset;
145 std::string data;
146 bool ok =
147 net::DataURL::Parse(GURL(tests[i].url), &mime_type, &charset, &data);
148 EXPECT_EQ(ok, tests[i].is_valid);
149 if (tests[i].is_valid) {
150 EXPECT_EQ(tests[i].mime_type, mime_type);
151 EXPECT_EQ(tests[i].charset, charset);
152 EXPECT_EQ(tests[i].data, data);
153 }
154 }
155 }
156