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 *metric;
44 } metrics[] = {
45 { "a", "m1" },
46 { "b", "m1" },
47 { "b", "m2" },
48 };
50 struct {
51 const char *host;
52 const char *service;
53 } services[] = {
54 { "a", "s1" },
55 { "a", "s2" },
56 { "b", "s1" },
57 { "b", "s3" },
58 };
60 struct {
61 const char *host;
62 const char *name;
63 sdb_data_t value;
64 } attrs[] = {
65 { "a", "k1", { SDB_TYPE_STRING, { .string = "v1" } } },
66 { "a", "k2", { SDB_TYPE_INTEGER, { .integer = 123 } } },
67 { "b", "k1", { SDB_TYPE_STRING, { .string = "v2" } } },
68 };
70 size_t i;
72 for (i = 0; i < SDB_STATIC_ARRAY_LEN(hosts); ++i) {
73 int status = sdb_store_host(hosts[i], 1);
74 fail_unless(status == 0,
75 "sdb_store_host(%s, 1) = %d; expected: 0",
76 hosts[i], status);
77 }
79 for (i = 0; i < SDB_STATIC_ARRAY_LEN(metrics); ++i) {
80 int status = sdb_store_metric(metrics[i].host,
81 metrics[i].metric, /* store */ NULL, 1);
82 fail_unless(status == 0,
83 "sdb_store_metric(%s, %s, NULL, 1) = %d; expected: 0",
84 metrics[i].host, metrics[i].metric, status);
85 }
87 for (i = 0; i < SDB_STATIC_ARRAY_LEN(services); ++i) {
88 int status = sdb_store_service(services[i].host,
89 services[i].service, 1);
90 fail_unless(status == 0,
91 "sdb_store_service(%s, %s, 1) = %d; expected: 0",
92 services[i].host, services[i].service, status);
93 }
95 for (i = 0; i < SDB_STATIC_ARRAY_LEN(attrs); ++i) {
96 int status = sdb_store_attribute(attrs[i].host,
97 attrs[i].name, &attrs[i].value, 1);
98 fail_unless(status == 0,
99 "sdb_store_attribute(%s, %s, <val>, 1) = %d; expected: 0",
100 attrs[i].host, attrs[i].name, status);
101 }
102 } /* populate */
104 START_TEST(test_store_match_name)
105 {
106 sdb_store_obj_t *obj;
108 struct {
109 int type;
110 const char *name;
111 _Bool re;
113 int expected;
114 } golden_data[] = {
115 { SDB_HOST, NULL, 0, 1 },
116 { SDB_HOST, NULL, 1, 1 },
117 { SDB_HOST, "a", 0, 1 },
118 { SDB_HOST, "a", 1, 1 },
119 { SDB_HOST, "b", 0, 0 },
120 { SDB_HOST, "b", 1, 0 },
121 { SDB_HOST, "^a$", 1, 1 },
122 { SDB_HOST, "^b$", 1, 0 },
123 { SDB_HOST, "^a$", 0, 0 },
124 { SDB_HOST, "^b$", 0, 0 },
125 { SDB_METRIC, NULL, 0, 1 },
126 { SDB_METRIC, NULL, 1, 1 },
127 { SDB_METRIC, "m1", 0, 1 },
128 { SDB_METRIC, "m1", 1, 1 },
129 { SDB_METRIC, "^m1$", 1, 1 },
130 { SDB_METRIC, "m", 1, 1 },
131 { SDB_METRIC, "s", 1, 0 },
132 { SDB_METRIC, "m2", 0, 0 },
133 { SDB_METRIC, "x1", 0, 0 },
134 { SDB_METRIC, "x1", 1, 0 },
135 { SDB_SERVICE, NULL, 0, 1 },
136 { SDB_SERVICE, NULL, 1, 1 },
137 { SDB_SERVICE, "s1", 0, 1 },
138 { SDB_SERVICE, "s2", 0, 1 },
139 { SDB_SERVICE, "s3", 0, 0 },
140 { SDB_SERVICE, "s4", 0, 0 },
141 { SDB_SERVICE, "^s1$", 1, 1 },
142 { SDB_SERVICE, "^s1$", 0, 0 },
143 { SDB_SERVICE, "x1", 0, 0 },
144 { SDB_SERVICE, "x1", 1, 0 },
145 { SDB_SERVICE, "x", 1, 0 },
146 { SDB_ATTRIBUTE, NULL, 0, 1 },
147 { SDB_ATTRIBUTE, NULL, 1, 1 },
148 { SDB_ATTRIBUTE, "k1", 0, 1 },
149 { SDB_ATTRIBUTE, "k2", 0, 1 },
150 { SDB_ATTRIBUTE, "k3", 0, 0 },
151 { SDB_ATTRIBUTE, "k4", 0, 0 },
152 { SDB_ATTRIBUTE, "k", 1, 1 },
153 { SDB_ATTRIBUTE, "1", 1, 1 },
154 { SDB_ATTRIBUTE, "3", 1, 0 },
155 };
157 size_t i;
159 obj = sdb_store_get_host("a");
160 fail_unless(obj != NULL,
161 "sdb_store_get_host(a) = NULL; expected: <host>");
163 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
164 sdb_store_matcher_t *m, *n;
165 char buf[1024];
166 int status;
168 m = sdb_store_name_matcher(golden_data[i].type,
169 golden_data[i].name, golden_data[i].re);
170 fail_unless(m != NULL,
171 "sdb_store_service_matcher(%d, %s, %d) = NULL; "
172 "expected: <matcher>", golden_data[i].type,
173 golden_data[i].name, golden_data[i].re);
175 status = sdb_store_matcher_matches(m, obj, /* filter */ NULL);
176 fail_unless(status == golden_data[i].expected,
177 "sdb_store_matcher_matches(%s, <host a>, NULL) = %d; "
178 "expected: %d",
179 sdb_store_matcher_tostring(m, buf, sizeof(buf)),
180 status, golden_data[i].expected);
182 n = sdb_store_inv_matcher(m);
183 fail_unless(n != NULL,
184 "sdb_store_inv_matcher() = NULL; expected: <matcher>");
185 sdb_object_deref(SDB_OBJ(m));
187 /* now match the inverted set of objects */
188 status = sdb_store_matcher_matches(n, obj, /* filter */ NULL);
189 fail_unless(status == !golden_data[i].expected,
190 "sdb_store_matcher_matches(%s, <host a>, NULL) = %d; "
191 "expected: %d",
192 sdb_store_matcher_tostring(n, buf, sizeof(buf)),
193 status, !golden_data[i].expected);
195 sdb_object_deref(SDB_OBJ(n));
196 }
198 sdb_object_deref(SDB_OBJ(obj));
199 }
200 END_TEST
202 START_TEST(test_store_match_attr)
203 {
204 sdb_store_obj_t *obj;
206 struct {
207 const char *attr_name;
208 const char *attr_value;
209 _Bool re;
211 int expected;
212 } golden_data[] = {
213 { "k1", NULL, 0, 1 },
214 { "k", NULL, 1, 0 },
215 { "1", NULL, 1, 0 },
216 { "k3", NULL, 0, 0 },
217 { "k3", NULL, 1, 0 },
218 { "k1", "v1", 0, 1 },
219 { "k1", "v1", 1, 1 },
220 { "k1", "^v1$", 1, 1 },
221 { "k1", "v", 1, 1 },
222 { "k1", "1", 1, 1 },
223 { "k1", "v2", 0, 0 },
224 { "k1", "v2", 1, 0 },
225 { "k", "v1", 0, 0 },
226 { "k", "v1", 1, 0 },
227 { "k3", "1", 0, 0 },
228 { "k3", "1", 1, 0 },
229 };
231 size_t i;
233 obj = sdb_store_get_host("a");
234 fail_unless(obj != NULL,
235 "sdb_store_get_host(a) = NULL; expected: <host>");
237 fail_unless(sdb_store_attr_matcher(NULL, "re", 0) == NULL,
238 "sdb_store_attr_matcher(NULL, \"re\", 0) = <m>; expected: NULL");
239 fail_unless(sdb_store_attr_matcher(NULL, "re", 1) == NULL,
240 "sdb_store_attr_matcher(NULL, \"re\", 1) = <m>; expected: NULL");
242 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
243 sdb_store_matcher_t *m, *n;
244 char buf[1024];
245 int status;
247 m = sdb_store_attr_matcher(golden_data[i].attr_name,
248 golden_data[i].attr_value, golden_data[i].re);
249 fail_unless(m != NULL,
250 "sdb_store_attr_matcher() = NULL; expected: <matcher>");
252 status = sdb_store_matcher_matches(m, obj, /* filter */ NULL);
253 fail_unless(status == golden_data[i].expected,
254 "sdb_store_matcher_matches({%s, <host a>, NULL) = %d; "
255 "expected: %d",
256 sdb_store_matcher_tostring(m, buf, sizeof(buf)),
257 status, golden_data[i].expected);
259 n = sdb_store_inv_matcher(m);
260 fail_unless(n != NULL,
261 "sdb_store_inv_matcher() = NULL; expected: <matcher>");
262 sdb_object_deref(SDB_OBJ(m));
264 /* now match the inverted set of objects */
265 status = sdb_store_matcher_matches(n, obj, /* filter */ NULL);
266 fail_unless(status == !golden_data[i].expected,
267 "sdb_store_matcher_matches({%s, <host a>, NULL) = %d; "
268 "expected: %d",
269 sdb_store_matcher_tostring(n, buf, sizeof(buf)),
270 status, !golden_data[i].expected);
272 sdb_object_deref(SDB_OBJ(n));
273 }
275 sdb_object_deref(SDB_OBJ(obj));
276 }
277 END_TEST
279 START_TEST(test_attr_cond)
280 {
281 sdb_store_obj_t *obj;
283 struct {
284 const char *attr;
285 const sdb_data_t value;
286 int expected_lt, expected_le, expected_eq, expected_ge, expected_gt;
287 } golden_data[] = {
288 { "k1", { SDB_TYPE_STRING, { .string = "v1" } }, 0, 1, 1, 1, 0 },
289 { "k1", { SDB_TYPE_STRING, { .string = "v2" } }, 1, 1, 0, 0, 0 },
290 { "k1", { SDB_TYPE_STRING, { .string = "v0" } }, 0, 0, 0, 1, 1 },
291 { "k1", { SDB_TYPE_STRING, { .string = "0" } }, 0, 0, 0, 1, 1 },
292 { "k2", { SDB_TYPE_INTEGER, { .integer = 123 } }, 0, 1, 1, 1, 0 },
293 { "k2", { SDB_TYPE_INTEGER, { .integer = 124 } }, 1, 1, 0, 0, 0 },
294 { "k2", { SDB_TYPE_INTEGER, { .integer = 122 } }, 0, 0, 0, 1, 1 },
295 /* key does not exist */
296 { "k3", { SDB_TYPE_STRING, { .string = "v1" } }, 0, 0, 0, 0, 0 },
297 { "k3", { SDB_TYPE_STRING, { .string = "123" } }, 0, 0, 0, 0, 0 },
298 { "k3", { SDB_TYPE_INTEGER, { .integer = 123 } }, 0, 0, 0, 0, 0 },
299 /* type mismatch */
300 { "k1", { SDB_TYPE_INTEGER, { .integer = 0 } }, 0, 0, 0, 1, 1 },
301 { "k2", { SDB_TYPE_STRING, { .string = "123" } }, 0, 1, 1, 1, 0 },
302 };
304 int status;
305 size_t i;
307 obj = sdb_store_get_host("a");
308 fail_unless(obj != NULL,
309 "sdb_store_get_host(a) = NULL; expected: <host>");
311 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
312 sdb_store_expr_t *expr;
313 sdb_store_cond_t *c;
314 char buf[1024];
315 size_t j;
317 struct {
318 sdb_store_matcher_t *(*matcher)(sdb_store_cond_t *);
319 int expected;
320 } tests[] = {
321 { sdb_store_lt_matcher, golden_data[i].expected_lt },
322 { sdb_store_le_matcher, golden_data[i].expected_le },
323 { sdb_store_eq_matcher, golden_data[i].expected_eq },
324 { sdb_store_ge_matcher, golden_data[i].expected_ge },
325 { sdb_store_gt_matcher, golden_data[i].expected_gt },
326 };
328 sdb_data_format(&golden_data[i].value,
329 buf, sizeof(buf), SDB_UNQUOTED);
331 expr = sdb_store_expr_constvalue(&golden_data[i].value);
332 fail_unless(expr != NULL,
333 "sdb_store_expr_constvalue(%s) = NULL; expected: <expr>",
334 buf);
336 c = sdb_store_attr_cond(golden_data[i].attr, expr);
337 sdb_object_deref(SDB_OBJ(expr));
338 fail_unless(c != NULL,
339 "sdb_store_attr_cond(%s, expr{%s}) = NULL; expected: <cond>",
340 golden_data[i].attr, buf);
342 for (j = 0; j < SDB_STATIC_ARRAY_LEN(tests); ++j) {
343 sdb_store_matcher_t *m;
344 char m_str[1024];
346 m = tests[j].matcher(c);
347 fail_unless(m != NULL,
348 "sdb_store_<cond>_matcher() = NULL; expected: <matcher>");
350 status = sdb_store_matcher_matches(m, obj, /* filter */ NULL);
351 fail_unless(status == tests[j].expected,
352 "sdb_store_matcher_matches(%s, <obj>, NULL) = %d; "
353 "expected: %d",
354 sdb_store_matcher_tostring(m, m_str, sizeof(m_str)),
355 status, tests[j].expected);
357 sdb_object_deref(SDB_OBJ(m));
358 }
360 sdb_object_deref(SDB_OBJ(c));
361 }
363 sdb_object_deref(SDB_OBJ(obj));
364 }
365 END_TEST
367 START_TEST(test_obj_cond)
368 {
369 struct {
370 const char *host;
371 int field;
372 const sdb_data_t value;
373 int expected_lt, expected_le, expected_eq, expected_ge, expected_gt;
374 } golden_data[] = {
375 { "b", SDB_FIELD_NAME,
376 { SDB_TYPE_STRING, { .string = "a" } }, 0, 0, 0, 1, 1 },
377 { "b", SDB_FIELD_NAME,
378 { SDB_TYPE_STRING, { .string = "b" } }, 0, 1, 1, 1, 0 },
379 { "b", SDB_FIELD_NAME,
380 { SDB_TYPE_STRING, { .string = "c" } }, 1, 1, 0, 0, 0 },
381 /* last-update = 1 for all objects */
382 { "a", SDB_FIELD_LAST_UPDATE,
383 { SDB_TYPE_DATETIME, { .datetime = 1 } }, 0, 1, 1, 1, 0 },
384 { "a", SDB_FIELD_LAST_UPDATE,
385 { SDB_TYPE_DATETIME, { .datetime = 2 } }, 1, 1, 0, 0, 0 },
386 { "a", SDB_FIELD_LAST_UPDATE,
387 { SDB_TYPE_DATETIME, { .datetime = 0 } }, 0, 0, 0, 1, 1 },
388 /* age > 0 for all objects */
389 { "a", SDB_FIELD_AGE,
390 { SDB_TYPE_DATETIME, { .datetime = 0 } }, 0, 0, 0, 1, 1 },
391 /* interval = 0 for all objects */
392 { "a", SDB_FIELD_INTERVAL,
393 { SDB_TYPE_DATETIME, { .datetime = 0 } }, 0, 1, 1, 1, 0 },
394 { "a", SDB_FIELD_INTERVAL,
395 { SDB_TYPE_DATETIME, { .datetime = 1 } }, 1, 1, 0, 0, 0 },
396 /* type mismatch */
397 { "a", SDB_FIELD_LAST_UPDATE,
398 { SDB_TYPE_INTEGER, { .integer = 0 } }, 0, 0, 0, 0, 0 },
399 { "a", SDB_FIELD_AGE,
400 { SDB_TYPE_INTEGER, { .integer = 0 } }, 0, 0, 0, 0, 0 },
401 { "a", SDB_FIELD_INTERVAL,
402 { SDB_TYPE_INTEGER, { .integer = 0 } }, 0, 0, 0, 0, 0 },
403 { "a", SDB_FIELD_BACKEND,
404 { SDB_TYPE_INTEGER, { .integer = 0 } }, 0, 0, 0, 0, 0 },
405 { "a", SDB_FIELD_BACKEND,
406 { SDB_TYPE_INTEGER, { .integer = 0 } }, 0, 0, 0, 0, 0 },
407 /* (64bit) integer value without zero-bytes */
408 { "a", SDB_FIELD_BACKEND,
409 { SDB_TYPE_INTEGER, { .integer = 0xffffffffffffffffL } },
410 0, 0, 0, 0, 0 },
411 };
413 int status;
414 size_t i;
416 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
417 sdb_store_obj_t *obj;
418 sdb_store_expr_t *expr;
419 sdb_store_cond_t *c;
420 char buf[1024];
421 size_t j;
423 struct {
424 sdb_store_matcher_t *(*matcher)(sdb_store_cond_t *);
425 int expected;
426 } tests[] = {
427 { sdb_store_lt_matcher, golden_data[i].expected_lt },
428 { sdb_store_le_matcher, golden_data[i].expected_le },
429 { sdb_store_eq_matcher, golden_data[i].expected_eq },
430 { sdb_store_ge_matcher, golden_data[i].expected_ge },
431 { sdb_store_gt_matcher, golden_data[i].expected_gt },
432 };
434 obj = sdb_store_get_host(golden_data[i].host);
435 fail_unless(obj != NULL,
436 "sdb_store_get_host(%s) = NULL; expected: <host>",
437 golden_data[i].host);
439 sdb_data_format(&golden_data[i].value,
440 buf, sizeof(buf), SDB_UNQUOTED);
442 expr = sdb_store_expr_constvalue(&golden_data[i].value);
443 fail_unless(expr != NULL,
444 "sdb_store_expr_constvalue(%s) = NULL; expected: <expr>",
445 buf);
447 c = sdb_store_obj_cond(golden_data[i].field, expr);
448 sdb_object_deref(SDB_OBJ(expr));
449 fail_unless(c != NULL,
450 "sdb_store_obj_cond(%d, expr{%s}) = NULL; expected: <cond>",
451 golden_data[i].field, buf);
453 for (j = 0; j < SDB_STATIC_ARRAY_LEN(tests); ++j) {
454 sdb_store_matcher_t *m;
455 char m_str[1024];
457 m = tests[j].matcher(c);
458 fail_unless(m != NULL,
459 "sdb_store_<cond>_matcher() = NULL; expected: <matcher>");
461 status = sdb_store_matcher_matches(m, obj, /* filter */ NULL);
462 fail_unless(status == tests[j].expected,
463 "sdb_store_matcher_matches(%s, <host '%s'>, NULL) = %d; "
464 "expected: %d",
465 sdb_store_matcher_tostring(m, m_str, sizeof(m_str)),
466 status, tests[j].expected);
468 sdb_object_deref(SDB_OBJ(m));
469 }
471 sdb_object_deref(SDB_OBJ(c));
472 sdb_object_deref(SDB_OBJ(obj));
473 }
474 }
475 END_TEST
477 START_TEST(test_store_match_op)
478 {
479 sdb_store_obj_t *obj;
481 sdb_store_matcher_t *always = sdb_store_name_matcher(SDB_HOST, "a", 0);
482 sdb_store_matcher_t *never = sdb_store_name_matcher(SDB_HOST, "z", 0);
484 struct {
485 const char *op;
486 sdb_store_matcher_t *left;
487 sdb_store_matcher_t *right;
488 int expected;
489 } golden_data[] = {
490 { "OR", always, always, 1 },
491 { "OR", always, never, 1 },
492 { "OR", never, always, 1 },
493 { "OR", never, never, 0 },
494 { "AND", always, always, 1 },
495 { "AND", always, never, 0 },
496 { "AND", never, always, 0 },
497 { "AND", never, never, 0 },
498 };
500 int status;
501 size_t i;
503 obj = sdb_store_get_host("a");
505 status = sdb_store_matcher_matches(always, obj, /* filter */ NULL);
506 fail_unless(status == 1,
507 "INTERNAL ERROR: 'always' did not match host");
508 status = sdb_store_matcher_matches(never, obj, /* filter */ NULL);
509 fail_unless(status == 0,
510 "INTERNAL ERROR: 'never' matches host");
512 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
513 sdb_store_matcher_t *m;
515 if (! strcmp(golden_data[i].op, "OR"))
516 m = sdb_store_dis_matcher(golden_data[i].left,
517 golden_data[i].right);
518 else if (! strcmp(golden_data[i].op, "AND"))
519 m = sdb_store_con_matcher(golden_data[i].left,
520 golden_data[i].right);
521 else {
522 fail("INTERNAL ERROR: unexpected operator %s", golden_data[i].op);
523 continue;
524 }
526 #define TO_NAME(v) (((v) == always) ? "always" \
527 : ((v) == never) ? "never" : "<unknown>")
529 status = sdb_store_matcher_matches(m, obj, /* filter */ NULL);
530 fail_unless(status == golden_data[i].expected,
531 "%s(%s, %s, NULL) = %d; expected: %d", golden_data[i].op,
532 TO_NAME(golden_data[i].left), TO_NAME(golden_data[i].right),
533 status, golden_data[i].expected);
535 #undef TO_NAME
537 sdb_object_deref(SDB_OBJ(m));
538 }
540 sdb_object_deref(SDB_OBJ(always));
541 sdb_object_deref(SDB_OBJ(never));
543 sdb_object_deref(SDB_OBJ(obj));
544 }
545 END_TEST
547 START_TEST(test_parse_cmp)
548 {
549 sdb_data_t hostname = { SDB_TYPE_STRING, { .string = "hostname" } };
550 sdb_data_t metricname = { SDB_TYPE_STRING, { .string = "metricname" } };
551 sdb_data_t srvname = { SDB_TYPE_STRING, { .string = "srvname" } };
552 sdb_data_t attrname = { SDB_TYPE_STRING, { .string = "attrname" } };
553 sdb_data_t attrvalue = { SDB_TYPE_INTEGER, { .integer = 4711 } };
555 sdb_store_matcher_t *check;
557 size_t i;
559 struct {
560 const char *obj_type;
561 const char *attr;
562 const char *op;
563 const sdb_data_t *value;
564 int expected;
565 } golden_data[] = {
566 { "host", NULL, "=", &hostname, MATCHER_NAME },
567 { "host", NULL, "!=", &hostname, MATCHER_NOT },
568 { "host", NULL, "=~", &hostname, MATCHER_NAME },
569 { "host", NULL, "!~", &hostname, MATCHER_NOT },
570 { "host", "attr", "=", &hostname, -1 },
571 { "host", "attr", "!=", &hostname, -1 },
572 { "host", "attr", "!=", &attrvalue, -1 },
573 { "host", "attr", "<=", &attrvalue, -1 },
574 { "host", NULL, "&^", &hostname, -1 },
575 { "host", NULL, "<", &hostname, -1 },
576 { "host", NULL, "<=", &hostname, -1 },
577 { "host", NULL, ">=", &hostname, -1 },
578 { "host", NULL, ">", &hostname, -1 },
579 { "host", NULL, "=", NULL, -1 },
580 { "metric", NULL, "=", &metricname, MATCHER_NAME },
581 { "metric", NULL, "!=", &metricname, MATCHER_NOT },
582 { "metric", NULL, "=~", &metricname, MATCHER_NAME },
583 { "metric", NULL, "!~", &metricname, MATCHER_NOT },
584 { "metric", "attr", "=", &metricname, -1 },
585 { "metric", "attr", "!=", &metricname, -1 },
586 { "metric", "attr", "!=", &attrvalue, -1 },
587 { "metric", "attr", "<=", &attrvalue, -1 },
588 { "metric", NULL, "&^", &metricname, -1 },
589 { "metric", NULL, "<", &metricname, -1 },
590 { "metric", NULL, "<=", &metricname, -1 },
591 { "metric", NULL, ">=", &metricname, -1 },
592 { "metric", NULL, ">", &metricname, -1 },
593 { "metric", NULL, "=", NULL, -1 },
594 { "service", NULL, "=", &srvname, MATCHER_NAME },
595 { "service", NULL, "!=", &srvname, MATCHER_NOT },
596 { "service", NULL, "=~", &srvname, MATCHER_NAME },
597 { "service", NULL, "!~", &srvname, MATCHER_NOT },
598 { "service", "attr", "=", &srvname, -1 },
599 { "service", "attr", "!=", &srvname, -1 },
600 { "service", "attr", "!=", &attrvalue, -1 },
601 { "service", "attr", "<=", &attrvalue, -1 },
602 { "service", NULL, "&^", &srvname, -1 },
603 { "service", NULL, "<", &srvname, -1 },
604 { "service", NULL, "<=", &srvname, -1 },
605 { "service", NULL, ">=", &srvname, -1 },
606 { "service", NULL, ">", &srvname, -1 },
607 { "service", NULL, "=", NULL, -1 },
608 { "attribute", NULL, "=", &attrname, MATCHER_NAME },
609 { "attribute", NULL, "!=", &attrname, MATCHER_NOT },
610 { "attribute", NULL, "=~", &attrname, MATCHER_NAME },
611 { "attribute", NULL, "!~", &attrname, MATCHER_NOT },
612 { "attribute", NULL, "<", &attrname, -1 },
613 { "attribute", NULL, "<=", &attrname, -1 },
614 { "attribute", NULL, ">=", &attrname, -1 },
615 { "attribute", NULL, ">", &attrname, -1 },
616 { "attribute", NULL, "=", NULL, -1 },
617 { "attribute", "attr", "=", &attrname, MATCHER_ATTR },
618 { "attribute", "attr", "!=", &attrname, MATCHER_NOT },
619 { "attribute", "attr", "=~", &attrname, MATCHER_ATTR },
620 { "attribute", "attr", "!~", &attrname, MATCHER_NOT },
621 { "attribute", "attr", "&^", &attrname, -1 },
622 { "attribute", "attr", "<", NULL, -1 },
623 { "attribute", "attr", "<", &attrname, MATCHER_LT },
624 { "attribute", "attr", "<=", &attrname, MATCHER_LE },
625 /* { "attribute", "attr", "=", &attrname, MATCHER_EQ }, */
626 { "attribute", "attr", ">=", &attrname, MATCHER_GE },
627 { "attribute", "attr", ">", &attrname, MATCHER_GT },
628 { "foo", NULL, "=", &attrname, -1 },
629 { "foo", "attr", "=", &attrname, -1 },
630 };
632 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
633 sdb_store_expr_t *expr;
634 char buf[1024];
636 if (sdb_data_format(golden_data[i].value,
637 buf, sizeof(buf), SDB_UNQUOTED) < 0)
638 snprintf(buf, sizeof(buf), "ERR");
640 expr = sdb_store_expr_constvalue(golden_data[i].value);
641 fail_unless(expr != NULL || golden_data[i].value == NULL,
642 "sdb_store_expr_constvalue(%s) = NULL; expected: <expr>",
643 buf);
645 check = sdb_store_matcher_parse_cmp(golden_data[i].obj_type,
646 golden_data[i].attr, golden_data[i].op, expr);
647 sdb_object_deref(SDB_OBJ(expr));
649 if (golden_data[i].expected == -1) {
650 fail_unless(check == NULL,
651 "sdb_store_matcher_parse_cmp(%s, %s, %s, expr{%s}) = %p; "
652 "expected: NULL", golden_data[i].obj_type,
653 golden_data[i].attr, golden_data[i].op, buf, check);
654 continue;
655 }
657 fail_unless(check != NULL,
658 "sdb_store_matcher_parse_cmp(%s, %s, %s, %s) = %p; "
659 "expected: <expr>", golden_data[i].obj_type,
660 golden_data[i].attr, golden_data[i].op, buf, check);
661 fail_unless(M(check)->type == golden_data[i].expected,
662 "sdb_store_matcher_parse_cmp(%s, %s, %s, %s) returned matcher "
663 "of type %d; expected: %d", golden_data[i].obj_type,
664 golden_data[i].attr, golden_data[i].op, buf,
665 M(check)->type, golden_data[i].expected);
667 sdb_object_deref(SDB_OBJ(check));
668 }
669 }
670 END_TEST
672 START_TEST(test_parse_field_cmp)
673 {
674 sdb_data_t datetime = { SDB_TYPE_DATETIME, { .datetime = 1 } };
675 sdb_data_t string = { SDB_TYPE_STRING, { .string = "s" } };
677 struct {
678 const char *field;
679 const char *op;
680 const sdb_data_t *value;
681 int expected;
682 } golden_data[] = {
683 { "name", "<", &string, MATCHER_LT },
684 { "name", "<=", &string, MATCHER_LE },
685 { "name", "=", &string, MATCHER_EQ },
686 { "name", ">=", &string, MATCHER_GE },
687 { "name", ">", &string, MATCHER_GT },
688 { "name", "!=", &string, MATCHER_NOT },
689 { "last_update", "<", &datetime, MATCHER_LT },
690 { "last_update", "<=", &datetime, MATCHER_LE },
691 { "last_update", "=", &datetime, MATCHER_EQ },
692 { "last_update", ">=", &datetime, MATCHER_GE },
693 { "last_update", ">", &datetime, MATCHER_GT },
694 { "last_update", "!=", &datetime, MATCHER_NOT },
695 { "age", "<", &datetime, MATCHER_LT },
696 { "age", "<=", &datetime, MATCHER_LE },
697 { "age", "=", &datetime, MATCHER_EQ },
698 { "age", ">=", &datetime, MATCHER_GE },
699 { "age", ">", &datetime, MATCHER_GT },
700 { "age", "!=", &datetime, MATCHER_NOT },
701 { "interval", "<", &datetime, MATCHER_LT },
702 { "interval", "<=", &datetime, MATCHER_LE },
703 { "interval", "=", &datetime, MATCHER_EQ },
704 { "interval", ">=", &datetime, MATCHER_GE },
705 { "interval", ">", &datetime, MATCHER_GT },
706 { "interval", "!=", &datetime, MATCHER_NOT },
707 { "backend", "=", &string, MATCHER_EQ },
708 { "backend", "!=", &string, MATCHER_NOT },
709 /* the behavior for other operators on .backend
710 * is currently unspecified */
711 { "last_update", "=", NULL, -1 },
712 { "last_update", "IS", NULL, -1 },
713 { "age", "=", NULL, -1 },
714 { "interval", "=", NULL, -1 },
715 { "backend", "=", NULL, -1 },
716 { "backend", "=~", &string, -1 },
717 };
719 size_t i;
721 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
722 sdb_store_matcher_t *check;
723 sdb_store_expr_t *expr;
724 char buf[1024];
726 if (sdb_data_format(golden_data[i].value,
727 buf, sizeof(buf), SDB_UNQUOTED) < 0)
728 snprintf(buf, sizeof(buf), "ERR");
730 expr = sdb_store_expr_constvalue(golden_data[i].value);
731 fail_unless(expr != NULL || golden_data[i].value == NULL,
732 "sdb_store_expr_constvalue(%s) = NULL; expected: <expr>",
733 buf);
735 check = sdb_store_matcher_parse_field_cmp(golden_data[i].field,
736 golden_data[i].op, expr);
737 sdb_object_deref(SDB_OBJ(expr));
739 if (golden_data[i].expected == -1) {
740 fail_unless(check == NULL,
741 "sdb_store_matcher_parse_field_cmp(%s, %s, expr{%s}) = %p; "
742 "expected: NULL", golden_data[i].field,
743 golden_data[i].op, buf, check);
744 continue;
745 }
747 fail_unless(check != NULL,
748 "sdb_store_matcher_parse_field_cmp(%s, %s, %s) = %p; "
749 "expected: NULL", golden_data[i].field,
750 golden_data[i].op, buf, check);
751 fail_unless(M(check)->type == golden_data[i].expected,
752 "sdb_store_matcher_parse_field_cmp(%s, %s, %s) returned "
753 "matcher of type %d; expected: %d", golden_data[i].field,
754 golden_data[i].op, buf, M(check)->type,
755 golden_data[i].expected);
757 sdb_object_deref(SDB_OBJ(check));
758 }
759 }
760 END_TEST
762 static int
763 scan_cb(sdb_store_obj_t *obj, void *user_data)
764 {
765 int *i = user_data;
767 fail_unless(obj != NULL,
768 "sdb_store_scan callback received NULL obj; expected: "
769 "<store base obj>");
770 fail_unless(i != NULL,
771 "sdb_store_scan callback received NULL user_data; "
772 "expected: <pointer to data>");
774 ++(*i);
775 return 0;
776 } /* scan_cb */
778 START_TEST(test_scan)
779 {
780 #define PTR_RE "0x[0-9a-f]+"
781 struct {
782 const char *query;
783 const char *filter;
784 int expected;
785 const char *tostring_re;
786 } golden_data[] = {
787 { "host = 'a'", NULL, 1,
788 "OBJ\\[host\\]\\{ NAME\\{ 'a', \\(nil\\) \\} \\}" },
789 { "host = 'a'", "host = 'x'", 0, /* filter never matches */
790 "OBJ\\[host\\]\\{ NAME\\{ 'a', \\(nil\\) \\} \\}" },
791 { "host = 'a'",
792 "NOT attribute['x'] = ''", 1, /* filter always matches */
793 "OBJ\\[host\\]\\{ NAME\\{ 'a', \\(nil\\) \\} \\}" },
794 { "host =~ 'a|b'", NULL, 2,
795 "OBJ\\[host\\]\\{ NAME\\{ NULL, "PTR_RE" \\} \\}" },
796 { "host =~ 'host'", NULL, 0,
797 "OBJ\\[host\\]\\{ NAME\\{ NULL, "PTR_RE" \\} \\}" },
798 { "host =~ '.'", NULL, 3,
799 "OBJ\\[host\\]\\{ NAME\\{ NULL, "PTR_RE" \\} \\}" },
800 { "metric = 'm1'", NULL, 2,
801 "OBJ\\[metric\\]\\{ NAME\\{ 'm1', \\(nil\\) } \\}" },
802 { "metric= 'm1'", "host = 'x'", 0, /* filter never matches */
803 "OBJ\\[metric\\]\\{ NAME\\{ 'm1', \\(nil\\) } \\}" },
804 { "metric = 'm1'",
805 "NOT attribute['x'] = ''", 2, /* filter always matches */
806 "OBJ\\[metric\\]\\{ NAME\\{ 'm1', \\(nil\\) } \\}" },
807 { "metric =~ 'm'", NULL, 2,
808 "OBJ\\[metric\\]\\{ NAME\\{ NULL, "PTR_RE" } \\}" },
809 { "metric !~ 'm'", NULL, 1,
810 "\\(NOT, OBJ\\[metric\\]\\{ NAME\\{ NULL, "PTR_RE" } \\}\\)" },
811 { "metric =~ 'x'", NULL, 0,
812 "OBJ\\[metric\\]\\{ NAME\\{ NULL, "PTR_RE" } \\}" },
813 { "service = 's1'", NULL, 2,
814 "OBJ\\[service\\]\\{ NAME\\{ 's1', \\(nil\\) } \\}" },
815 { "service = 's1'", "host = 'x'", 0, /* filter never matches */
816 "OBJ\\[service\\]\\{ NAME\\{ 's1', \\(nil\\) } \\}" },
817 { "service = 's1'",
818 "NOT attribute['x'] = ''", 2, /* filter always matches */
819 "OBJ\\[service\\]\\{ NAME\\{ 's1', \\(nil\\) } \\}" },
820 { "service =~ 's'", NULL, 2,
821 "OBJ\\[service\\]\\{ NAME\\{ NULL, "PTR_RE" } \\}" },
822 { "service !~ 's'", NULL, 1,
823 "\\(NOT, OBJ\\[service\\]\\{ NAME\\{ NULL, "PTR_RE" } \\}\\)" },
824 { "attribute = 'k1'", NULL, 2,
825 "OBJ\\[attribute\\]\\{ NAME\\{ 'k1', \\(nil\\) \\} " },
826 { "attribute = 'k1'", "host = 'x'", 0, /* filter never matches */
827 "OBJ\\[attribute\\]\\{ NAME\\{ 'k1', \\(nil\\) \\} " },
828 { "attribute = 'k1'",
829 "NOT attribute['x'] = ''", 2, /* filter always matches */
830 "OBJ\\[attribute\\]\\{ NAME\\{ 'k1', \\(nil\\) \\} " },
831 { "attribute = 'x'", NULL, 0,
832 "OBJ\\[attribute\\]\\{ NAME\\{ 'x', \\(nil\\) \\}" },
833 { "attribute['k1'] = 'v1'", NULL, 1,
834 "CMP_MATCHER\\(15\\)" },
835 { "attribute['k1'] IS NULL", NULL, 1,
836 "\\(IS NULL\\)" },
837 { "attribute['x1'] IS NULL", NULL, 3,
838 "\\(IS NULL\\)" },
839 { "attribute['k1'] IS NOT NULL", NULL, 2,
840 "\\(IS NOT NULL\\)" },
841 { "attribute['x1'] IS NOT NULL", NULL, 0,
842 "\\(IS NOT NULL\\)" },
843 { "attribute['k2'] < 123", NULL, 0,
844 "CMP_MATCHER\\(13\\)" },
845 { "attribute['k2'] <= 123", NULL, 1,
846 "CMP_MATCHER\\(14\\)" },
847 { "attribute['k2'] >= 123", NULL, 1,
848 "CMP_MATCHER\\(17\\)" },
849 { "attribute['k2'] > 123", NULL, 0,
850 "CMP_MATCHER\\(18\\)" },
851 { "attribute['k2'] = 123", NULL, 1,
852 "CMP_MATCHER\\(15\\)" },
853 { "attribute['k2'] != 123", NULL, 0,
854 "CMP_MATCHER\\(16\\)" },
855 { "attribute['k1'] != 'v1'", NULL, 1,
856 "CMP_MATCHER\\(16\\)" },
857 { "attribute['k1'] != 'v2'", NULL, 1,
858 "CMP_MATCHER\\(16\\)" },
859 { "attribute != 'x' "
860 "AND attribute['y'] !~ 'x'", NULL, 3,
861 "\\(AND, "
862 "\\(NOT, OBJ\\[attribute\\]\\{ NAME\\{ 'x', \\(nil\\) \\} \\}\\), "
863 "CMP_MATCHER\\(21\\)" },
864 };
866 int check, n;
867 size_t i;
869 n = 0;
870 check = sdb_store_scan(/* matcher */ NULL, /* filter */ NULL,
871 scan_cb, &n);
872 fail_unless(check == 0,
873 "sdb_store_scan() = %d; expected: 0", check);
874 fail_unless(n == 3,
875 "sdb_store_scan called callback %d times; expected: 3", (int)n);
877 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
878 sdb_store_matcher_t *m, *filter = NULL;
879 char buf[4096];
881 m = sdb_fe_parse_matcher(golden_data[i].query, -1);
882 fail_unless(m != NULL,
883 "sdb_fe_parse_matcher(%s, -1) = NULL; expected: <matcher>",
884 golden_data[i].query);
886 if (golden_data[i].filter) {
887 filter = sdb_fe_parse_matcher(golden_data[i].filter, -1);
888 fail_unless(filter != NULL,
889 "sdb_fe_parse_matcher(%s, -1) = NULL; "
890 "expected: <matcher>", golden_data[i].filter);
891 }
893 fail_unless(sdb_regmatches(golden_data[i].tostring_re,
894 sdb_store_matcher_tostring(m, buf, sizeof(buf))) == 0,
895 "sdb_fe_parse_matcher(%s, -1) = %s; expected: %s",
896 golden_data[i].query,
897 sdb_store_matcher_tostring(m, buf, sizeof(buf)),
898 golden_data[i].tostring_re);
900 n = 0;
901 sdb_store_scan(m, filter, scan_cb, &n);
902 fail_unless(n == golden_data[i].expected,
903 "sdb_store_scan(matcher{%s}, filter{%s}) found %d hosts; "
904 "expected: %d", golden_data[i].query, golden_data[i].filter,
905 n, golden_data[i].expected);
906 sdb_object_deref(SDB_OBJ(filter));
907 sdb_object_deref(SDB_OBJ(m));
908 }
909 }
910 END_TEST
912 Suite *
913 core_store_lookup_suite(void)
914 {
915 Suite *s = suite_create("core::store_lookup");
916 TCase *tc;
918 tc = tcase_create("core");
919 tcase_add_checked_fixture(tc, populate, sdb_store_clear);
920 tcase_add_test(tc, test_store_match_name);
921 tcase_add_test(tc, test_store_match_attr);
922 tcase_add_test(tc, test_attr_cond);
923 tcase_add_test(tc, test_obj_cond);
924 tcase_add_test(tc, test_store_match_op);
925 tcase_add_test(tc, test_parse_cmp);
926 tcase_add_test(tc, test_parse_field_cmp);
927 tcase_add_test(tc, test_scan);
928 suite_add_tcase(s, tc);
930 return s;
931 } /* core_store_lookup_suite */
933 /* vim: set tw=78 sw=4 ts=4 noexpandtab : */