Code

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