1 // Copyright 2017 PDFium 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 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
6
7 #include "xfa/fde/css/cfde_cssstylesheet.h"
8
9 #include <memory>
10 #include <vector>
11
12 #include "testing/gtest/include/gtest/gtest.h"
13 #include "third_party/base/ptr_util.h"
14 #include "third_party/base/stl_util.h"
15 #include "xfa/fde/css/cfde_cssdeclaration.h"
16 #include "xfa/fde/css/cfde_cssenumvalue.h"
17 #include "xfa/fde/css/cfde_cssnumbervalue.h"
18 #include "xfa/fde/css/cfde_cssstylerule.h"
19 #include "xfa/fde/css/cfde_cssvaluelist.h"
20
21 class CFDE_CSSStyleSheetTest : public testing::Test {
22 public:
SetUp()23 void SetUp() override {
24 sheet_ = pdfium::MakeUnique<CFDE_CSSStyleSheet>();
25 decl_ = nullptr;
26 }
27
TearDown()28 void TearDown() override { decl_ = nullptr; }
29
LoadAndVerifyDecl(const FX_WCHAR * buf,const std::vector<CFX_WideString> & selectors,size_t decl_count)30 void LoadAndVerifyDecl(const FX_WCHAR* buf,
31 const std::vector<CFX_WideString>& selectors,
32 size_t decl_count) {
33 ASSERT(sheet_);
34
35 EXPECT_TRUE(sheet_->LoadBuffer(buf, FXSYS_wcslen(buf)));
36 EXPECT_EQ(sheet_->CountRules(), 1);
37
38 CFDE_CSSStyleRule* style = sheet_->GetRule(0);
39 EXPECT_EQ(selectors.size(), style->CountSelectorLists());
40
41 for (size_t i = 0; i < selectors.size(); i++) {
42 uint32_t hash = FX_HashCode_GetW(selectors[i].AsStringC(), true);
43 EXPECT_EQ(hash, style->GetSelectorList(i)->GetNameHash());
44 }
45
46 decl_ = style->GetDeclaration();
47 EXPECT_EQ(decl_->PropertyCountForTesting(), decl_count);
48 }
49
VerifyFloat(FDE_CSSProperty prop,float val,FDE_CSSNumberType type)50 void VerifyFloat(FDE_CSSProperty prop, float val, FDE_CSSNumberType type) {
51 ASSERT(decl_);
52
53 bool important;
54 CFX_RetainPtr<CFDE_CSSValue> v = decl_->GetProperty(prop, &important);
55 EXPECT_EQ(v->GetType(), FDE_CSSPrimitiveType::Number);
56 EXPECT_EQ(v.As<CFDE_CSSNumberValue>()->Kind(), type);
57 EXPECT_EQ(v.As<CFDE_CSSNumberValue>()->Value(), val);
58 }
59
VerifyEnum(FDE_CSSProperty prop,FDE_CSSPropertyValue val)60 void VerifyEnum(FDE_CSSProperty prop, FDE_CSSPropertyValue val) {
61 ASSERT(decl_);
62
63 bool important;
64 CFX_RetainPtr<CFDE_CSSValue> v = decl_->GetProperty(prop, &important);
65 EXPECT_EQ(v->GetType(), FDE_CSSPrimitiveType::Enum);
66 EXPECT_EQ(v.As<CFDE_CSSEnumValue>()->Value(), val);
67 }
68
VerifyList(FDE_CSSProperty prop,std::vector<FDE_CSSPropertyValue> values)69 void VerifyList(FDE_CSSProperty prop,
70 std::vector<FDE_CSSPropertyValue> values) {
71 ASSERT(decl_);
72
73 bool important;
74 CFX_RetainPtr<CFDE_CSSValueList> list =
75 decl_->GetProperty(prop, &important).As<CFDE_CSSValueList>();
76 EXPECT_EQ(list->CountValues(), pdfium::CollectionSize<int32_t>(values));
77
78 for (size_t i = 0; i < values.size(); i++) {
79 CFX_RetainPtr<CFDE_CSSValue> val = list->GetValue(i);
80 EXPECT_EQ(val->GetType(), FDE_CSSPrimitiveType::Enum);
81 EXPECT_EQ(val.As<CFDE_CSSEnumValue>()->Value(), values[i]);
82 }
83 }
84
85 std::unique_ptr<CFDE_CSSStyleSheet> sheet_;
86 CFDE_CSSDeclaration* decl_;
87 };
88
TEST_F(CFDE_CSSStyleSheetTest,ParseMultipleSelectors)89 TEST_F(CFDE_CSSStyleSheetTest, ParseMultipleSelectors) {
90 const FX_WCHAR* buf =
91 L"a { border: 10px; }\nb { text-decoration: underline; }";
92 EXPECT_TRUE(sheet_->LoadBuffer(buf, FXSYS_wcslen(buf)));
93 EXPECT_EQ(2, sheet_->CountRules());
94
95 CFDE_CSSStyleRule* style = sheet_->GetRule(0);
96 EXPECT_EQ(1UL, style->CountSelectorLists());
97
98 bool found_selector = false;
99 uint32_t hash = FX_HashCode_GetW(L"a", true);
100 for (size_t i = 0; i < style->CountSelectorLists(); i++) {
101 if (style->GetSelectorList(i)->GetNameHash() == hash) {
102 found_selector = true;
103 break;
104 }
105 }
106 EXPECT_TRUE(found_selector);
107
108 decl_ = style->GetDeclaration();
109 EXPECT_EQ(4UL, decl_->PropertyCountForTesting());
110
111 VerifyFloat(FDE_CSSProperty::BorderLeftWidth, 10.0,
112 FDE_CSSNumberType::Pixels);
113 VerifyFloat(FDE_CSSProperty::BorderRightWidth, 10.0,
114 FDE_CSSNumberType::Pixels);
115 VerifyFloat(FDE_CSSProperty::BorderTopWidth, 10.0, FDE_CSSNumberType::Pixels);
116 VerifyFloat(FDE_CSSProperty::BorderBottomWidth, 10.0,
117 FDE_CSSNumberType::Pixels);
118
119 style = sheet_->GetRule(1);
120 EXPECT_EQ(1UL, style->CountSelectorLists());
121
122 found_selector = false;
123 hash = FX_HashCode_GetW(L"b", true);
124 for (size_t i = 0; i < style->CountSelectorLists(); i++) {
125 if (style->GetSelectorList(i)->GetNameHash() == hash) {
126 found_selector = true;
127 break;
128 }
129 }
130 EXPECT_TRUE(found_selector);
131
132 decl_ = style->GetDeclaration();
133 EXPECT_EQ(1UL, decl_->PropertyCountForTesting());
134 VerifyList(FDE_CSSProperty::TextDecoration,
135 {FDE_CSSPropertyValue::Underline});
136 }
137
TEST_F(CFDE_CSSStyleSheetTest,ParseChildSelectors)138 TEST_F(CFDE_CSSStyleSheetTest, ParseChildSelectors) {
139 const FX_WCHAR* buf = L"a b c { border: 10px; }";
140 EXPECT_TRUE(sheet_->LoadBuffer(buf, FXSYS_wcslen(buf)));
141 EXPECT_EQ(1, sheet_->CountRules());
142
143 CFDE_CSSStyleRule* style = sheet_->GetRule(0);
144 EXPECT_EQ(1UL, style->CountSelectorLists());
145
146 auto sel = style->GetSelectorList(0);
147 EXPECT_TRUE(sel != nullptr);
148 EXPECT_EQ(FX_HashCode_GetW(L"c", true), sel->GetNameHash());
149
150 sel = sel->GetNextSelector();
151 EXPECT_TRUE(sel != nullptr);
152 EXPECT_EQ(FX_HashCode_GetW(L"b", true), sel->GetNameHash());
153
154 sel = sel->GetNextSelector();
155 EXPECT_TRUE(sel != nullptr);
156 EXPECT_EQ(FX_HashCode_GetW(L"a", true), sel->GetNameHash());
157
158 sel = sel->GetNextSelector();
159 EXPECT_TRUE(sel == nullptr);
160
161 decl_ = style->GetDeclaration();
162 EXPECT_EQ(4UL, decl_->PropertyCountForTesting());
163
164 VerifyFloat(FDE_CSSProperty::BorderLeftWidth, 10.0,
165 FDE_CSSNumberType::Pixels);
166 VerifyFloat(FDE_CSSProperty::BorderRightWidth, 10.0,
167 FDE_CSSNumberType::Pixels);
168 VerifyFloat(FDE_CSSProperty::BorderTopWidth, 10.0, FDE_CSSNumberType::Pixels);
169 VerifyFloat(FDE_CSSProperty::BorderBottomWidth, 10.0,
170 FDE_CSSNumberType::Pixels);
171 }
172
TEST_F(CFDE_CSSStyleSheetTest,ParseUnhandledSelectors)173 TEST_F(CFDE_CSSStyleSheetTest, ParseUnhandledSelectors) {
174 const FX_WCHAR* buf = L"a > b { padding: 0; }";
175 EXPECT_TRUE(sheet_->LoadBuffer(buf, FXSYS_wcslen(buf)));
176 EXPECT_EQ(0, sheet_->CountRules());
177
178 buf = L"a[first] { padding: 0; }";
179 EXPECT_TRUE(sheet_->LoadBuffer(buf, FXSYS_wcslen(buf)));
180 EXPECT_EQ(0, sheet_->CountRules());
181
182 buf = L"a+b { padding: 0; }";
183 EXPECT_TRUE(sheet_->LoadBuffer(buf, FXSYS_wcslen(buf)));
184 EXPECT_EQ(0, sheet_->CountRules());
185
186 buf = L"a ^ b { padding: 0; }";
187 EXPECT_TRUE(sheet_->LoadBuffer(buf, FXSYS_wcslen(buf)));
188 EXPECT_EQ(0, sheet_->CountRules());
189 }
190
TEST_F(CFDE_CSSStyleSheetTest,ParseMultipleSelectorsCombined)191 TEST_F(CFDE_CSSStyleSheetTest, ParseMultipleSelectorsCombined) {
192 LoadAndVerifyDecl(L"a, b, c { border: 5px; }", {L"a", L"b", L"c"}, 4);
193 }
194
TEST_F(CFDE_CSSStyleSheetTest,ParseBorder)195 TEST_F(CFDE_CSSStyleSheetTest, ParseBorder) {
196 LoadAndVerifyDecl(L"a { border: 5px; }", {L"a"}, 4);
197 VerifyFloat(FDE_CSSProperty::BorderLeftWidth, 5.0, FDE_CSSNumberType::Pixels);
198 VerifyFloat(FDE_CSSProperty::BorderRightWidth, 5.0,
199 FDE_CSSNumberType::Pixels);
200 VerifyFloat(FDE_CSSProperty::BorderTopWidth, 5.0, FDE_CSSNumberType::Pixels);
201 VerifyFloat(FDE_CSSProperty::BorderBottomWidth, 5.0,
202 FDE_CSSNumberType::Pixels);
203 }
204
TEST_F(CFDE_CSSStyleSheetTest,ParseBorderFull)205 TEST_F(CFDE_CSSStyleSheetTest, ParseBorderFull) {
206 LoadAndVerifyDecl(L"a { border: 5px solid red; }", {L"a"}, 4);
207 VerifyFloat(FDE_CSSProperty::BorderLeftWidth, 5.0, FDE_CSSNumberType::Pixels);
208 VerifyFloat(FDE_CSSProperty::BorderRightWidth, 5.0,
209 FDE_CSSNumberType::Pixels);
210 VerifyFloat(FDE_CSSProperty::BorderTopWidth, 5.0, FDE_CSSNumberType::Pixels);
211 VerifyFloat(FDE_CSSProperty::BorderBottomWidth, 5.0,
212 FDE_CSSNumberType::Pixels);
213 }
214
TEST_F(CFDE_CSSStyleSheetTest,ParseBorderLeft)215 TEST_F(CFDE_CSSStyleSheetTest, ParseBorderLeft) {
216 LoadAndVerifyDecl(L"a { border-left: 2.5pc; }", {L"a"}, 1);
217 VerifyFloat(FDE_CSSProperty::BorderLeftWidth, 2.5, FDE_CSSNumberType::Picas);
218 }
219
TEST_F(CFDE_CSSStyleSheetTest,ParseBorderLeftThick)220 TEST_F(CFDE_CSSStyleSheetTest, ParseBorderLeftThick) {
221 LoadAndVerifyDecl(L"a { border-left: thick; }", {L"a"}, 1);
222 VerifyEnum(FDE_CSSProperty::BorderLeftWidth, FDE_CSSPropertyValue::Thick);
223 }
224
TEST_F(CFDE_CSSStyleSheetTest,ParseBorderRight)225 TEST_F(CFDE_CSSStyleSheetTest, ParseBorderRight) {
226 LoadAndVerifyDecl(L"a { border-right: 2.5pc; }", {L"a"}, 1);
227 VerifyFloat(FDE_CSSProperty::BorderRightWidth, 2.5, FDE_CSSNumberType::Picas);
228 }
229
TEST_F(CFDE_CSSStyleSheetTest,ParseBorderTop)230 TEST_F(CFDE_CSSStyleSheetTest, ParseBorderTop) {
231 LoadAndVerifyDecl(L"a { border-top: 2.5pc; }", {L"a"}, 1);
232 VerifyFloat(FDE_CSSProperty::BorderTopWidth, 2.5, FDE_CSSNumberType::Picas);
233 }
234
TEST_F(CFDE_CSSStyleSheetTest,ParseBorderBottom)235 TEST_F(CFDE_CSSStyleSheetTest, ParseBorderBottom) {
236 LoadAndVerifyDecl(L"a { border-bottom: 2.5pc; }", {L"a"}, 1);
237 VerifyFloat(FDE_CSSProperty::BorderBottomWidth, 2.5,
238 FDE_CSSNumberType::Picas);
239 }
240