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 <assert.h>
35 #include <check.h>
36 #include <string.h>
38 static void
39 populate(void)
40 {
41 const char *hosts[] = { "a", "b", "c" };
43 struct {
44 const char *host;
45 const char *metric;
46 } metrics[] = {
47 { "a", "m1" },
48 { "b", "m1" },
49 { "b", "m2" },
50 };
52 struct {
53 const char *host;
54 const char *service;
55 } services[] = {
56 { "a", "s1" },
57 { "a", "s2" },
58 { "b", "s1" },
59 { "b", "s3" },
60 };
62 struct {
63 const char *host;
64 const char *name;
65 sdb_data_t value;
66 } attrs[] = {
67 { "a", "k1", { SDB_TYPE_STRING, { .string = "v1" } } },
68 { "a", "k2", { SDB_TYPE_INTEGER, { .integer = 123 } } },
69 { "b", "k1", { SDB_TYPE_STRING, { .string = "v2" } } },
70 };
72 size_t i;
74 for (i = 0; i < SDB_STATIC_ARRAY_LEN(hosts); ++i) {
75 int status = sdb_store_host(hosts[i], 1);
76 fail_unless(status == 0,
77 "sdb_store_host(%s, 1) = %d; expected: 0",
78 hosts[i], status);
79 }
81 for (i = 0; i < SDB_STATIC_ARRAY_LEN(metrics); ++i) {
82 int status = sdb_store_metric(metrics[i].host,
83 metrics[i].metric, /* store */ NULL, 1);
84 fail_unless(status == 0,
85 "sdb_store_metric(%s, %s, NULL, 1) = %d; expected: 0",
86 metrics[i].host, metrics[i].metric, status);
87 }
89 for (i = 0; i < SDB_STATIC_ARRAY_LEN(services); ++i) {
90 int status = sdb_store_service(services[i].host,
91 services[i].service, 1);
92 fail_unless(status == 0,
93 "sdb_store_service(%s, %s, 1) = %d; expected: 0",
94 services[i].host, services[i].service, status);
95 }
97 for (i = 0; i < SDB_STATIC_ARRAY_LEN(attrs); ++i) {
98 int status = sdb_store_attribute(attrs[i].host,
99 attrs[i].name, &attrs[i].value, 1);
100 fail_unless(status == 0,
101 "sdb_store_attribute(%s, %s, <val>, 1) = %d; expected: 0",
102 attrs[i].host, attrs[i].name, status);
103 }
104 } /* populate */
106 START_TEST(test_store_match_name)
107 {
108 sdb_store_obj_t *obj;
110 struct {
111 int type;
112 const char *name;
113 _Bool re;
115 int expected;
116 } golden_data[] = {
117 { SDB_HOST, NULL, 0, 1 },
118 { SDB_HOST, NULL, 1, 1 },
119 { SDB_HOST, "a", 0, 1 },
120 { SDB_HOST, "a", 1, 1 },
121 { SDB_HOST, "b", 0, 0 },
122 { SDB_HOST, "b", 1, 0 },
123 { SDB_HOST, "^a$", 1, 1 },
124 { SDB_HOST, "^b$", 1, 0 },
125 { SDB_HOST, "^a$", 0, 0 },
126 { SDB_HOST, "^b$", 0, 0 },
127 { SDB_METRIC, NULL, 0, 1 },
128 { SDB_METRIC, NULL, 1, 1 },
129 { SDB_METRIC, "m1", 0, 1 },
130 { SDB_METRIC, "m1", 1, 1 },
131 { SDB_METRIC, "^m1$", 1, 1 },
132 { SDB_METRIC, "m", 1, 1 },
133 { SDB_METRIC, "s", 1, 0 },
134 { SDB_METRIC, "m2", 0, 0 },
135 { SDB_METRIC, "x1", 0, 0 },
136 { SDB_METRIC, "x1", 1, 0 },
137 { SDB_SERVICE, NULL, 0, 1 },
138 { SDB_SERVICE, NULL, 1, 1 },
139 { SDB_SERVICE, "s1", 0, 1 },
140 { SDB_SERVICE, "s2", 0, 1 },
141 { SDB_SERVICE, "s3", 0, 0 },
142 { SDB_SERVICE, "s4", 0, 0 },
143 { SDB_SERVICE, "^s1$", 1, 1 },
144 { SDB_SERVICE, "^s1$", 0, 0 },
145 { SDB_SERVICE, "x1", 0, 0 },
146 { SDB_SERVICE, "x1", 1, 0 },
147 { SDB_SERVICE, "x", 1, 0 },
148 { SDB_ATTRIBUTE, NULL, 0, 1 },
149 { SDB_ATTRIBUTE, NULL, 1, 1 },
150 { SDB_ATTRIBUTE, "k1", 0, 1 },
151 { SDB_ATTRIBUTE, "k2", 0, 1 },
152 { SDB_ATTRIBUTE, "k3", 0, 0 },
153 { SDB_ATTRIBUTE, "k4", 0, 0 },
154 { SDB_ATTRIBUTE, "k", 1, 1 },
155 { SDB_ATTRIBUTE, "1", 1, 1 },
156 { SDB_ATTRIBUTE, "3", 1, 0 },
157 };
159 size_t i;
161 obj = sdb_store_get_host("a");
162 fail_unless(obj != NULL,
163 "sdb_store_get_host(a) = NULL; expected: <host>");
165 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
166 sdb_store_matcher_t *m, *n;
167 int status;
169 m = sdb_store_name_matcher(golden_data[i].type,
170 golden_data[i].name, golden_data[i].re);
171 fail_unless(m != NULL,
172 "sdb_store_service_matcher(%d, %s, %d) = NULL; "
173 "expected: <matcher>", golden_data[i].type,
174 golden_data[i].name, golden_data[i].re);
176 status = sdb_store_matcher_matches(m, obj, /* filter */ NULL);
177 fail_unless(status == golden_data[i].expected,
178 "sdb_store_matcher_matches(%s->%s, <host a>, NULL) = %d; "
179 "expected: %d", SDB_STORE_TYPE_TO_NAME(golden_data[i].type),
180 golden_data[i].name, 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->%s, <host a>, NULL) = %d; "
191 "expected: %d", SDB_STORE_TYPE_TO_NAME(golden_data[i].type),
192 golden_data[i].name, 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 int status;
245 m = sdb_store_attr_matcher(golden_data[i].attr_name,
246 golden_data[i].attr_value, golden_data[i].re);
247 fail_unless(m != NULL,
248 "sdb_store_attr_matcher() = NULL; expected: <matcher>");
250 status = sdb_store_matcher_matches(m, obj, /* filter */ NULL);
251 fail_unless(status == golden_data[i].expected,
252 "sdb_store_matcher_matches(attribute[%s] = %s, "
253 "<host a>, NULL) = %d; expected: %d", golden_data[i].attr_name,
254 golden_data[i].attr_value, status, golden_data[i].expected);
256 n = sdb_store_inv_matcher(m);
257 fail_unless(n != NULL,
258 "sdb_store_inv_matcher() = NULL; expected: <matcher>");
259 sdb_object_deref(SDB_OBJ(m));
261 /* now match the inverted set of objects */
262 status = sdb_store_matcher_matches(n, obj, /* filter */ NULL);
263 fail_unless(status == !golden_data[i].expected,
264 "sdb_store_matcher_matches(attribute[%s] = %s, "
265 "<host a>, NULL) = %d; expected: %d",
266 golden_data[i].attr_name, golden_data[i].attr_value,
267 status, !golden_data[i].expected);
269 sdb_object_deref(SDB_OBJ(n));
270 }
272 sdb_object_deref(SDB_OBJ(obj));
273 }
274 END_TEST
276 START_TEST(test_attr_cond)
277 {
278 sdb_store_obj_t *obj;
280 struct {
281 const char *attr;
282 const sdb_data_t value;
283 int expected_lt, expected_le, expected_eq, expected_ge, expected_gt;
284 } golden_data[] = {
285 { "k1", { SDB_TYPE_STRING, { .string = "v1" } }, 0, 1, 1, 1, 0 },
286 { "k1", { SDB_TYPE_STRING, { .string = "v2" } }, 1, 1, 0, 0, 0 },
287 { "k1", { SDB_TYPE_STRING, { .string = "v0" } }, 0, 0, 0, 1, 1 },
288 { "k1", { SDB_TYPE_STRING, { .string = "0" } }, 0, 0, 0, 1, 1 },
289 { "k2", { SDB_TYPE_INTEGER, { .integer = 123 } }, 0, 1, 1, 1, 0 },
290 { "k2", { SDB_TYPE_INTEGER, { .integer = 124 } }, 1, 1, 0, 0, 0 },
291 { "k2", { SDB_TYPE_INTEGER, { .integer = 122 } }, 0, 0, 0, 1, 1 },
292 /* key does not exist */
293 { "k3", { SDB_TYPE_STRING, { .string = "v1" } }, 0, 0, 0, 0, 0 },
294 { "k3", { SDB_TYPE_STRING, { .string = "123" } }, 0, 0, 0, 0, 0 },
295 { "k3", { SDB_TYPE_INTEGER, { .integer = 123 } }, 0, 0, 0, 0, 0 },
296 /* type mismatch */
297 { "k1", { SDB_TYPE_INTEGER, { .integer = 0 } }, 0, 0, 0, 1, 1 },
298 { "k2", { SDB_TYPE_STRING, { .string = "123" } }, 0, 1, 1, 1, 0 },
299 };
301 int status;
302 size_t i;
304 obj = sdb_store_get_host("a");
305 fail_unless(obj != NULL,
306 "sdb_store_get_host(a) = NULL; expected: <host>");
308 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
309 sdb_store_expr_t *expr;
310 sdb_store_cond_t *c;
311 char buf[1024];
312 size_t j;
314 struct {
315 sdb_store_matcher_t *(*matcher)(sdb_store_cond_t *);
316 int expected;
317 } tests[] = {
318 { sdb_store_lt_matcher, golden_data[i].expected_lt },
319 { sdb_store_le_matcher, golden_data[i].expected_le },
320 { sdb_store_eq_matcher, golden_data[i].expected_eq },
321 { sdb_store_ge_matcher, golden_data[i].expected_ge },
322 { sdb_store_gt_matcher, golden_data[i].expected_gt },
323 };
325 sdb_data_format(&golden_data[i].value,
326 buf, sizeof(buf), SDB_UNQUOTED);
328 expr = sdb_store_expr_constvalue(&golden_data[i].value);
329 fail_unless(expr != NULL,
330 "sdb_store_expr_constvalue(%s) = NULL; expected: <expr>",
331 buf);
333 c = sdb_store_attr_cond(golden_data[i].attr, expr);
334 sdb_object_deref(SDB_OBJ(expr));
335 fail_unless(c != NULL,
336 "sdb_store_attr_cond(%s, expr{%s}) = NULL; expected: <cond>",
337 golden_data[i].attr, buf);
339 for (j = 0; j < SDB_STATIC_ARRAY_LEN(tests); ++j) {
340 sdb_store_matcher_t *m;
342 m = tests[j].matcher(c);
343 fail_unless(m != NULL,
344 "sdb_store_<cond>_matcher() = NULL; expected: <matcher>");
346 status = sdb_store_matcher_matches(m, obj, /* filter */ NULL);
347 fail_unless(status == tests[j].expected,
348 "sdb_store_matcher_matches(<m>, <obj>, NULL) = %d; "
349 "expected: %d", status, tests[j].expected);
351 sdb_object_deref(SDB_OBJ(m));
352 }
354 sdb_object_deref(SDB_OBJ(c));
355 }
357 sdb_object_deref(SDB_OBJ(obj));
358 }
359 END_TEST
361 START_TEST(test_cmp_obj)
362 {
363 struct {
364 const char *host;
365 int field;
366 const sdb_data_t value;
367 int expected_lt, expected_le, expected_eq, expected_ge, expected_gt;
368 } golden_data[] = {
369 { "b", SDB_FIELD_NAME,
370 { SDB_TYPE_STRING, { .string = "a" } }, 0, 0, 0, 1, 1 },
371 { "b", SDB_FIELD_NAME,
372 { SDB_TYPE_STRING, { .string = "b" } }, 0, 1, 1, 1, 0 },
373 { "b", SDB_FIELD_NAME,
374 { SDB_TYPE_STRING, { .string = "c" } }, 1, 1, 0, 0, 0 },
375 /* last-update = 1 for all objects */
376 { "a", SDB_FIELD_LAST_UPDATE,
377 { SDB_TYPE_DATETIME, { .datetime = 1 } }, 0, 1, 1, 1, 0 },
378 { "a", SDB_FIELD_LAST_UPDATE,
379 { SDB_TYPE_DATETIME, { .datetime = 2 } }, 1, 1, 0, 0, 0 },
380 { "a", SDB_FIELD_LAST_UPDATE,
381 { SDB_TYPE_DATETIME, { .datetime = 0 } }, 0, 0, 0, 1, 1 },
382 /* age > 0 for all objects */
383 { "a", SDB_FIELD_AGE,
384 { SDB_TYPE_DATETIME, { .datetime = 0 } }, 0, 0, 0, 1, 1 },
385 /* interval = 0 for all objects */
386 { "a", SDB_FIELD_INTERVAL,
387 { SDB_TYPE_DATETIME, { .datetime = 0 } }, 0, 1, 1, 1, 0 },
388 { "a", SDB_FIELD_INTERVAL,
389 { SDB_TYPE_DATETIME, { .datetime = 1 } }, 1, 1, 0, 0, 0 },
390 /* type mismatch */
391 /* TODO: let matchers only use data_strcmp for attributes
392 * everything else has a well-known type
393 { "a", SDB_FIELD_LAST_UPDATE,
394 { SDB_TYPE_INTEGER, { .integer = 0 } }, 0, 0, 0, 0, 0 },
395 { "a", SDB_FIELD_AGE,
396 { SDB_TYPE_INTEGER, { .integer = 0 } }, 0, 0, 0, 0, 0 },
397 { "a", SDB_FIELD_INTERVAL,
398 { SDB_TYPE_INTEGER, { .integer = 0 } }, 0, 0, 0, 0, 0 },
399 { "a", SDB_FIELD_BACKEND,
400 { SDB_TYPE_INTEGER, { .integer = 0 } }, 0, 0, 0, 0, 0 },
401 { "a", SDB_FIELD_BACKEND,
402 { SDB_TYPE_INTEGER, { .integer = 0 } }, 0, 0, 0, 0, 0 },
403 */
404 /* (64bit) integer value without zero-bytes */
405 /*
406 { "a", SDB_FIELD_BACKEND,
407 { SDB_TYPE_INTEGER, { .integer = 0xffffffffffffffffL } },
408 0, 0, 0, 0, 0 },
409 */
410 };
412 int status;
413 size_t i;
415 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
416 sdb_store_obj_t *host;
417 sdb_store_expr_t *field;
418 sdb_store_expr_t *value;
419 char value_str[1024];
420 size_t j;
422 struct {
423 sdb_store_matcher_t *(*matcher)(sdb_store_expr_t *,
424 sdb_store_expr_t *);
425 int expected;
426 } tests[] = {
427 { sdb_store_cmp_lt, golden_data[i].expected_lt },
428 { sdb_store_cmp_le, golden_data[i].expected_le },
429 { sdb_store_cmp_eq, golden_data[i].expected_eq },
430 { sdb_store_cmp_ge, golden_data[i].expected_ge },
431 { sdb_store_cmp_gt, golden_data[i].expected_gt },
432 };
433 char *op_str[] = { "<", "<=", "=", ">=", ">" };
435 assert(SDB_STATIC_ARRAY_LEN(tests) == SDB_STATIC_ARRAY_LEN(op_str));
437 host = sdb_store_get_host(golden_data[i].host);
438 fail_unless(host != NULL,
439 "sdb_store_get_host(%s) = NULL; expected: <host>",
440 golden_data[i].host);
442 sdb_data_format(&golden_data[i].value,
443 value_str, sizeof(value_str), SDB_UNQUOTED);
445 field = sdb_store_expr_fieldvalue(golden_data[i].field);
446 fail_unless(field != NULL,
447 "sdb_store_expr_fieldvalue(%d) = NULL; "
448 "expected: <expr>", golden_data[i].field);
450 value = sdb_store_expr_constvalue(&golden_data[i].value);
451 fail_unless(value != NULL,
452 "sdb_store_expr_constvalue(%s) = NULL; "
453 "expected: <expr>", value_str);
455 for (j = 0; j < SDB_STATIC_ARRAY_LEN(tests); ++j) {
456 char m_str[1024];
457 sdb_store_matcher_t *m;
459 snprintf(m_str, sizeof(m_str), "%s %s %s",
460 SDB_FIELD_TO_NAME(golden_data[i].field),
461 op_str[j], value_str);
463 m = tests[j].matcher(field, value);
464 fail_unless(m != NULL,
465 "sdb_store_<cond>_matcher() = NULL; expected: <matcher>");
467 status = sdb_store_matcher_matches(m, host, /* filter */ NULL);
468 fail_unless(status == tests[j].expected,
469 "sdb_store_matcher_matches(<%s>, <host '%s'>, NULL) = %d; "
470 "expected: %d", m_str, golden_data[i].host, status,
471 tests[j].expected);
473 sdb_object_deref(SDB_OBJ(m));
474 }
476 sdb_object_deref(SDB_OBJ(field));
477 sdb_object_deref(SDB_OBJ(value));
478 sdb_object_deref(SDB_OBJ(host));
479 }
480 }
481 END_TEST
483 START_TEST(test_store_match_op)
484 {
485 sdb_store_obj_t *obj;
487 sdb_store_matcher_t *always = sdb_store_name_matcher(SDB_HOST, "a", 0);
488 sdb_store_matcher_t *never = sdb_store_name_matcher(SDB_HOST, "z", 0);
490 struct {
491 const char *op;
492 sdb_store_matcher_t *left;
493 sdb_store_matcher_t *right;
494 int expected;
495 } golden_data[] = {
496 { "OR", always, always, 1 },
497 { "OR", always, never, 1 },
498 { "OR", never, always, 1 },
499 { "OR", never, never, 0 },
500 { "AND", always, always, 1 },
501 { "AND", always, never, 0 },
502 { "AND", never, always, 0 },
503 { "AND", never, never, 0 },
504 };
506 int status;
507 size_t i;
509 obj = sdb_store_get_host("a");
511 status = sdb_store_matcher_matches(always, obj, /* filter */ NULL);
512 fail_unless(status == 1,
513 "INTERNAL ERROR: 'always' did not match host");
514 status = sdb_store_matcher_matches(never, obj, /* filter */ NULL);
515 fail_unless(status == 0,
516 "INTERNAL ERROR: 'never' matches host");
518 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
519 sdb_store_matcher_t *m;
521 if (! strcmp(golden_data[i].op, "OR"))
522 m = sdb_store_dis_matcher(golden_data[i].left,
523 golden_data[i].right);
524 else if (! strcmp(golden_data[i].op, "AND"))
525 m = sdb_store_con_matcher(golden_data[i].left,
526 golden_data[i].right);
527 else {
528 fail("INTERNAL ERROR: unexpected operator %s", golden_data[i].op);
529 continue;
530 }
532 #define TO_NAME(v) (((v) == always) ? "always" \
533 : ((v) == never) ? "never" : "<unknown>")
535 status = sdb_store_matcher_matches(m, obj, /* filter */ NULL);
536 fail_unless(status == golden_data[i].expected,
537 "%s(%s, %s, NULL) = %d; expected: %d", golden_data[i].op,
538 TO_NAME(golden_data[i].left), TO_NAME(golden_data[i].right),
539 status, golden_data[i].expected);
541 #undef TO_NAME
543 sdb_object_deref(SDB_OBJ(m));
544 }
546 sdb_object_deref(SDB_OBJ(always));
547 sdb_object_deref(SDB_OBJ(never));
549 sdb_object_deref(SDB_OBJ(obj));
550 }
551 END_TEST
553 START_TEST(test_parse_cmp)
554 {
555 sdb_data_t hostname = { SDB_TYPE_STRING, { .string = "hostname" } };
556 sdb_data_t metricname = { SDB_TYPE_STRING, { .string = "metricname" } };
557 sdb_data_t srvname = { SDB_TYPE_STRING, { .string = "srvname" } };
558 sdb_data_t attrname = { SDB_TYPE_STRING, { .string = "attrname" } };
559 sdb_data_t attrvalue = { SDB_TYPE_INTEGER, { .integer = 4711 } };
561 sdb_store_matcher_t *check;
563 size_t i;
565 struct {
566 const char *obj_type;
567 const char *attr;
568 const char *op;
569 const sdb_data_t *value;
570 int expected;
571 } golden_data[] = {
572 { "host", NULL, "=", &hostname, MATCHER_NAME },
573 { "host", NULL, "!=", &hostname, MATCHER_NOT },
574 { "host", NULL, "=~", &hostname, MATCHER_NAME },
575 { "host", NULL, "!~", &hostname, MATCHER_NOT },
576 { "host", "attr", "=", &hostname, -1 },
577 { "host", "attr", "!=", &hostname, -1 },
578 { "host", "attr", "!=", &attrvalue, -1 },
579 { "host", "attr", "<=", &attrvalue, -1 },
580 { "host", NULL, "&^", &hostname, -1 },
581 { "host", NULL, "<", &hostname, -1 },
582 { "host", NULL, "<=", &hostname, -1 },
583 { "host", NULL, ">=", &hostname, -1 },
584 { "host", NULL, ">", &hostname, -1 },
585 { "host", NULL, "=", NULL, -1 },
586 { "metric", NULL, "=", &metricname, MATCHER_NAME },
587 { "metric", NULL, "!=", &metricname, MATCHER_NOT },
588 { "metric", NULL, "=~", &metricname, MATCHER_NAME },
589 { "metric", NULL, "!~", &metricname, MATCHER_NOT },
590 { "metric", "attr", "=", &metricname, -1 },
591 { "metric", "attr", "!=", &metricname, -1 },
592 { "metric", "attr", "!=", &attrvalue, -1 },
593 { "metric", "attr", "<=", &attrvalue, -1 },
594 { "metric", NULL, "&^", &metricname, -1 },
595 { "metric", NULL, "<", &metricname, -1 },
596 { "metric", NULL, "<=", &metricname, -1 },
597 { "metric", NULL, ">=", &metricname, -1 },
598 { "metric", NULL, ">", &metricname, -1 },
599 { "metric", NULL, "=", NULL, -1 },
600 { "service", NULL, "=", &srvname, MATCHER_NAME },
601 { "service", NULL, "!=", &srvname, MATCHER_NOT },
602 { "service", NULL, "=~", &srvname, MATCHER_NAME },
603 { "service", NULL, "!~", &srvname, MATCHER_NOT },
604 { "service", "attr", "=", &srvname, -1 },
605 { "service", "attr", "!=", &srvname, -1 },
606 { "service", "attr", "!=", &attrvalue, -1 },
607 { "service", "attr", "<=", &attrvalue, -1 },
608 { "service", NULL, "&^", &srvname, -1 },
609 { "service", NULL, "<", &srvname, -1 },
610 { "service", NULL, "<=", &srvname, -1 },
611 { "service", NULL, ">=", &srvname, -1 },
612 { "service", NULL, ">", &srvname, -1 },
613 { "service", NULL, "=", NULL, -1 },
614 { "attribute", NULL, "=", &attrname, MATCHER_NAME },
615 { "attribute", NULL, "!=", &attrname, MATCHER_NOT },
616 { "attribute", NULL, "=~", &attrname, MATCHER_NAME },
617 { "attribute", NULL, "!~", &attrname, MATCHER_NOT },
618 { "attribute", NULL, "<", &attrname, -1 },
619 { "attribute", NULL, "<=", &attrname, -1 },
620 { "attribute", NULL, ">=", &attrname, -1 },
621 { "attribute", NULL, ">", &attrname, -1 },
622 { "attribute", NULL, "=", NULL, -1 },
623 { "attribute", "attr", "=", &attrname, MATCHER_ATTR },
624 { "attribute", "attr", "!=", &attrname, MATCHER_NOT },
625 { "attribute", "attr", "=~", &attrname, MATCHER_ATTR },
626 { "attribute", "attr", "!~", &attrname, MATCHER_NOT },
627 { "attribute", "attr", "&^", &attrname, -1 },
628 { "attribute", "attr", "<", NULL, -1 },
629 { "attribute", "attr", "<", &attrname, MATCHER_LT },
630 { "attribute", "attr", "<=", &attrname, MATCHER_LE },
631 /* { "attribute", "attr", "=", &attrname, MATCHER_EQ }, */
632 { "attribute", "attr", ">=", &attrname, MATCHER_GE },
633 { "attribute", "attr", ">", &attrname, MATCHER_GT },
634 { "foo", NULL, "=", &attrname, -1 },
635 { "foo", "attr", "=", &attrname, -1 },
636 };
638 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
639 sdb_store_expr_t *expr;
640 char buf[1024];
642 if (sdb_data_format(golden_data[i].value,
643 buf, sizeof(buf), SDB_UNQUOTED) < 0)
644 snprintf(buf, sizeof(buf), "ERR");
646 expr = sdb_store_expr_constvalue(golden_data[i].value);
647 fail_unless(expr != NULL || golden_data[i].value == NULL,
648 "sdb_store_expr_constvalue(%s) = NULL; expected: <expr>",
649 buf);
651 check = sdb_store_matcher_parse_cmp(golden_data[i].obj_type,
652 golden_data[i].attr, golden_data[i].op, expr);
653 sdb_object_deref(SDB_OBJ(expr));
655 if (golden_data[i].expected == -1) {
656 fail_unless(check == NULL,
657 "sdb_store_matcher_parse_cmp(%s, %s, %s, expr{%s}) = %p; "
658 "expected: NULL", golden_data[i].obj_type,
659 golden_data[i].attr, golden_data[i].op, buf, check);
660 continue;
661 }
663 fail_unless(check != NULL,
664 "sdb_store_matcher_parse_cmp(%s, %s, %s, %s) = %p; "
665 "expected: <expr>", golden_data[i].obj_type,
666 golden_data[i].attr, golden_data[i].op, buf, check);
667 fail_unless(M(check)->type == golden_data[i].expected,
668 "sdb_store_matcher_parse_cmp(%s, %s, %s, %s) returned matcher "
669 "of type %d; expected: %d", golden_data[i].obj_type,
670 golden_data[i].attr, golden_data[i].op, buf,
671 M(check)->type, golden_data[i].expected);
673 sdb_object_deref(SDB_OBJ(check));
674 }
675 }
676 END_TEST
678 START_TEST(test_parse_field_cmp)
679 {
680 sdb_data_t datetime = { SDB_TYPE_DATETIME, { .datetime = 1 } };
681 sdb_data_t string = { SDB_TYPE_STRING, { .string = "s" } };
683 struct {
684 const char *field;
685 const char *op;
686 const sdb_data_t *value;
687 int expected;
688 } golden_data[] = {
689 { "name", "<", &string, MATCHER_LT },
690 { "name", "<=", &string, MATCHER_LE },
691 { "name", "=", &string, MATCHER_EQ },
692 { "name", ">=", &string, MATCHER_GE },
693 { "name", ">", &string, MATCHER_GT },
694 { "name", "!=", &string, MATCHER_NOT },
695 { "last_update", "<", &datetime, MATCHER_LT },
696 { "last_update", "<=", &datetime, MATCHER_LE },
697 { "last_update", "=", &datetime, MATCHER_EQ },
698 { "last_update", ">=", &datetime, MATCHER_GE },
699 { "last_update", ">", &datetime, MATCHER_GT },
700 { "last_update", "!=", &datetime, MATCHER_NOT },
701 { "age", "<", &datetime, MATCHER_LT },
702 { "age", "<=", &datetime, MATCHER_LE },
703 { "age", "=", &datetime, MATCHER_EQ },
704 { "age", ">=", &datetime, MATCHER_GE },
705 { "age", ">", &datetime, MATCHER_GT },
706 { "age", "!=", &datetime, MATCHER_NOT },
707 { "interval", "<", &datetime, MATCHER_LT },
708 { "interval", "<=", &datetime, MATCHER_LE },
709 { "interval", "=", &datetime, MATCHER_EQ },
710 { "interval", ">=", &datetime, MATCHER_GE },
711 { "interval", ">", &datetime, MATCHER_GT },
712 { "interval", "!=", &datetime, MATCHER_NOT },
713 { "backend", "=", &string, MATCHER_EQ },
714 { "backend", "!=", &string, MATCHER_NOT },
715 /* the behavior for other operators on .backend
716 * is currently unspecified */
717 { "last_update", "=", NULL, -1 },
718 { "last_update", "IS", NULL, -1 },
719 { "age", "=", NULL, -1 },
720 { "interval", "=", NULL, -1 },
721 { "backend", "=", NULL, -1 },
722 { "backend", "=~", &string, -1 },
723 };
725 size_t i;
727 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
728 sdb_store_matcher_t *check;
729 sdb_store_expr_t *expr;
730 char buf[1024];
732 if (sdb_data_format(golden_data[i].value,
733 buf, sizeof(buf), SDB_UNQUOTED) < 0)
734 snprintf(buf, sizeof(buf), "ERR");
736 expr = sdb_store_expr_constvalue(golden_data[i].value);
737 fail_unless(expr != NULL || golden_data[i].value == NULL,
738 "sdb_store_expr_constvalue(%s) = NULL; expected: <expr>",
739 buf);
741 check = sdb_store_matcher_parse_field_cmp(golden_data[i].field,
742 golden_data[i].op, expr);
743 sdb_object_deref(SDB_OBJ(expr));
745 if (golden_data[i].expected == -1) {
746 fail_unless(check == NULL,
747 "sdb_store_matcher_parse_field_cmp(%s, %s, expr{%s}) = %p; "
748 "expected: NULL", golden_data[i].field,
749 golden_data[i].op, buf, check);
750 continue;
751 }
753 fail_unless(check != NULL,
754 "sdb_store_matcher_parse_field_cmp(%s, %s, %s) = %p; "
755 "expected: NULL", golden_data[i].field,
756 golden_data[i].op, buf, check);
757 fail_unless(M(check)->type == golden_data[i].expected,
758 "sdb_store_matcher_parse_field_cmp(%s, %s, %s) returned "
759 "matcher of type %d; expected: %d", golden_data[i].field,
760 golden_data[i].op, buf, M(check)->type,
761 golden_data[i].expected);
763 sdb_object_deref(SDB_OBJ(check));
764 }
765 }
766 END_TEST
768 static int
769 scan_cb(sdb_store_obj_t *obj, void *user_data)
770 {
771 int *i = user_data;
773 fail_unless(obj != NULL,
774 "sdb_store_scan callback received NULL obj; expected: "
775 "<store base obj>");
776 fail_unless(i != NULL,
777 "sdb_store_scan callback received NULL user_data; "
778 "expected: <pointer to data>");
780 ++(*i);
781 return 0;
782 } /* scan_cb */
784 START_TEST(test_scan)
785 {
786 struct {
787 const char *query;
788 const char *filter;
789 int expected;
790 } golden_data[] = {
791 { "host = 'a'", NULL, 1 },
792 { "host = 'a'", "host = 'x'", 0 }, /* filter never matches */
793 { "host = 'a'",
794 "NOT attribute['x'] = ''", 1 }, /* filter always matches */
795 { "host =~ 'a|b'", NULL, 2 },
796 { "host =~ 'host'", NULL, 0 },
797 { "host =~ '.'", NULL, 3 },
798 { "metric = 'm1'", NULL, 2 },
799 { "metric= 'm1'", "host = 'x'", 0 }, /* filter never matches */
800 { "metric = 'm1'",
801 "NOT attribute['x'] = ''", 2 }, /* filter always matches */
802 { "metric =~ 'm'", NULL, 2 },
803 { "metric !~ 'm'", NULL, 1 },
804 { "metric =~ 'x'", NULL, 0 },
805 { "service = 's1'", NULL, 2 },
806 { "service = 's1'", "host = 'x'", 0 }, /* filter never matches */
807 { "service = 's1'",
808 "NOT attribute['x'] = ''", 2 }, /* filter always matches */
809 { "service =~ 's'", NULL, 2 },
810 { "service !~ 's'", NULL, 1 },
811 { "attribute = 'k1'", NULL, 2 },
812 { "attribute = 'k1'", "host = 'x'", 0 }, /* filter never matches */
813 { "attribute = 'k1'",
814 "NOT attribute['x'] = ''", 2 }, /* filter always matches */
815 { "attribute = 'x'", NULL, 0 },
816 { "attribute['k1'] = 'v1'", NULL, 1 },
817 { "attribute['k1'] =~ 'v'", NULL, 2 },
818 { "attribute['k1'] !~ 'v'", NULL, 1 },
819 { "attribute['x1'] =~ 'v'", NULL, 0 },
820 { "attribute['x1'] =~ 'NULL'", NULL, 0 },
821 { "attribute['x1'] !~ 'v'", NULL, 3 },
822 { "attribute['k1'] IS NULL", NULL, 1 },
823 { "attribute['x1'] IS NULL", NULL, 3 },
824 { "attribute['k1'] IS NOT NULL", NULL, 2 },
825 { "attribute['x1'] IS NOT NULL", NULL, 0 },
826 { "attribute['k2'] < 123", NULL, 0 },
827 { "attribute['k2'] <= 123", NULL, 1 },
828 { "attribute['k2'] >= 123", NULL, 1 },
829 { "attribute['k2'] > 123", NULL, 0 },
830 { "attribute['k2'] = 123", NULL, 1 },
831 { "attribute['k2'] != 123", NULL, 0 },
832 { "attribute['k1'] != 'v1'", NULL, 1 },
833 { "attribute['k1'] != 'v2'", NULL, 1 },
834 { "attribute != 'x' "
835 "AND attribute['y'] !~ 'x'", NULL, 3 },
836 };
838 int check, n;
839 size_t i;
841 n = 0;
842 check = sdb_store_scan(/* matcher */ NULL, /* filter */ NULL,
843 scan_cb, &n);
844 fail_unless(check == 0,
845 "sdb_store_scan() = %d; expected: 0", check);
846 fail_unless(n == 3,
847 "sdb_store_scan called callback %d times; expected: 3", (int)n);
849 for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
850 sdb_store_matcher_t *m, *filter = NULL;
852 m = sdb_fe_parse_matcher(golden_data[i].query, -1);
853 fail_unless(m != NULL,
854 "sdb_fe_parse_matcher(%s, -1) = NULL; expected: <matcher>",
855 golden_data[i].query);
857 if (golden_data[i].filter) {
858 filter = sdb_fe_parse_matcher(golden_data[i].filter, -1);
859 fail_unless(filter != NULL,
860 "sdb_fe_parse_matcher(%s, -1) = NULL; "
861 "expected: <matcher>", golden_data[i].filter);
862 }
864 n = 0;
865 sdb_store_scan(m, filter, scan_cb, &n);
866 fail_unless(n == golden_data[i].expected,
867 "sdb_store_scan(matcher{%s}, filter{%s}) found %d hosts; "
868 "expected: %d", golden_data[i].query, golden_data[i].filter,
869 n, golden_data[i].expected);
870 sdb_object_deref(SDB_OBJ(filter));
871 sdb_object_deref(SDB_OBJ(m));
872 }
873 }
874 END_TEST
876 Suite *
877 core_store_lookup_suite(void)
878 {
879 Suite *s = suite_create("core::store_lookup");
880 TCase *tc;
882 tc = tcase_create("core");
883 tcase_add_checked_fixture(tc, populate, sdb_store_clear);
884 tcase_add_test(tc, test_store_match_name);
885 tcase_add_test(tc, test_store_match_attr);
886 tcase_add_test(tc, test_attr_cond);
887 tcase_add_test(tc, test_cmp_obj);
888 tcase_add_test(tc, test_store_match_op);
889 tcase_add_test(tc, test_parse_cmp);
890 tcase_add_test(tc, test_parse_field_cmp);
891 tcase_add_test(tc, test_scan);
892 suite_add_tcase(s, tc);
894 return s;
895 } /* core_store_lookup_suite */
897 /* vim: set tw=78 sw=4 ts=4 noexpandtab : */