754eb13f18808fba615ed1e809ac4d4b39a526e7
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 };
69 size_t i;
71 for (i = 0; i < SDB_STATIC_ARRAY_LEN(hosts); ++i) {
72 int status = sdb_store_host(hosts[i], 1);
73 fail_unless(status == 0,
74 "sdb_store_host(%s, 1) = %d; expected: 0",
75 hosts[i], status);
76 }
78 for (i = 0; i < SDB_STATIC_ARRAY_LEN(metrics); ++i) {
79 int status = sdb_store_metric(metrics[i].host,
80 metrics[i].metric, /* store */ NULL, 1);
81 fail_unless(status == 0,
82 "sdb_store_metric(%s, %s, NULL, 1) = %d; expected: 0",
83 metrics[i].host, metrics[i].metric, status);
84 }
86 for (i = 0; i < SDB_STATIC_ARRAY_LEN(services); ++i) {
87 int status = sdb_store_service(services[i].host,
88 services[i].service, 1);
89 fail_unless(status == 0,
90 "sdb_store_service(%s, %s, 1) = %d; expected: 0",
91 services[i].host, services[i].service, status);
92 }
94 for (i = 0; i < SDB_STATIC_ARRAY_LEN(attrs); ++i) {
95 int status = sdb_store_attribute(attrs[i].host,
96 attrs[i].name, &attrs[i].value, 1);
97 fail_unless(status == 0,
98 "sdb_store_attribute(%s, %s, <val>, 1) = %d; expected: 0",
99 attrs[i].host, attrs[i].name, status);
100 }
101 } /* populate */
103 START_TEST(test_store_match_name)
104 {
105 sdb_store_obj_t *obj;
107 struct {
108 int type;
109 const char *name;
110 _Bool re;
112 int expected;
113 } golden_data[] = {
114 { SDB_HOST, NULL, 0, 1 },
115 { SDB_HOST, NULL, 1, 1 },
116 { SDB_HOST, "a", 0, 1 },
117 { SDB_HOST, "a", 1, 1 },
118 { SDB_HOST, "b", 0, 0 },
119 { SDB_HOST, "b", 1, 0 },
120 { SDB_HOST, "^a$", 1, 1 },
121 { SDB_HOST, "^b$", 1, 0 },
122 { SDB_HOST, "^a$", 0, 0 },
123 { SDB_HOST, "^b$", 0, 0 },
124 { SDB_METRIC, NULL, 0, 1 },
125 { SDB_METRIC, NULL, 1, 1 },
126 { SDB_METRIC, "m1", 0, 1 },
127 { SDB_METRIC, "m1", 1, 1 },
128 { SDB_METRIC, "^m1$", 1, 1 },
129 { SDB_METRIC, "m", 1, 1 },
130 { SDB_METRIC, "s", 1, 0 },
131 { SDB_METRIC, "m2", 0, 0 },
132 { SDB_METRIC, "x1", 0, 0 },
133 { SDB_METRIC, "x1", 1, 0 },
134 { SDB_SERVICE, NULL, 0, 1 },
135 { SDB_SERVICE, NULL, 1, 1 },
136 { SDB_SERVICE, "s1", 0, 1 },
137 { SDB_SERVICE, "s2", 0, 1 },
138 { SDB_SERVICE, "s3", 0, 0 },
139 { SDB_SERVICE, "s4", 0, 0 },
140 { SDB_SERVICE, "^s1$", 1, 1 },
141 { SDB_SERVICE, "^s1$", 0, 0 },
142 { SDB_SERVICE, "x1", 0, 0 },
143 { SDB_SERVICE, "x1", 1, 0 },
144 { SDB_SERVICE, "x", 1, 0 },
145 { SDB_ATTRIBUTE, NULL, 0, 1 },
146 { SDB_ATTRIBUTE, NULL, 1, 1 },
147 { SDB_ATTRIBUTE, "k1", 0, 1 },
148 { SDB_ATTRIBUTE, "k2", 0, 1 },
149 { SDB_ATTRIBUTE, "k3", 0, 0 },
150 { SDB_ATTRIBUTE, "k4", 0, 0 },
151 { SDB_ATTRIBUTE, "k", 1, 1 },
152 { SDB_ATTRIBUTE, "1", 1, 1 },
153 { SDB_ATTRIBUTE, "3", 1, 0 },
154 };
156 size_t i;
158 obj = sdb_store_get_host("a");
159 fail_unless(obj != NULL,
160 "sdb_store_get_host(a) = NULL; expected: <host>");
162 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
163 sdb_store_matcher_t *m, *n;
164 char buf[1024];
165 int status;
167 m = sdb_store_name_matcher(golden_data[i].type,
168 golden_data[i].name, golden_data[i].re);
169 fail_unless(m != NULL,
170 "sdb_store_service_matcher(%d, %s, %d) = NULL; "
171 "expected: <matcher>", golden_data[i].type,
172 golden_data[i].name, golden_data[i].re);
174 status = sdb_store_matcher_matches(m, obj, /* filter */ NULL);
175 fail_unless(status == golden_data[i].expected,
176 "sdb_store_matcher_matches(%s, <host a>, NULL) = %d; "
177 "expected: %d",
178 sdb_store_matcher_tostring(m, buf, sizeof(buf)),
179 status, golden_data[i].expected);
181 n = sdb_store_inv_matcher(m);
182 fail_unless(n != NULL,
183 "sdb_store_inv_matcher() = NULL; expected: <matcher>");
184 sdb_object_deref(SDB_OBJ(m));
186 /* now match the inverted set of objects */
187 status = sdb_store_matcher_matches(n, obj, /* filter */ NULL);
188 fail_unless(status == !golden_data[i].expected,
189 "sdb_store_matcher_matches(%s, <host a>, NULL) = %d; "
190 "expected: %d",
191 sdb_store_matcher_tostring(n, buf, sizeof(buf)),
192 status, !golden_data[i].expected);
194 sdb_object_deref(SDB_OBJ(n));
195 }
197 sdb_object_deref(SDB_OBJ(obj));
198 }
199 END_TEST
201 START_TEST(test_store_match_attr)
202 {
203 sdb_store_obj_t *obj;
205 struct {
206 const char *attr_name;
207 const char *attr_value;
208 _Bool re;
210 int expected;
211 } golden_data[] = {
212 { "k1", NULL, 0, 1 },
213 { "k", NULL, 1, 0 },
214 { "1", NULL, 1, 0 },
215 { "k3", NULL, 0, 0 },
216 { "k3", NULL, 1, 0 },
217 { "k1", "v1", 0, 1 },
218 { "k1", "v1", 1, 1 },
219 { "k1", "^v1$", 1, 1 },
220 { "k1", "v", 1, 1 },
221 { "k1", "1", 1, 1 },
222 { "k1", "v2", 0, 0 },
223 { "k1", "v2", 1, 0 },
224 { "k", "v1", 0, 0 },
225 { "k", "v1", 1, 0 },
226 { "k3", "1", 0, 0 },
227 { "k3", "1", 1, 0 },
228 };
230 size_t i;
232 obj = sdb_store_get_host("a");
233 fail_unless(obj != NULL,
234 "sdb_store_get_host(a) = NULL; expected: <host>");
236 fail_unless(sdb_store_attr_matcher(NULL, "re", 0) == NULL,
237 "sdb_store_attr_matcher(NULL, \"re\", 0) = <m>; expected: NULL");
238 fail_unless(sdb_store_attr_matcher(NULL, "re", 1) == NULL,
239 "sdb_store_attr_matcher(NULL, \"re\", 1) = <m>; expected: NULL");
241 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
242 sdb_store_matcher_t *m, *n;
243 char buf[1024];
244 int status;
246 m = sdb_store_attr_matcher(golden_data[i].attr_name,
247 golden_data[i].attr_value, golden_data[i].re);
248 fail_unless(m != NULL,
249 "sdb_store_attr_matcher() = NULL; expected: <matcher>");
251 status = sdb_store_matcher_matches(m, obj, /* filter */ NULL);
252 fail_unless(status == golden_data[i].expected,
253 "sdb_store_matcher_matches({%s, <host a>, NULL) = %d; "
254 "expected: %d",
255 sdb_store_matcher_tostring(m, buf, sizeof(buf)),
256 status, golden_data[i].expected);
258 n = sdb_store_inv_matcher(m);
259 fail_unless(n != NULL,
260 "sdb_store_inv_matcher() = NULL; expected: <matcher>");
261 sdb_object_deref(SDB_OBJ(m));
263 /* now match the inverted set of objects */
264 status = sdb_store_matcher_matches(n, obj, /* filter */ NULL);
265 fail_unless(status == !golden_data[i].expected,
266 "sdb_store_matcher_matches({%s, <host a>, NULL) = %d; "
267 "expected: %d",
268 sdb_store_matcher_tostring(n, buf, sizeof(buf)),
269 status, !golden_data[i].expected);
271 sdb_object_deref(SDB_OBJ(n));
272 }
274 sdb_object_deref(SDB_OBJ(obj));
275 }
276 END_TEST
278 START_TEST(test_attr_cond)
279 {
280 sdb_store_obj_t *obj;
282 struct {
283 const char *attr;
284 const sdb_data_t value;
285 int expected_lt, expected_le, expected_eq, expected_ge, expected_gt;
286 } golden_data[] = {
287 { "k1", { SDB_TYPE_STRING, { .string = "v1" } }, 0, 1, 1, 1, 0 },
288 { "k1", { SDB_TYPE_STRING, { .string = "v2" } }, 1, 1, 0, 0, 0 },
289 { "k1", { SDB_TYPE_STRING, { .string = "v0" } }, 0, 0, 0, 1, 1 },
290 { "k2", { SDB_TYPE_INTEGER, { .integer = 123 } }, 0, 1, 1, 1, 0 },
291 { "k2", { SDB_TYPE_INTEGER, { .integer = 124 } }, 1, 1, 0, 0, 0 },
292 { "k2", { SDB_TYPE_INTEGER, { .integer = 122 } }, 0, 0, 0, 1, 1 },
293 /* key does not exist */
294 { "k3", { SDB_TYPE_STRING, { .string = "v1" } }, 0, 0, 0, 0, 0 },
295 { "k3", { SDB_TYPE_STRING, { .string = "123" } }, 0, 0, 0, 0, 0 },
296 { "k3", { SDB_TYPE_INTEGER, { .integer = 123 } }, 0, 0, 0, 0, 0 },
297 /* type mismatch */
298 { "k1", { SDB_TYPE_INTEGER, { .integer = 0 } }, 0, 0, 0, 0, 0 },
299 { "k2", { SDB_TYPE_STRING, { .string = "123" } }, 0, 0, 0, 0, 0 },
300 };
302 int status;
303 size_t i;
305 obj = sdb_store_get_host("a");
306 fail_unless(obj != NULL,
307 "sdb_store_get_host(a) = NULL; expected: <host>");
309 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
310 sdb_store_expr_t *expr;
311 sdb_store_cond_t *c;
312 char buf[1024];
313 size_t j;
315 struct {
316 sdb_store_matcher_t *(*matcher)(sdb_store_cond_t *);
317 int expected;
318 } tests[] = {
319 { sdb_store_lt_matcher, golden_data[i].expected_lt },
320 { sdb_store_le_matcher, golden_data[i].expected_le },
321 { sdb_store_eq_matcher, golden_data[i].expected_eq },
322 { sdb_store_ge_matcher, golden_data[i].expected_ge },
323 { sdb_store_gt_matcher, golden_data[i].expected_gt },
324 };
326 sdb_data_format(&golden_data[i].value,
327 buf, sizeof(buf), SDB_UNQUOTED);
329 expr = sdb_store_expr_constvalue(&golden_data[i].value);
330 fail_unless(expr != NULL,
331 "sdb_store_expr_constvalue(%s) = NULL; expected: <expr>",
332 buf);
334 c = sdb_store_attr_cond(golden_data[i].attr, expr);
335 sdb_object_deref(SDB_OBJ(expr));
336 fail_unless(c != NULL,
337 "sdb_store_attr_cond(%s, expr{%s}) = NULL; expected: <cond>",
338 golden_data[i].attr, buf);
340 for (j = 0; j < SDB_STATIC_ARRAY_LEN(tests); ++j) {
341 sdb_store_matcher_t *m;
342 char m_str[1024];
344 m = tests[j].matcher(c);
345 fail_unless(m != NULL,
346 "sdb_store_<cond>_matcher() = NULL; expected: <matcher>");
348 status = sdb_store_matcher_matches(m, obj, /* filter */ NULL);
349 fail_unless(status == tests[j].expected,
350 "sdb_store_matcher_matches(%s, <obj>, NULL) = %d; "
351 "expected: %d",
352 sdb_store_matcher_tostring(m, m_str, sizeof(m_str)),
353 status, tests[j].expected);
355 sdb_object_deref(SDB_OBJ(m));
356 }
358 sdb_object_deref(SDB_OBJ(c));
359 }
361 sdb_object_deref(SDB_OBJ(obj));
362 }
363 END_TEST
365 START_TEST(test_obj_cond)
366 {
367 struct {
368 const char *host;
369 int field;
370 const sdb_data_t value;
371 int expected_lt, expected_le, expected_eq, expected_ge, expected_gt;
372 } golden_data[] = {
373 /* last-update = 1 for all objects */
374 { "a", SDB_FIELD_LAST_UPDATE,
375 { SDB_TYPE_DATETIME, { .datetime = 1 } }, 0, 1, 1, 1, 0 },
376 { "a", SDB_FIELD_LAST_UPDATE,
377 { SDB_TYPE_DATETIME, { .datetime = 2 } }, 1, 1, 0, 0, 0 },
378 { "a", SDB_FIELD_LAST_UPDATE,
379 { SDB_TYPE_DATETIME, { .datetime = 0 } }, 0, 0, 0, 1, 1 },
380 /* age > 0 for all objects */
381 { "a", SDB_FIELD_AGE,
382 { SDB_TYPE_DATETIME, { .datetime = 0 } }, 0, 0, 0, 1, 1 },
383 /* interval = 0 for all objects */
384 { "a", SDB_FIELD_INTERVAL,
385 { SDB_TYPE_DATETIME, { .datetime = 0 } }, 0, 1, 1, 1, 0 },
386 { "a", SDB_FIELD_INTERVAL,
387 { SDB_TYPE_DATETIME, { .datetime = 1 } }, 1, 1, 0, 0, 0 },
388 /* type mismatch */
389 { "a", SDB_FIELD_LAST_UPDATE,
390 { SDB_TYPE_INTEGER, { .integer = 0 } }, 0, 0, 0, 0, 0 },
391 { "a", SDB_FIELD_AGE,
392 { SDB_TYPE_INTEGER, { .integer = 0 } }, 0, 0, 0, 0, 0 },
393 { "a", SDB_FIELD_INTERVAL,
394 { SDB_TYPE_INTEGER, { .integer = 0 } }, 0, 0, 0, 0, 0 },
395 { "a", SDB_FIELD_BACKEND,
396 { SDB_TYPE_INTEGER, { .integer = 0 } }, 0, 0, 0, 0, 0 },
397 };
399 int status;
400 size_t i;
402 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
403 sdb_store_obj_t *obj;
404 sdb_store_expr_t *expr;
405 sdb_store_cond_t *c;
406 char buf[1024];
407 size_t j;
409 struct {
410 sdb_store_matcher_t *(*matcher)(sdb_store_cond_t *);
411 int expected;
412 } tests[] = {
413 { sdb_store_lt_matcher, golden_data[i].expected_lt },
414 { sdb_store_le_matcher, golden_data[i].expected_le },
415 { sdb_store_eq_matcher, golden_data[i].expected_eq },
416 { sdb_store_ge_matcher, golden_data[i].expected_ge },
417 { sdb_store_gt_matcher, golden_data[i].expected_gt },
418 };
420 obj = sdb_store_get_host(golden_data[i].host);
421 fail_unless(obj != NULL,
422 "sdb_store_get_host(%s) = NULL; expected: <host>",
423 golden_data[i].host);
425 sdb_data_format(&golden_data[i].value,
426 buf, sizeof(buf), SDB_UNQUOTED);
428 expr = sdb_store_expr_constvalue(&golden_data[i].value);
429 fail_unless(expr != NULL,
430 "sdb_store_expr_constvalue(%s) = NULL; expected: <expr>",
431 buf);
433 c = sdb_store_obj_cond(golden_data[i].field, expr);
434 sdb_object_deref(SDB_OBJ(expr));
435 fail_unless(c != NULL,
436 "sdb_store_obj_cond(%d, expr{%s}) = NULL; expected: <cond>",
437 golden_data[i].field, buf);
439 for (j = 0; j < SDB_STATIC_ARRAY_LEN(tests); ++j) {
440 sdb_store_matcher_t *m;
441 char m_str[1024];
443 m = tests[j].matcher(c);
444 fail_unless(m != NULL,
445 "sdb_store_<cond>_matcher() = NULL; expected: <matcher>");
447 status = sdb_store_matcher_matches(m, obj, /* filter */ NULL);
448 fail_unless(status == tests[j].expected,
449 "sdb_store_matcher_matches(%s, <obj>, NULL) = %d; "
450 "expected: %d",
451 sdb_store_matcher_tostring(m, m_str, sizeof(m_str)),
452 status, tests[j].expected);
454 sdb_object_deref(SDB_OBJ(m));
455 }
457 sdb_object_deref(SDB_OBJ(c));
458 sdb_object_deref(SDB_OBJ(obj));
459 }
460 }
461 END_TEST
463 START_TEST(test_store_match_op)
464 {
465 sdb_store_obj_t *obj;
467 sdb_store_matcher_t *always = sdb_store_name_matcher(SDB_HOST, "a", 0);
468 sdb_store_matcher_t *never = sdb_store_name_matcher(SDB_HOST, "z", 0);
470 struct {
471 const char *op;
472 sdb_store_matcher_t *left;
473 sdb_store_matcher_t *right;
474 int expected;
475 } golden_data[] = {
476 { "OR", always, always, 1 },
477 { "OR", always, never, 1 },
478 { "OR", never, always, 1 },
479 { "OR", never, never, 0 },
480 { "AND", always, always, 1 },
481 { "AND", always, never, 0 },
482 { "AND", never, always, 0 },
483 { "AND", never, never, 0 },
484 };
486 int status;
487 size_t i;
489 obj = sdb_store_get_host("a");
491 status = sdb_store_matcher_matches(always, obj, /* filter */ NULL);
492 fail_unless(status == 1,
493 "INTERNAL ERROR: 'always' did not match host");
494 status = sdb_store_matcher_matches(never, obj, /* filter */ NULL);
495 fail_unless(status == 0,
496 "INTERNAL ERROR: 'never' matches host");
498 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
499 sdb_store_matcher_t *m;
501 if (! strcmp(golden_data[i].op, "OR"))
502 m = sdb_store_dis_matcher(golden_data[i].left,
503 golden_data[i].right);
504 else if (! strcmp(golden_data[i].op, "AND"))
505 m = sdb_store_con_matcher(golden_data[i].left,
506 golden_data[i].right);
507 else {
508 fail("INTERNAL ERROR: unexpected operator %s", golden_data[i].op);
509 continue;
510 }
512 #define TO_NAME(v) (((v) == always) ? "always" \
513 : ((v) == never) ? "never" : "<unknown>")
515 status = sdb_store_matcher_matches(m, obj, /* filter */ NULL);
516 fail_unless(status == golden_data[i].expected,
517 "%s(%s, %s, NULL) = %d; expected: %d", golden_data[i].op,
518 TO_NAME(golden_data[i].left), TO_NAME(golden_data[i].right),
519 status, golden_data[i].expected);
521 #undef TO_NAME
523 sdb_object_deref(SDB_OBJ(m));
524 }
526 sdb_object_deref(SDB_OBJ(always));
527 sdb_object_deref(SDB_OBJ(never));
529 sdb_object_deref(SDB_OBJ(obj));
530 }
531 END_TEST
533 START_TEST(test_parse_cmp)
534 {
535 sdb_data_t hostname = { SDB_TYPE_STRING, { .string = "hostname" } };
536 sdb_data_t metricname = { SDB_TYPE_STRING, { .string = "metricname" } };
537 sdb_data_t srvname = { SDB_TYPE_STRING, { .string = "srvname" } };
538 sdb_data_t attrname = { SDB_TYPE_STRING, { .string = "attrname" } };
539 sdb_data_t attrvalue = { SDB_TYPE_INTEGER, { .integer = 4711 } };
541 sdb_store_matcher_t *check;
543 size_t i;
545 struct {
546 const char *obj_type;
547 const char *attr;
548 const char *op;
549 const sdb_data_t *value;
550 int expected;
551 } golden_data[] = {
552 { "host", NULL, "=", &hostname, MATCHER_NAME },
553 { "host", NULL, "!=", &hostname, MATCHER_NOT },
554 { "host", NULL, "=~", &hostname, MATCHER_NAME },
555 { "host", NULL, "!~", &hostname, MATCHER_NOT },
556 { "host", "attr", "=", &hostname, -1 },
557 { "host", "attr", "!=", &hostname, -1 },
558 { "host", "attr", "!=", &attrvalue, -1 },
559 { "host", "attr", "<=", &attrvalue, -1 },
560 { "host", NULL, "&^", &hostname, -1 },
561 { "host", NULL, "<", &hostname, -1 },
562 { "host", NULL, "<=", &hostname, -1 },
563 { "host", NULL, ">=", &hostname, -1 },
564 { "host", NULL, ">", &hostname, -1 },
565 { "host", NULL, "=", NULL, -1 },
566 { "metric", NULL, "=", &metricname, MATCHER_NAME },
567 { "metric", NULL, "!=", &metricname, MATCHER_NOT },
568 { "metric", NULL, "=~", &metricname, MATCHER_NAME },
569 { "metric", NULL, "!~", &metricname, MATCHER_NOT },
570 { "metric", "attr", "=", &metricname, -1 },
571 { "metric", "attr", "!=", &metricname, -1 },
572 { "metric", "attr", "!=", &attrvalue, -1 },
573 { "metric", "attr", "<=", &attrvalue, -1 },
574 { "metric", NULL, "&^", &metricname, -1 },
575 { "metric", NULL, "<", &metricname, -1 },
576 { "metric", NULL, "<=", &metricname, -1 },
577 { "metric", NULL, ">=", &metricname, -1 },
578 { "metric", NULL, ">", &metricname, -1 },
579 { "metric", NULL, "=", NULL, -1 },
580 { "service", NULL, "=", &srvname, MATCHER_NAME },
581 { "service", NULL, "!=", &srvname, MATCHER_NOT },
582 { "service", NULL, "=~", &srvname, MATCHER_NAME },
583 { "service", NULL, "!~", &srvname, MATCHER_NOT },
584 { "service", "attr", "=", &srvname, -1 },
585 { "service", "attr", "!=", &srvname, -1 },
586 { "service", "attr", "!=", &attrvalue, -1 },
587 { "service", "attr", "<=", &attrvalue, -1 },
588 { "service", NULL, "&^", &srvname, -1 },
589 { "service", NULL, "<", &srvname, -1 },
590 { "service", NULL, "<=", &srvname, -1 },
591 { "service", NULL, ">=", &srvname, -1 },
592 { "service", NULL, ">", &srvname, -1 },
593 { "service", NULL, "=", NULL, -1 },
594 { "attribute", NULL, "=", &attrname, MATCHER_NAME },
595 { "attribute", NULL, "!=", &attrname, MATCHER_NOT },
596 { "attribute", NULL, "=~", &attrname, MATCHER_NAME },
597 { "attribute", NULL, "!~", &attrname, MATCHER_NOT },
598 { "attribute", NULL, "<", &attrname, -1 },
599 { "attribute", NULL, "<=", &attrname, -1 },
600 { "attribute", NULL, ">=", &attrname, -1 },
601 { "attribute", NULL, ">", &attrname, -1 },
602 { "attribute", NULL, "=", NULL, -1 },
603 { "attribute", "attr", "=", &attrname, MATCHER_ATTR },
604 { "attribute", "attr", "!=", &attrname, MATCHER_NOT },
605 { "attribute", "attr", "=~", &attrname, MATCHER_ATTR },
606 { "attribute", "attr", "!~", &attrname, MATCHER_NOT },
607 { "attribute", "attr", "&^", &attrname, -1 },
608 { "attribute", "attr", "<", NULL, -1 },
609 { "attribute", "attr", "<", &attrname, MATCHER_LT },
610 { "attribute", "attr", "<=", &attrname, MATCHER_LE },
611 /* { "attribute", "attr", "=", &attrname, MATCHER_EQ }, */
612 { "attribute", "attr", ">=", &attrname, MATCHER_GE },
613 { "attribute", "attr", ">", &attrname, MATCHER_GT },
614 { "attribute", "attr", "IS", NULL, MATCHER_ISNULL },
615 { "attribute", "attr", "IS", &attrname, -1 },
616 { "foo", NULL, "=", &attrname, -1 },
617 { "foo", "attr", "=", &attrname, -1 },
618 };
620 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
621 sdb_store_expr_t *expr;
622 char buf[1024];
624 if (sdb_data_format(golden_data[i].value,
625 buf, sizeof(buf), SDB_UNQUOTED) < 0)
626 snprintf(buf, sizeof(buf), "ERR");
628 expr = sdb_store_expr_constvalue(golden_data[i].value);
629 fail_unless(expr != NULL || golden_data[i].value == NULL,
630 "sdb_store_expr_constvalue(%s) = NULL; expected: <expr>",
631 buf);
633 check = sdb_store_matcher_parse_cmp(golden_data[i].obj_type,
634 golden_data[i].attr, golden_data[i].op, expr);
635 sdb_object_deref(SDB_OBJ(expr));
637 if (golden_data[i].expected == -1) {
638 fail_unless(check == NULL,
639 "sdb_store_matcher_parse_cmp(%s, %s, %s, expr{%s}) = %p; "
640 "expected: NULL", golden_data[i].obj_type,
641 golden_data[i].attr, golden_data[i].op, buf, check);
642 continue;
643 }
645 fail_unless(check != NULL,
646 "sdb_store_matcher_parse_cmp(%s, %s, %s, %s) = %p; "
647 "expected: NULL", golden_data[i].obj_type,
648 golden_data[i].attr, golden_data[i].op, buf, check);
649 fail_unless(M(check)->type == golden_data[i].expected,
650 "sdb_store_matcher_parse_cmp(%s, %s, %s, %s) returned matcher "
651 "of type %d; expected: %d", golden_data[i].obj_type,
652 golden_data[i].attr, golden_data[i].op, buf,
653 M(check)->type, golden_data[i].expected);
655 sdb_object_deref(SDB_OBJ(check));
656 }
657 }
658 END_TEST
660 START_TEST(test_parse_field_cmp)
661 {
662 sdb_data_t datetime = { SDB_TYPE_DATETIME, { .datetime = 1 } };
663 sdb_data_t string = { SDB_TYPE_STRING, { .string = "s" } };
665 struct {
666 const char *field;
667 const char *op;
668 const sdb_data_t *value;
669 int expected;
670 } golden_data[] = {
671 { "last_update", "<", &datetime, MATCHER_LT },
672 { "last_update", "<=", &datetime, MATCHER_LE },
673 { "last_update", "=", &datetime, MATCHER_EQ },
674 { "last_update", ">=", &datetime, MATCHER_GE },
675 { "last_update", ">", &datetime, MATCHER_GT },
676 { "last_update", "!=", &datetime, MATCHER_NOT },
677 { "age", "<", &datetime, MATCHER_LT },
678 { "age", "<=", &datetime, MATCHER_LE },
679 { "age", "=", &datetime, MATCHER_EQ },
680 { "age", ">=", &datetime, MATCHER_GE },
681 { "age", ">", &datetime, MATCHER_GT },
682 { "age", "!=", &datetime, MATCHER_NOT },
683 { "interval", "<", &datetime, MATCHER_LT },
684 { "interval", "<=", &datetime, MATCHER_LE },
685 { "interval", "=", &datetime, MATCHER_EQ },
686 { "interval", ">=", &datetime, MATCHER_GE },
687 { "interval", ">", &datetime, MATCHER_GT },
688 { "interval", "!=", &datetime, MATCHER_NOT },
689 { "backend", "=", &string, MATCHER_EQ },
690 { "backend", "!=", &string, MATCHER_NOT },
691 /* the behavior for other operators on :backend
692 * is currently unspecified */
693 { "last_update", "=", NULL, -1 },
694 { "last_update", "IS", NULL, -1 },
695 { "age", "=", NULL, -1 },
696 { "interval", "=", NULL, -1 },
697 { "backend", "=", NULL, -1 },
698 { "backend", "=~", &string, -1 },
699 };
701 size_t i;
703 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
704 sdb_store_matcher_t *check;
705 sdb_store_expr_t *expr;
706 char buf[1024];
708 if (sdb_data_format(golden_data[i].value,
709 buf, sizeof(buf), SDB_UNQUOTED) < 0)
710 snprintf(buf, sizeof(buf), "ERR");
712 expr = sdb_store_expr_constvalue(golden_data[i].value);
713 fail_unless(expr != NULL || golden_data[i].value == NULL,
714 "sdb_store_expr_constvalue(%s) = NULL; expected: <expr>",
715 buf);
717 check = sdb_store_matcher_parse_field_cmp(golden_data[i].field,
718 golden_data[i].op, expr);
719 sdb_object_deref(SDB_OBJ(expr));
721 if (golden_data[i].expected == -1) {
722 fail_unless(check == NULL,
723 "sdb_store_matcher_parse_field_cmp(%s, %s, expr{%s}) = %p; "
724 "expected: NULL", golden_data[i].field,
725 golden_data[i].op, buf, check);
726 continue;
727 }
729 fail_unless(check != NULL,
730 "sdb_store_matcher_parse_field_cmp(%s, %s, %s) = %p; "
731 "expected: NULL", golden_data[i].field,
732 golden_data[i].op, buf, check);
733 fail_unless(M(check)->type == golden_data[i].expected,
734 "sdb_store_matcher_parse_field_cmp(%s, %s, %s) returned "
735 "matcher of type %d; expected: %d", golden_data[i].field,
736 golden_data[i].op, buf, M(check)->type,
737 golden_data[i].expected);
739 sdb_object_deref(SDB_OBJ(check));
740 }
741 }
742 END_TEST
744 static int
745 scan_cb(sdb_store_obj_t *obj, void *user_data)
746 {
747 int *i = user_data;
749 fail_unless(obj != NULL,
750 "sdb_store_scan callback received NULL obj; expected: "
751 "<store base obj>");
752 fail_unless(i != NULL,
753 "sdb_store_scan callback received NULL user_data; "
754 "expected: <pointer to data>");
756 ++(*i);
757 return 0;
758 } /* scan_cb */
760 START_TEST(test_scan)
761 {
762 #define PTR_RE "0x[0-9a-f]+"
763 struct {
764 const char *query;
765 const char *filter;
766 int expected;
767 const char *tostring_re;
768 } golden_data[] = {
769 { "host = 'a'", NULL, 1,
770 "OBJ\\[host\\]\\{ NAME\\{ 'a', \\(nil\\) \\} \\}" },
771 { "host = 'a'", "host = 'x'", 0, /* filter never matches */
772 "OBJ\\[host\\]\\{ NAME\\{ 'a', \\(nil\\) \\} \\}" },
773 { "host = 'a'",
774 "NOT attribute.x = ''", 1, /* filter always matches */
775 "OBJ\\[host\\]\\{ NAME\\{ 'a', \\(nil\\) \\} \\}" },
776 { "host =~ 'a|b'", NULL, 2,
777 "OBJ\\[host\\]\\{ NAME\\{ NULL, "PTR_RE" \\} \\}" },
778 { "host =~ 'host'", NULL, 0,
779 "OBJ\\[host\\]\\{ NAME\\{ NULL, "PTR_RE" \\} \\}" },
780 { "host =~ '.'", NULL, 3,
781 "OBJ\\[host\\]\\{ NAME\\{ NULL, "PTR_RE" \\} \\}" },
782 { "metric = 'm1'", NULL, 2,
783 "OBJ\\[metric\\]\\{ NAME\\{ 'm1', \\(nil\\) } \\}" },
784 { "metric= 'm1'", "host = 'x'", 0, /* filter never matches */
785 "OBJ\\[metric\\]\\{ NAME\\{ 'm1', \\(nil\\) } \\}" },
786 { "metric = 'm1'",
787 "NOT attribute.x = ''", 2, /* filter always matches */
788 "OBJ\\[metric\\]\\{ NAME\\{ 'm1', \\(nil\\) } \\}" },
789 { "metric =~ 'm'", NULL, 2,
790 "OBJ\\[metric\\]\\{ NAME\\{ NULL, "PTR_RE" } \\}" },
791 { "metric !~ 'm'", NULL, 1,
792 "\\(NOT, OBJ\\[metric\\]\\{ NAME\\{ NULL, "PTR_RE" } \\}\\)" },
793 { "metric =~ 'x'", NULL, 0,
794 "OBJ\\[metric\\]\\{ NAME\\{ NULL, "PTR_RE" } \\}" },
795 { "service = 's1'", NULL, 2,
796 "OBJ\\[service\\]\\{ NAME\\{ 's1', \\(nil\\) } \\}" },
797 { "service = 's1'", "host = 'x'", 0, /* filter never matches */
798 "OBJ\\[service\\]\\{ NAME\\{ 's1', \\(nil\\) } \\}" },
799 { "service = 's1'",
800 "NOT attribute.x = ''", 2, /* filter always matches */
801 "OBJ\\[service\\]\\{ NAME\\{ 's1', \\(nil\\) } \\}" },
802 { "service =~ 's'", NULL, 2,
803 "OBJ\\[service\\]\\{ NAME\\{ NULL, "PTR_RE" } \\}" },
804 { "service !~ 's'", NULL, 1,
805 "\\(NOT, OBJ\\[service\\]\\{ NAME\\{ NULL, "PTR_RE" } \\}\\)" },
806 { "attribute = 'k1'", NULL, 1,
807 "OBJ\\[attribute\\]\\{ NAME\\{ 'k1', \\(nil\\) \\} " },
808 { "attribute = 'k1'", "host = 'x'", 0, /* filter never matches */
809 "OBJ\\[attribute\\]\\{ NAME\\{ 'k1', \\(nil\\) \\} " },
810 { "attribute = 'k1'",
811 "NOT attribute.x = ''", 1, /* filter always matches */
812 "OBJ\\[attribute\\]\\{ NAME\\{ 'k1', \\(nil\\) \\} " },
813 { "attribute = 'x'", NULL, 0,
814 "OBJ\\[attribute\\]\\{ NAME\\{ 'x', \\(nil\\) \\}" },
815 { "attribute.k1 = 'v1'", NULL, 1,
816 "ATTR\\[k1\\]\\{ VALUE\\{ 'v1', \\(nil\\) \\} \\}" },
817 { "attribute.k1 IS NULL", NULL, 2,
818 "\\(IS NULL, ATTR\\[k1\\]\\)" },
819 { "attribute.x1 IS NULL", NULL, 3,
820 "\\(IS NULL, ATTR\\[x1\\]\\)" },
821 { "attribute.k1 IS NOT NULL", NULL, 1,
822 "\\(NOT, \\(IS NULL, ATTR\\[k1\\]\\)\\)" },
823 { "attribute.x1 IS NOT NULL", NULL, 0,
824 "\\(NOT, \\(IS NULL, ATTR\\[x1\\]\\)\\)" },
825 { "attribute.k2 < 123", NULL, 0,
826 "ATTR\\[k2\\]\\{ < 123 \\}" },
827 { "attribute.k2 <= 123", NULL, 1,
828 "ATTR\\[k2\\]\\{ <= 123 \\}" },
829 { "attribute.k2 >= 123", NULL, 1,
830 "ATTR\\[k2\\]\\{ >= 123 \\}" },
831 { "attribute.k2 > 123", NULL, 0,
832 "ATTR\\[k2\\]\\{ > 123 \\}" },
833 { "attribute.k2 = 123", NULL, 1,
834 "ATTR\\[k2\\]\\{ = 123 \\}" },
835 { "attribute.k2 != 123", NULL, 2,
836 "\\(NOT, ATTR\\[k2\\]\\{ = 123 \\}\\)" },
837 { "attribute.k1 != 'v1'", NULL, 2,
838 "\\(NOT, ATTR\\[k1\\]\\{ VALUE\\{ 'v1', \\(nil\\) \\} \\}\\)" },
839 { "attribute.k1 != 'v2'", NULL, 3,
840 "\\(NOT, ATTR\\[k1\\]\\{ VALUE\\{ 'v2', \\(nil\\) \\} \\}\\)" },
841 { "attribute != 'x' "
842 "AND attribute.y !~ 'x'", NULL, 3,
843 "\\(AND, "
844 "\\(NOT, OBJ\\[attribute\\]\\{ NAME\\{ 'x', \\(nil\\) \\} \\}\\), "
845 "\\(NOT, ATTR\\[y\\]\\{ VALUE\\{ NULL, "PTR_RE" \\} \\}\\)\\)" },
846 };
848 int check, n;
849 size_t i;
851 n = 0;
852 check = sdb_store_scan(/* matcher */ NULL, /* filter */ NULL,
853 scan_cb, &n);
854 fail_unless(check == 0,
855 "sdb_store_scan() = %d; expected: 0", check);
856 fail_unless(n == 3,
857 "sdb_store_scan called callback %d times; expected: 3", (int)n);
859 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
860 sdb_store_matcher_t *m, *filter = NULL;
861 char buf[4096];
863 m = sdb_fe_parse_matcher(golden_data[i].query, -1);
864 fail_unless(m != NULL,
865 "sdb_fe_parse_matcher(%s, -1) = NULL; expected: <matcher>",
866 golden_data[i].query);
868 if (golden_data[i].filter) {
869 filter = sdb_fe_parse_matcher(golden_data[i].filter, -1);
870 fail_unless(filter != NULL,
871 "sdb_fe_parse_matcher(%s, -1) = NULL; "
872 "expected: <matcher>", golden_data[i].filter);
873 }
875 fail_unless(sdb_regmatches(golden_data[i].tostring_re,
876 sdb_store_matcher_tostring(m, buf, sizeof(buf))) == 0,
877 "sdb_fe_parse_matcher(%s, -1) = %s; expected: %s",
878 golden_data[i].query,
879 sdb_store_matcher_tostring(m, buf, sizeof(buf)),
880 golden_data[i].tostring_re);
882 n = 0;
883 sdb_store_scan(m, filter, scan_cb, &n);
884 fail_unless(n == golden_data[i].expected,
885 "sdb_store_scan(matcher{%s}, filter{%s}) found %d hosts; "
886 "expected: %d", golden_data[i].query, golden_data[i].filter,
887 n, golden_data[i].expected);
888 sdb_object_deref(SDB_OBJ(filter));
889 sdb_object_deref(SDB_OBJ(m));
890 }
891 }
892 END_TEST
894 Suite *
895 core_store_lookup_suite(void)
896 {
897 Suite *s = suite_create("core::store_lookup");
898 TCase *tc;
900 tc = tcase_create("core");
901 tcase_add_checked_fixture(tc, populate, sdb_store_clear);
902 tcase_add_test(tc, test_store_match_name);
903 tcase_add_test(tc, test_store_match_attr);
904 tcase_add_test(tc, test_attr_cond);
905 tcase_add_test(tc, test_obj_cond);
906 tcase_add_test(tc, test_store_match_op);
907 tcase_add_test(tc, test_parse_cmp);
908 tcase_add_test(tc, test_parse_field_cmp);
909 tcase_add_test(tc, test_scan);
910 suite_add_tcase(s, tc);
912 return s;
913 } /* core_store_lookup_suite */
915 /* vim: set tw=78 sw=4 ts=4 noexpandtab : */