1 /*
2 * SysDB - t/unit/core/store_lookup_test.c
3 * Copyright (C) 2014 Sebastian 'tokkee' Harl <sh@tokkee.org>
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
16 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
17 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
18 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR
19 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
20 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
21 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
22 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
23 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
24 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
25 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26 */
28 #include "core/store.h"
29 #include "core/store-private.h"
30 #include "frontend/parser.h"
31 #include "libsysdb_test.h"
33 #include <check.h>
34 #include <string.h>
36 static void
37 populate(void)
38 {
39 const char *hosts[] = { "a", "b", "c" };
41 struct {
42 const char *host;
43 const char *service;
44 } services[] = {
45 { "a", "s1" },
46 { "a", "s2" },
47 { "b", "s1" },
48 { "b", "s3" },
49 };
51 struct {
52 const char *host;
53 const char *name;
54 sdb_data_t value;
55 } attrs[] = {
56 { "a", "k1", { SDB_TYPE_STRING, { .string = "v1" } } },
57 };
59 size_t i;
61 for (i = 0; i < SDB_STATIC_ARRAY_LEN(hosts); ++i) {
62 int status = sdb_store_host(hosts[i], 1);
63 fail_unless(status == 0,
64 "sdb_store_host(%s, 1) = %d; expected: 0",
65 hosts[i], status);
66 }
68 for (i = 0; i < SDB_STATIC_ARRAY_LEN(services); ++i) {
69 int status = sdb_store_service(services[i].host,
70 services[i].service, 1);
71 fail_unless(status == 0,
72 "sdb_store_service(%s, %s, 1) = %d; expected: 0",
73 services[i].host, services[i].service, status);
74 }
76 for (i = 0; i < SDB_STATIC_ARRAY_LEN(attrs); ++i) {
77 int status = sdb_store_attribute(attrs[i].host,
78 attrs[i].name, &attrs[i].value, 1);
79 fail_unless(status == 0,
80 "sdb_store_attribute(%s, %s, <val>, 1) = %d; expected: 0",
81 attrs[i].host, attrs[i].name, status);
82 }
83 } /* populate */
85 START_TEST(test_store_match_name)
86 {
87 sdb_store_base_t *obj;
89 struct {
90 int type;
91 const char *name;
92 _Bool re;
94 int expected;
95 } golden_data[] = {
96 { SDB_HOST, NULL, 0, 1 },
97 { SDB_HOST, NULL, 1, 1 },
98 { SDB_HOST, "a", 0, 1 },
99 { SDB_HOST, "a", 1, 1 },
100 { SDB_HOST, "b", 0, 0 },
101 { SDB_HOST, "b", 1, 0 },
102 { SDB_HOST, "^a$", 1, 1 },
103 { SDB_HOST, "^b$", 1, 0 },
104 { SDB_HOST, "^a$", 0, 0 },
105 { SDB_HOST, "^b$", 0, 0 },
106 { SDB_SERVICE, NULL, 0, 1 },
107 { SDB_SERVICE, NULL, 1, 1 },
108 { SDB_SERVICE, "s1", 0, 1 },
109 { SDB_SERVICE, "s2", 0, 1 },
110 { SDB_SERVICE, "s3", 0, 0 },
111 { SDB_SERVICE, "s4", 0, 0 },
112 { SDB_SERVICE, "^s1$", 1, 1 },
113 { SDB_SERVICE, "^s1$", 0, 0 },
114 { SDB_SERVICE, "x1", 0, 0 },
115 { SDB_SERVICE, "x1", 1, 0 },
116 { SDB_SERVICE, "x", 1, 0 },
117 { SDB_ATTRIBUTE, NULL, 0, 1 },
118 { SDB_ATTRIBUTE, NULL, 1, 1 },
119 { SDB_ATTRIBUTE, "k1", 0, 1 },
120 { SDB_ATTRIBUTE, "k2", 0, 0 },
121 { SDB_ATTRIBUTE, "k3", 0, 0 },
122 { SDB_ATTRIBUTE, "k", 1, 1 },
123 { SDB_ATTRIBUTE, "1", 1, 1 },
124 { SDB_ATTRIBUTE, "2", 1, 0 },
125 };
127 size_t i;
129 obj = sdb_store_get_host("a");
130 fail_unless(obj != NULL,
131 "sdb_store_get_host(a) = NULL; expected: <host>");
133 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
134 sdb_store_matcher_t *m, *n;
135 char buf[1024];
136 int status;
138 m = sdb_store_name_matcher(golden_data[i].type,
139 golden_data[i].name, golden_data[i].re);
140 fail_unless(m != NULL,
141 "sdb_store_service_matcher(%d, %s, %d) = NULL; expected: <matcher>",
142 golden_data[i].type, golden_data[i].name, golden_data[i].re);
144 status = sdb_store_matcher_matches(m, obj);
145 fail_unless(status == golden_data[i].expected,
146 "sdb_store_matcher_matches(%s, <host a>) = %d; expected: %d",
147 sdb_store_matcher_tostring(m, buf, sizeof(buf)),
148 status, golden_data[i].expected);
150 n = sdb_store_inv_matcher(m);
151 fail_unless(n != NULL,
152 "sdb_store_inv_matcher() = NULL; expected: <matcher>");
153 sdb_object_deref(SDB_OBJ(m));
155 /* now match the inverted set of objects */
156 status = sdb_store_matcher_matches(n, obj);
157 fail_unless(status == !golden_data[i].expected,
158 "sdb_store_matcher_matches(%s, <host a>) = %d; expected: %d",
159 sdb_store_matcher_tostring(n, buf, sizeof(buf)),
160 status, !golden_data[i].expected);
162 sdb_object_deref(SDB_OBJ(n));
163 }
165 sdb_object_deref(SDB_OBJ(obj));
166 }
167 END_TEST
169 START_TEST(test_store_match_attr)
170 {
171 sdb_store_base_t *obj;
173 struct {
174 const char *attr_name;
175 const char *attr_value;
176 _Bool re;
178 int expected;
179 } golden_data[] = {
180 { "k1", NULL, 0, 1 },
181 { "k", NULL, 1, 0 },
182 { "1", NULL, 1, 0 },
183 { "k2", NULL, 0, 0 },
184 { "k2", NULL, 1, 0 },
185 { "k3", NULL, 0, 0 },
186 { "k3", NULL, 1, 0 },
187 { "k1", "v1", 0, 1 },
188 { "k1", "v1", 1, 1 },
189 { "k1", "^v1$", 1, 1 },
190 { "k1", "v", 1, 1 },
191 { "k1", "1", 1, 1 },
192 { "k1", "v2", 0, 0 },
193 { "k1", "v2", 1, 0 },
194 { "k", "v1", 0, 0 },
195 { "k", "v1", 1, 0 },
196 { "k2", "1", 0, 0 },
197 { "k2", "1", 1, 0 },
198 };
200 size_t i;
202 obj = sdb_store_get_host("a");
203 fail_unless(obj != NULL,
204 "sdb_store_get_host(a) = NULL; expected: <host>");
206 fail_unless(sdb_store_attr_matcher(NULL, "re", 0) == NULL,
207 "sdb_store_attr_matcher(NULL, \"re\", 0) = <m>; expected: NULL");
208 fail_unless(sdb_store_attr_matcher(NULL, "re", 1) == NULL,
209 "sdb_store_attr_matcher(NULL, \"re\", 1) = <m>; expected: NULL");
211 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
212 sdb_store_matcher_t *m, *n;
213 char buf[1024];
214 int status;
216 m = sdb_store_attr_matcher(golden_data[i].attr_name,
217 golden_data[i].attr_value, golden_data[i].re);
218 fail_unless(m != NULL,
219 "sdb_store_attr_matcher() = NULL; expected: <matcher>");
221 status = sdb_store_matcher_matches(m, obj);
222 fail_unless(status == golden_data[i].expected,
223 "sdb_store_matcher_matches({%s, <host a>) = %d; expected: %d",
224 sdb_store_matcher_tostring(m, buf, sizeof(buf)),
225 status, golden_data[i].expected);
227 n = sdb_store_inv_matcher(m);
228 fail_unless(n != NULL,
229 "sdb_store_inv_matcher() = NULL; expected: <matcher>");
230 sdb_object_deref(SDB_OBJ(m));
232 /* now match the inverted set of objects */
233 status = sdb_store_matcher_matches(n, obj);
234 fail_unless(status == !golden_data[i].expected,
235 "sdb_store_matcher_matches({%s, <host a>) = %d; expected: %d",
236 sdb_store_matcher_tostring(n, buf, sizeof(buf)),
237 status, !golden_data[i].expected);
239 sdb_object_deref(SDB_OBJ(n));
240 }
242 sdb_object_deref(SDB_OBJ(obj));
243 }
244 END_TEST
246 START_TEST(test_store_match_op)
247 {
248 sdb_store_base_t *obj;
250 sdb_store_matcher_t *always = sdb_store_name_matcher(SDB_HOST, "a", 0);
251 sdb_store_matcher_t *never = sdb_store_name_matcher(SDB_HOST, "z", 0);
253 struct {
254 const char *op;
255 sdb_store_matcher_t *left;
256 sdb_store_matcher_t *right;
257 int expected;
258 } golden_data[] = {
259 { "OR", always, always, 1 },
260 { "OR", always, never, 1 },
261 { "OR", never, always, 1 },
262 { "OR", never, never, 0 },
263 { "AND", always, always, 1 },
264 { "AND", always, never, 0 },
265 { "AND", never, always, 0 },
266 { "AND", never, never, 0 },
267 };
269 int status;
270 size_t i;
272 obj = sdb_store_get_host("a");
274 status = sdb_store_matcher_matches(always, obj);
275 fail_unless(status == 1,
276 "INTERNAL ERROR: 'always' did not match host");
277 status = sdb_store_matcher_matches(never, obj);
278 fail_unless(status == 0,
279 "INTERNAL ERROR: 'never' matches host");
281 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
282 sdb_store_matcher_t *m;
284 if (! strcmp(golden_data[i].op, "OR"))
285 m = sdb_store_dis_matcher(golden_data[i].left,
286 golden_data[i].right);
287 else if (! strcmp(golden_data[i].op, "AND"))
288 m = sdb_store_con_matcher(golden_data[i].left,
289 golden_data[i].right);
290 else {
291 fail("INTERNAL ERROR: unexpected operator %s", golden_data[i].op);
292 continue;
293 }
295 #define TO_NAME(v) (((v) == always) ? "always" \
296 : ((v) == never) ? "never" : "<unknown>")
298 status = sdb_store_matcher_matches(m, obj);
299 fail_unless(status == golden_data[i].expected,
300 "%s(%s, %s) = %d; expected: %d", golden_data[i].op,
301 TO_NAME(golden_data[i].left), TO_NAME(golden_data[i].right),
302 status, golden_data[i].expected);
304 #undef TO_NAME
306 sdb_object_deref(SDB_OBJ(m));
307 }
309 sdb_object_deref(SDB_OBJ(always));
310 sdb_object_deref(SDB_OBJ(never));
312 sdb_object_deref(SDB_OBJ(obj));
313 }
314 END_TEST
316 START_TEST(test_parse_cmp)
317 {
318 sdb_store_matcher_t *check;
320 size_t i;
322 struct {
323 const char *obj_type;
324 const char *attr;
325 const char *op;
326 const char *value;
327 int expected;
328 } golden_data[] = {
329 { "host", "name", "=", "hostname", MATCHER_NAME },
330 { "host", "name", "!=", "hostname", MATCHER_NOT },
331 { "host", "name", "=~", "hostname", MATCHER_NAME },
332 { "host", "name", "!~", "hostname", MATCHER_NOT },
333 { "host", "attr", "=", "hostname", -1 },
334 { "host", "attr", "!=", "hostname", -1 },
335 { "host", "name", "&^", "hostname", -1 },
336 { "service", "name", "=", "srvname", MATCHER_NAME },
337 { "service", "name", "!=", "srvname", MATCHER_NOT },
338 { "service", "name", "=~", "srvname", MATCHER_NAME },
339 { "service", "name", "!~", "srvname", MATCHER_NOT },
340 { "service", "attr", "=", "srvname", -1 },
341 { "service", "attr", "!=", "srvname", -1 },
342 { "service", "name", "&^", "srvname", -1 },
343 { "attribute", "name", "=", "attrname", MATCHER_NAME },
344 { "attribute", "name", "!=", "attrname", MATCHER_NOT },
345 { "attribute", "name", "=~", "attrname", MATCHER_NAME },
346 { "attribute", "name", "!~", "attrname", MATCHER_NOT },
347 { "attribute", "attr", "=", "attrname", MATCHER_ATTR },
348 { "attribute", "attr", "!=", "attrname", MATCHER_NOT },
349 { "attribute", "attr", "=~", "attrname", MATCHER_ATTR },
350 { "attribute", "attr", "!~", "attrname", MATCHER_NOT },
351 { "attribute", "attr", "&^", "attrname", -1 },
352 };
354 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
355 check = sdb_store_matcher_parse_cmp(golden_data[i].obj_type,
356 golden_data[i].attr, golden_data[i].op, golden_data[i].value);
358 if (golden_data[i].expected == -1) {
359 fail_unless(check == NULL,
360 "sdb_store_matcher_parse_cmp(%s, %s, %s, %s) = %p; "
361 "expected: NULL", golden_data[i].obj_type,
362 golden_data[i].attr, golden_data[i].op,
363 golden_data[i].value, check);
364 continue;
365 }
367 fail_unless(check != NULL,
368 "sdb_store_matcher_parse_cmp(%s, %s, %s, %s) = %p; "
369 "expected: NULL", golden_data[i].obj_type,
370 golden_data[i].attr, golden_data[i].op,
371 golden_data[i].value, check);
372 fail_unless(M(check)->type == golden_data[i].expected,
373 "sdb_store_matcher_parse_cmp(%s, %s, %s, %s) returned matcher "
374 "of type %d; expected: %d", golden_data[i].obj_type,
375 golden_data[i].attr, golden_data[i].op, golden_data[i].value,
376 M(check)->type, golden_data[i].expected);
378 sdb_object_deref(SDB_OBJ(check));
379 }
380 }
381 END_TEST
383 static int
384 lookup_cb(sdb_store_base_t *obj, void *user_data)
385 {
386 int *i = user_data;
388 fail_unless(obj != NULL,
389 "sdb_store_lookup callback received NULL obj; expected: "
390 "<store base obj>");
391 fail_unless(i != NULL,
392 "sdb_store_lookup callback received NULL user_data; "
393 "expected: <pointer to data>");
395 ++(*i);
396 return 0;
397 } /* lookup_cb */
399 START_TEST(test_lookup)
400 {
401 #define PTR_RE "0x[0-9a-f]+"
402 struct {
403 const char *query;
404 int expected;
405 const char *tostring_re;
406 } golden_data[] = {
407 { "host.name = 'a'", 1,
408 "OBJ\\[host\\]\\{ NAME\\{ 'a', \\(nil\\) \\} \\}" },
409 { "host.name =~ 'a|b'", 2,
410 "OBJ\\[host\\]\\{ NAME\\{ NULL, "PTR_RE" \\} \\}" },
411 { "host.name =~ 'host'", 0,
412 "OBJ\\[host\\]\\{ NAME\\{ NULL, "PTR_RE" \\} \\}" },
413 { "host.name =~ '.'", 3,
414 "OBJ\\[host\\]\\{ NAME\\{ NULL, "PTR_RE" \\} \\}" },
415 { "service.name = 's1'", 2,
416 "OBJ\\[service\\]\\{ NAME\\{ 's1', \\(nil\\) } \\}" },
417 { "service.name =~ 's'", 2,
418 "OBJ\\[service\\]\\{ NAME\\{ NULL, "PTR_RE" } \\}" },
419 { "service.name !~ 's'", 1,
420 "\\(NOT, OBJ\\[service\\]\\{ NAME\\{ NULL, "PTR_RE" } \\}\\)" },
421 { "attribute.name = 'k1'", 1,
422 "OBJ\\[attribute\\]\\{ NAME\\{ 'k1', \\(nil\\) \\} " },
423 { "attribute.name = 'x'", 0,
424 "OBJ\\[attribute\\]\\{ NAME\\{ 'x', \\(nil\\) \\}" },
425 { "attribute.k1 = 'v1'", 1,
426 "ATTR\\[k1\\]\\{ VALUE\\{ 'v1', \\(nil\\) \\} \\}" },
427 { "attribute.k1 != 'v1'", 2,
428 "\\(NOT, ATTR\\[k1\\]\\{ VALUE\\{ 'v1', \\(nil\\) \\} \\}\\)" },
429 { "attribute.k1 != 'v2'", 3,
430 "\\(NOT, ATTR\\[k1\\]\\{ VALUE\\{ 'v2', \\(nil\\) \\} \\}\\)" },
431 { "attribute.name != 'x' "
432 "AND attribute.y !~ 'x'", 3,
433 "\\(AND, "
434 "\\(NOT, OBJ\\[attribute\\]\\{ NAME\\{ 'x', \\(nil\\) \\} \\}\\), "
435 "\\(NOT, ATTR\\[y\\]\\{ VALUE\\{ NULL, "PTR_RE" \\} \\}\\)\\)" },
436 };
438 int check, n;
439 size_t i;
441 n = 0;
442 check = sdb_store_lookup(NULL, lookup_cb, &n);
443 fail_unless(check == 0,
444 "sdb_store_lookup() = %d; expected: 0", check);
445 fail_unless(n == 3,
446 "sdb_store_lookup called callback %d times; expected: 3", (int)n);
448 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
449 sdb_store_matcher_t *m;
450 char buf[4096];
452 m = sdb_fe_parse_matcher(golden_data[i].query, -1);
453 fail_unless(m != NULL,
454 "sdb_fe_parse_matcher(%s, -1) = NULL; expected: <matcher>",
455 golden_data[i].query);
456 fail_unless(sdb_regmatches(golden_data[i].tostring_re,
457 sdb_store_matcher_tostring(m, buf, sizeof(buf))) == 0,
458 "sdb_fe_parse_matcher(%s, -1) = %s; expected: %s",
459 golden_data[i].query,
460 sdb_store_matcher_tostring(m, buf, sizeof(buf)),
461 golden_data[i].tostring_re);
463 n = 0;
464 sdb_store_lookup(m, lookup_cb, &n);
465 fail_unless(n == golden_data[i].expected,
466 "sdb_store_lookup(matcher{%s}) found %d hosts; expected: %d",
467 golden_data[i].query, n, golden_data[i].expected);
468 sdb_object_deref(SDB_OBJ(m));
469 }
470 }
471 END_TEST
473 Suite *
474 core_store_lookup_suite(void)
475 {
476 Suite *s = suite_create("core::store_lookup");
477 TCase *tc;
479 tc = tcase_create("core");
480 tcase_add_checked_fixture(tc, populate, sdb_store_clear);
481 tcase_add_test(tc, test_store_match_name);
482 tcase_add_test(tc, test_store_match_attr);
483 tcase_add_test(tc, test_store_match_op);
484 tcase_add_test(tc, test_parse_cmp);
485 tcase_add_test(tc, test_lookup);
486 suite_add_tcase(s, tc);
488 return s;
489 } /* core_store_lookup_suite */
491 /* vim: set tw=78 sw=4 ts=4 noexpandtab : */