1 // Copyright 2021 gRPC authors.
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 // http://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
14
15 #include "src/core/lib/matchers/matchers.h"
16
17 #include <gtest/gtest.h>
18
19 namespace grpc_core {
20
TEST(StringMatcherTest,ExactMatchCaseSensitive)21 TEST(StringMatcherTest, ExactMatchCaseSensitive) {
22 auto string_matcher =
23 StringMatcher::Create(StringMatcher::Type::kExact,
24 /*matcher=*/"exact", /*case_sensitive=*/true);
25 ASSERT_TRUE(string_matcher.ok());
26 EXPECT_TRUE(string_matcher->Match("exact"));
27 EXPECT_FALSE(string_matcher->Match("Exact"));
28 EXPECT_FALSE(string_matcher->Match("exacz"));
29 }
30
TEST(StringMatcherTest,ExactMatchCaseInsensitive)31 TEST(StringMatcherTest, ExactMatchCaseInsensitive) {
32 auto string_matcher =
33 StringMatcher::Create(StringMatcher::Type::kExact,
34 /*matcher=*/"exact", /*case_sensitive=*/false);
35 ASSERT_TRUE(string_matcher.ok());
36 EXPECT_TRUE(string_matcher->Match("Exact"));
37 EXPECT_FALSE(string_matcher->Match("Exacz"));
38 }
39
TEST(StringMatcherTest,PrefixMatchCaseSensitive)40 TEST(StringMatcherTest, PrefixMatchCaseSensitive) {
41 auto string_matcher = StringMatcher::Create(StringMatcher::Type::kPrefix,
42 /*matcher=*/"prefix",
43 /*case_sensitive=*/true);
44 ASSERT_TRUE(string_matcher.ok());
45 EXPECT_TRUE(string_matcher->Match("prefix-test"));
46 EXPECT_FALSE(string_matcher->Match("xx-prefix-test"));
47 EXPECT_FALSE(string_matcher->Match("Prefix-test"));
48 EXPECT_FALSE(string_matcher->Match("pre-test"));
49 }
50
TEST(StringMatcherTest,PrefixMatchCaseInsensitive)51 TEST(StringMatcherTest, PrefixMatchCaseInsensitive) {
52 auto string_matcher = StringMatcher::Create(StringMatcher::Type::kPrefix,
53 /*matcher=*/"prefix",
54 /*case_sensitive=*/false);
55 ASSERT_TRUE(string_matcher.ok());
56 EXPECT_TRUE(string_matcher->Match("PREfix-test"));
57 EXPECT_FALSE(string_matcher->Match("xx-PREfix-test"));
58 EXPECT_FALSE(string_matcher->Match("PRE-test"));
59 }
60
TEST(StringMatcherTest,SuffixMatchCaseSensitive)61 TEST(StringMatcherTest, SuffixMatchCaseSensitive) {
62 auto string_matcher = StringMatcher::Create(StringMatcher::Type::kSuffix,
63 /*matcher=*/"suffix",
64 /*case_sensitive=*/true);
65 ASSERT_TRUE(string_matcher.ok());
66 EXPECT_TRUE(string_matcher->Match("test-suffix"));
67 EXPECT_FALSE(string_matcher->Match("test-Suffix"));
68 EXPECT_FALSE(string_matcher->Match("test-suffix-xx"));
69 EXPECT_FALSE(string_matcher->Match("test-suffiz"));
70 }
71
TEST(StringMatcherTest,SuffixMatchCaseInSensitive)72 TEST(StringMatcherTest, SuffixMatchCaseInSensitive) {
73 auto string_matcher = StringMatcher::Create(StringMatcher::Type::kSuffix,
74 /*matcher=*/"suffix",
75 /*case_sensitive=*/false);
76 ASSERT_TRUE(string_matcher.ok());
77 EXPECT_TRUE(string_matcher->Match("Test-SUFFIX"));
78 EXPECT_FALSE(string_matcher->Match("Test-SUFFIX-xx"));
79 EXPECT_FALSE(string_matcher->Match("Test-SUFFIZ"));
80 }
81
TEST(StringMatcherTest,InvalidRegex)82 TEST(StringMatcherTest, InvalidRegex) {
83 auto string_matcher = StringMatcher::Create(StringMatcher::Type::kSafeRegex,
84 /*matcher=*/"a[b-a]",
85 /*case_sensitive=*/true);
86 EXPECT_FALSE(string_matcher.ok());
87 EXPECT_EQ(string_matcher.status().code(), absl::StatusCode::kInvalidArgument);
88 EXPECT_EQ(string_matcher.status().message(),
89 "Invalid regex string specified in matcher.");
90 }
91
TEST(StringMatcherTest,SafeRegexMatchCaseSensitive)92 TEST(StringMatcherTest, SafeRegexMatchCaseSensitive) {
93 auto string_matcher = StringMatcher::Create(StringMatcher::Type::kSafeRegex,
94 /*matcher=*/"regex.*",
95 /*case_sensitive=*/true);
96 ASSERT_TRUE(string_matcher.ok());
97 EXPECT_TRUE(string_matcher->Match("regex-test"));
98 EXPECT_FALSE(string_matcher->Match("xx-regex-test"));
99 EXPECT_FALSE(string_matcher->Match("Regex-test"));
100 EXPECT_FALSE(string_matcher->Match("test-regex"));
101 }
102
TEST(StringMatcherTest,SafeRegexMatchCaseInSensitive)103 TEST(StringMatcherTest, SafeRegexMatchCaseInSensitive) {
104 auto string_matcher = StringMatcher::Create(StringMatcher::Type::kSafeRegex,
105 /*matcher=*/"regex.*",
106 /*case_sensitive=*/false);
107 ASSERT_TRUE(string_matcher.ok());
108 EXPECT_TRUE(string_matcher->Match("regex-test"));
109 EXPECT_TRUE(string_matcher->Match("Regex-test"));
110 EXPECT_FALSE(string_matcher->Match("xx-Regex-test"));
111 EXPECT_FALSE(string_matcher->Match("test-regex"));
112 }
113
TEST(StringMatcherTest,ContainsMatchCaseSensitive)114 TEST(StringMatcherTest, ContainsMatchCaseSensitive) {
115 auto string_matcher = StringMatcher::Create(StringMatcher::Type::kContains,
116 /*matcher=*/"contains",
117 /*case_sensitive=*/true);
118 ASSERT_TRUE(string_matcher.ok());
119 EXPECT_TRUE(string_matcher->Match("test-contains"));
120 EXPECT_TRUE(string_matcher->Match("test-contains-test"));
121 EXPECT_FALSE(string_matcher->Match("test-Contains"));
122 EXPECT_FALSE(string_matcher->Match("test-containz"));
123 }
124
TEST(StringMatcherTest,ContainsMatchCaseInSensitive)125 TEST(StringMatcherTest, ContainsMatchCaseInSensitive) {
126 auto string_matcher = StringMatcher::Create(StringMatcher::Type::kContains,
127 /*matcher=*/"contains",
128 /*case_sensitive=*/false);
129 ASSERT_TRUE(string_matcher.ok());
130 EXPECT_TRUE(string_matcher->Match("Test-Contains"));
131 EXPECT_TRUE(string_matcher->Match("Test-Contains-Test"));
132 EXPECT_FALSE(string_matcher->Match("Test-Containz"));
133 }
134
TEST(HeaderMatcherTest,StringMatcher)135 TEST(HeaderMatcherTest, StringMatcher) {
136 auto header_matcher =
137 HeaderMatcher::Create(/*name=*/"key", HeaderMatcher::Type::kExact,
138 /*matcher=*/"exact");
139 ASSERT_TRUE(header_matcher.ok());
140 EXPECT_TRUE(header_matcher->Match("exact"));
141 EXPECT_FALSE(header_matcher->Match("Exact"));
142 EXPECT_FALSE(header_matcher->Match("exacz"));
143 }
144
TEST(HeaderMatcherTest,StringMatcherWithInvertMatch)145 TEST(HeaderMatcherTest, StringMatcherWithInvertMatch) {
146 auto header_matcher =
147 HeaderMatcher::Create(/*name=*/"key", HeaderMatcher::Type::kExact,
148 /*matcher=*/"exact",
149 /*range_start=*/0, /*range_end=*/0,
150 /*present_match=*/false, /*invert_match=*/true);
151 ASSERT_TRUE(header_matcher.ok());
152 EXPECT_FALSE(header_matcher->Match("exact"));
153 EXPECT_TRUE(header_matcher->Match("Exact"));
154 EXPECT_TRUE(header_matcher->Match("exacz"));
155 }
156
TEST(HeaderMatcherTest,InvalidRegex)157 TEST(HeaderMatcherTest, InvalidRegex) {
158 auto header_matcher =
159 HeaderMatcher::Create(/*name=*/"key", HeaderMatcher::Type::kSafeRegex,
160 /*matcher=*/"a[b-a]",
161 /*range_start=*/0, /*range_end=*/0,
162 /*present_match=*/false, /*invert_match=*/true);
163 EXPECT_FALSE(header_matcher.ok());
164 EXPECT_EQ(header_matcher.status().code(), absl::StatusCode::kInvalidArgument);
165 EXPECT_EQ(header_matcher.status().message(),
166 "Invalid regex string specified in matcher.");
167 }
168
TEST(HeaderMatcherTest,RangeMatcherValidRange)169 TEST(HeaderMatcherTest, RangeMatcherValidRange) {
170 auto header_matcher =
171 HeaderMatcher::Create(/*name=*/"key", HeaderMatcher::Type::kRange,
172 /*matcher=*/"", /*range_start=*/10,
173 /*range_end*/ 20);
174 ASSERT_TRUE(header_matcher.ok());
175 EXPECT_TRUE(header_matcher->Match("16"));
176 EXPECT_TRUE(header_matcher->Match("10"));
177 EXPECT_FALSE(header_matcher->Match("3"));
178 EXPECT_FALSE(header_matcher->Match("20"));
179 }
180
TEST(HeaderMatcherTest,RangeMatcherInvalidRange)181 TEST(HeaderMatcherTest, RangeMatcherInvalidRange) {
182 auto header_matcher =
183 HeaderMatcher::Create(/*name=*/"key", HeaderMatcher::Type::kRange,
184 /*matcher=*/"", /*range_start=*/20,
185 /*range_end*/ 10);
186 EXPECT_FALSE(header_matcher.ok());
187 EXPECT_EQ(header_matcher.status().code(), absl::StatusCode::kInvalidArgument);
188 EXPECT_EQ(
189 header_matcher.status().message(),
190 "Invalid range specifier specified: end cannot be smaller than start.");
191 }
192
TEST(HeaderMatcherTest,PresentMatcherTrue)193 TEST(HeaderMatcherTest, PresentMatcherTrue) {
194 auto header_matcher =
195 HeaderMatcher::Create(/*name=*/"key", HeaderMatcher::Type::kPresent,
196 /*matcher=*/"", /*range_start=*/0,
197 /*range_end=*/0, /*present_match=*/true);
198 ASSERT_TRUE(header_matcher.ok());
199 EXPECT_TRUE(header_matcher->Match("any_value"));
200 EXPECT_FALSE(header_matcher->Match(absl::nullopt));
201 }
202
TEST(HeaderMatcherTest,PresentMatcherFalse)203 TEST(HeaderMatcherTest, PresentMatcherFalse) {
204 auto header_matcher =
205 HeaderMatcher::Create(/*name=*/"key", HeaderMatcher::Type::kPresent,
206 /*matcher=*/"", /*range_start=*/0,
207 /*range_end=*/0, /*present_match=*/false);
208 ASSERT_TRUE(header_matcher.ok());
209 EXPECT_FALSE(header_matcher->Match("any_value"));
210 EXPECT_TRUE(header_matcher->Match(absl::nullopt));
211 }
212
213 } // namespace grpc_core
214
main(int argc,char ** argv)215 int main(int argc, char** argv) {
216 ::testing::InitGoogleTest(&argc, argv);
217 return RUN_ALL_TESTS();
218 }
219