X-Git-Url: https://git.tokkee.org/?p=sysdb.git;a=blobdiff_plain;f=t%2Fcore%2Fstore_test.c;h=29c5b1b67d7a8eee65639078e65fc7800f6a0bcf;hp=c27310d4e6720dd5819483e00af7668f3e498287;hb=3ba81224dac4d31ea5ad651db569543825ea5078;hpb=ebc8f66f721bbe6a23e32f6712003b34f588b508 diff --git a/t/core/store_test.c b/t/core/store_test.c index c27310d..29c5b1b 100644 --- a/t/core/store_test.c +++ b/t/core/store_test.c @@ -26,11 +26,32 @@ */ #include "core/store.h" +#include "core/store-private.h" #include "libsysdb_test.h" #include #include +static void +populate(void) +{ + sdb_data_t datum; + + sdb_store_host("h1", 1); + sdb_store_host("h2", 1); + + datum.type = SDB_TYPE_STRING; + datum.data.string = "v1"; + sdb_store_attribute("h1", "k1", &datum, 1); + datum.data.string = "v2"; + sdb_store_attribute("h1", "k2", &datum, 1); + datum.data.string = "v3"; + sdb_store_attribute("h1", "k3", &datum, 1); + + sdb_store_service("h2", "s1", 1); + sdb_store_service("h2", "s2", 1); +} /* populate */ + START_TEST(test_store_host) { struct { @@ -81,12 +102,73 @@ START_TEST(test_store_host) } END_TEST +START_TEST(test_store_get_host) +{ + char *golden_hosts[] = { "a", "b", "c" }; + char *unknown_hosts[] = { "x", "y", "z" }; + size_t i; + + for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_hosts); ++i) { + int status = sdb_store_host(golden_hosts[i], 1); + fail_unless(status >= 0, + "sdb_store_host(%s) = %d; expected: >=0", + golden_hosts[i], status); + } + + for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_hosts); ++i) { + sdb_store_base_t *sobj1, *sobj2; + int ref_cnt; + + fail_unless(sdb_store_has_host(golden_hosts[i]), + "sdb_store_has_host(%s) = FALSE; expected: TRUE", + golden_hosts[i]); + + sobj1 = sdb_store_get_host(golden_hosts[i]); + fail_unless(sobj1 != NULL, + "sdb_store_get_host(%s) = NULL; expected: ", + golden_hosts[i]); + ref_cnt = SDB_OBJ(sobj1)->ref_cnt; + + fail_unless(ref_cnt > 1, + "sdb_store_get_host(%s) did not increment ref count: " + "got: %d; expected: >1", golden_hosts[i], ref_cnt); + + sobj2 = sdb_store_get_host(golden_hosts[i]); + fail_unless(sobj2 != NULL, + "sdb_store_get_host(%s) = NULL; expected: ", + golden_hosts[i]); + + fail_unless(sobj1 == sobj2, + "sdb_store_get_host(%s) returned different objects " + "in successive calls", golden_hosts[i]); + fail_unless(SDB_OBJ(sobj2)->ref_cnt == ref_cnt + 1, + "sdb_store_get_hosts(%s) did not increment ref count " + "(first call: %d; second call: %d)", + golden_hosts[i], ref_cnt, SDB_OBJ(sobj2)->ref_cnt); + + sdb_object_deref(SDB_OBJ(sobj1)); + sdb_object_deref(SDB_OBJ(sobj2)); + } + for (i = 0; i < SDB_STATIC_ARRAY_LEN(unknown_hosts); ++i) { + sdb_store_base_t *sobj; + + fail_unless(!sdb_store_has_host(unknown_hosts[i]), + "sdb_store_has_host(%s) = TRUE; expected: FALSE", + unknown_hosts[i]); + + sobj = sdb_store_get_host(unknown_hosts[i]); + fail_unless(!sobj, "sdb_store_get_host(%s) = ; expected: NULL", + unknown_hosts[i], sobj ? SDB_OBJ(sobj)->name : "NULL"); + } +} +END_TEST + START_TEST(test_store_attr) { struct { const char *host; const char *key; - const char *value; + char *value; sdb_time_t last_update; int expected; } golden_data[] = { @@ -105,10 +187,15 @@ START_TEST(test_store_attr) sdb_store_host("l", 1); sdb_store_host("m", 1); for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) { + sdb_data_t datum; int status; + /* XXX: test other types as well */ + datum.type = SDB_TYPE_STRING; + datum.data.string = golden_data[i].value; + status = sdb_store_attribute(golden_data[i].host, - golden_data[i].key, golden_data[i].value, + golden_data[i].key, &datum, golden_data[i].last_update); fail_unless(status == golden_data[i].expected, "sdb_store_attribute(%s, %s, %s, %d) = %d; expected: %d", @@ -146,52 +233,20 @@ START_TEST(test_store_service) status = sdb_store_service(golden_data[i].host, golden_data[i].svc, golden_data[i].last_update); fail_unless(status == golden_data[i].expected, - "sdb_store_attribute(%s, %s, %d) = %d; expected: %d", + "sdb_store_service(%s, %s, %d) = %d; expected: %d", golden_data[i].host, golden_data[i].svc, golden_data[i].last_update, status, golden_data[i].expected); } } END_TEST -START_TEST(test_store_tojson) +static void +verify_json_output(sdb_strbuf_t *buf, const char *expected, int flags) { - sdb_strbuf_t *buf; - - int status, pos; + int pos; size_t len1, len2; size_t i; - const char *expected = "{\"hosts\":[" - "{\"name\": \"h1\", \"last_update\": \"1970-01-01 00:00:00 +0000\", " - "\"attributes\": [" - "{\"name\": \"k1\", \"value\": \"v1\", \"last_update\": \"1970-01-01 00:00:00 +0000\"}," - "{\"name\": \"k2\", \"value\": \"v2\", \"last_update\": \"1970-01-01 00:00:00 +0000\"}," - "{\"name\": \"k3\", \"value\": \"v3\", \"last_update\": \"1970-01-01 00:00:00 +0000\"}," - "], " - "\"services\": []}," - "{\"name\": \"h2\", \"last_update\": \"1970-01-01 00:00:00 +0000\", " - "\"attributes\": [], " - "\"services\": [" - "{\"name\": \"s1\", \"last_update\": \"1970-01-01 00:00:00 +0000\"}," - "{\"name\": \"s2\", \"last_update\": \"1970-01-01 00:00:00 +0000\"}," - "]}" - "]}"; - - sdb_store_host("h1", 1); - sdb_store_host("h2", 1); - - sdb_store_attribute("h1", "k1", "v1", 1); - sdb_store_attribute("h1", "k2", "v2", 1); - sdb_store_attribute("h1", "k3", "v3", 1); - - sdb_store_service("h2", "s1", 1); - sdb_store_service("h2", "s2", 1); - - buf = sdb_strbuf_create(0); - status = sdb_store_tojson(buf); - fail_unless(status == 0, - "sdb_store_tojson() = %d; expected: 0", status); - len1 = strlen(sdb_strbuf_string(buf)); len2 = strlen(expected); @@ -207,9 +262,226 @@ START_TEST(test_store_tojson) } fail_unless(pos == -1, - "sdb_store_tojson() returned unexpected result\n" + "sdb_store_tojson(%x) returned unexpected result\n" " got: %s\n %*s\n expected: %s", - sdb_strbuf_string(buf), pos + 1, "^", expected); + flags, sdb_strbuf_string(buf), pos + 1, "^", expected); +} /* verify_json_output */ + +START_TEST(test_store_tojson) +{ + sdb_strbuf_t *buf; + size_t i; + + struct { + int flags; + const char *expected; + } golden_data[] = { + { 0, "{\"hosts\":[" + "{\"name\": \"h1\", \"last_update\": \"1970-01-01 00:00:00 +0000\", " + "\"update_interval\": \"0s\", " + "\"attributes\": [" + "{\"name\": \"k1\", \"value\": \"v1\", " + "\"last_update\": \"1970-01-01 00:00:00 +0000\", " + "\"update_interval\": \"0s\"}," + "{\"name\": \"k2\", \"value\": \"v2\", " + "\"last_update\": \"1970-01-01 00:00:00 +0000\", " + "\"update_interval\": \"0s\"}," + "{\"name\": \"k3\", \"value\": \"v3\", " + "\"last_update\": \"1970-01-01 00:00:00 +0000\", " + "\"update_interval\": \"0s\"}" + "], " + "\"services\": []}," + "{\"name\": \"h2\", \"last_update\": \"1970-01-01 00:00:00 +0000\", " + "\"update_interval\": \"0s\", " + "\"attributes\": [], " + "\"services\": [" + "{\"name\": \"s1\", " + "\"last_update\": \"1970-01-01 00:00:00 +0000\", " + "\"update_interval\": \"0s\"}," + "{\"name\": \"s2\", " + "\"last_update\": \"1970-01-01 00:00:00 +0000\", " + "\"update_interval\": \"0s\"}" + "]}" + "]}" }, + { SDB_SKIP_SERVICES, + "{\"hosts\":[" + "{\"name\": \"h1\", \"last_update\": \"1970-01-01 00:00:00 +0000\", " + "\"update_interval\": \"0s\", " + "\"attributes\": [" + "{\"name\": \"k1\", \"value\": \"v1\", " + "\"last_update\": \"1970-01-01 00:00:00 +0000\", " + "\"update_interval\": \"0s\"}," + "{\"name\": \"k2\", \"value\": \"v2\", " + "\"last_update\": \"1970-01-01 00:00:00 +0000\", " + "\"update_interval\": \"0s\"}," + "{\"name\": \"k3\", \"value\": \"v3\", " + "\"last_update\": \"1970-01-01 00:00:00 +0000\", " + "\"update_interval\": \"0s\"}" + "]}," + "{\"name\": \"h2\", \"last_update\": \"1970-01-01 00:00:00 +0000\", " + "\"update_interval\": \"0s\", " + "\"attributes\": []}" + "]}" }, + { SDB_SKIP_ATTRIBUTES, + "{\"hosts\":[" + "{\"name\": \"h1\", \"last_update\": \"1970-01-01 00:00:00 +0000\", " + "\"update_interval\": \"0s\", " + "\"services\": []}," + "{\"name\": \"h2\", \"last_update\": \"1970-01-01 00:00:00 +0000\", " + "\"update_interval\": \"0s\", " + "\"services\": [" + "{\"name\": \"s1\", " + "\"last_update\": \"1970-01-01 00:00:00 +0000\", " + "\"update_interval\": \"0s\"}," + "{\"name\": \"s2\", " + "\"last_update\": \"1970-01-01 00:00:00 +0000\", " + "\"update_interval\": \"0s\"}" + "]}" + "]}" }, + { SDB_SKIP_SERVICES | SDB_SKIP_ATTRIBUTES, + "{\"hosts\":[" + "{\"name\": \"h1\", \"last_update\": \"1970-01-01 00:00:00 +0000\", " + "\"update_interval\": \"0s\"}," + "{\"name\": \"h2\", \"last_update\": \"1970-01-01 00:00:00 +0000\", " + "\"update_interval\": \"0s\"}" + "]}" }, + }; + + buf = sdb_strbuf_create(0); + fail_unless(buf != NULL, "INTERNAL ERROR: failed to create string buffer"); + populate(); + + for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) { + int status; + + sdb_strbuf_clear(buf); + + status = sdb_store_tojson(buf, golden_data[i].flags); + fail_unless(status == 0, + "sdb_store_tojson(%x) = %d; expected: 0", + golden_data[i].flags, status); + + verify_json_output(buf, golden_data[i].expected, golden_data[i].flags); + } + sdb_strbuf_destroy(buf); +} +END_TEST + +START_TEST(test_interval) +{ + sdb_store_base_t *host; + + /* 10 us interval */ + sdb_store_host("host", 10); + sdb_store_host("host", 20); + sdb_store_host("host", 30); + sdb_store_host("host", 40); + + host = sdb_store_get_host("host"); + fail_unless(host != NULL, + "INTERNAL ERROR: store doesn't have host after adding it"); + + fail_unless(host->interval == 10, + "sdb_store_host() did not calculate interval correctly: " + "got: %"PRIscTIME"; expected: %"PRIscTIME, host->interval, 10); + + /* multiple updates for the same timestamp don't modify the interval */ + sdb_store_host("host", 40); + sdb_store_host("host", 40); + sdb_store_host("host", 40); + sdb_store_host("host", 40); + + fail_unless(host->interval == 10, + "sdb_store_host() changed interval when doing multiple updates " + "using the same timestamp; got: %"PRIscTIME"; " + "expected: %"PRIscTIME, host->interval, 10); + + /* multiple updates using an timestamp don't modify the interval */ + sdb_store_host("host", 20); + sdb_store_host("host", 20); + sdb_store_host("host", 20); + sdb_store_host("host", 20); + + fail_unless(host->interval == 10, + "sdb_store_host() changed interval when doing multiple updates " + "using an old timestamp; got: %"PRIscTIME"; expected: %"PRIscTIME, + host->interval, 10); + + /* new interval: 20 us */ + sdb_store_host("host", 60); + fail_unless(host->interval == 11, + "sdb_store_host() did not calculate interval correctly: " + "got: %"PRIscTIME"; expected: %"PRIscTIME, host->interval, 11); + + /* new interval: 40 us */ + sdb_store_host("host", 100); + fail_unless(host->interval == 13, + "sdb_store_host() did not calculate interval correctly: " + "got: %"PRIscTIME"; expected: %"PRIscTIME, host->interval, 11); + + sdb_object_deref(SDB_OBJ(host)); +} +END_TEST + +static int +iter_incr(sdb_store_base_t *obj, void *user_data) +{ + intptr_t *i = user_data; + + fail_unless(obj != NULL, + "sdb_store_iterate callback received NULL obj; expected: " + ""); + fail_unless(i != NULL, + "sdb_store_iterate callback received NULL user_data; " + "expected: "); + + ++(*i); + return 0; +} /* iter_incr */ + +static int +iter_error(sdb_store_base_t *obj, void *user_data) +{ + intptr_t *i = user_data; + + fail_unless(obj != NULL, + "sdb_store_iterate callback received NULL obj; expected: " + ""); + fail_unless(i != NULL, + "sdb_store_iterate callback received NULL user_data; " + "expected: "); + + ++(*i); + return -1; +} /* iter_error */ + +START_TEST(test_iterate) +{ + intptr_t i = 0; + int check; + + /* empty store */ + check = sdb_store_iterate(iter_incr, &i); + fail_unless(check == -1, + "sdb_store_iterate(), empty store = %d; expected: -1", check); + fail_unless(i == 0, + "sdb_store_iterate called callback %d times; expected: 0", (int)i); + + populate(); + + check = sdb_store_iterate(iter_incr, &i); + fail_unless(check == 0, + "sdb_store_iterate() = %d; expected: 0", check); + fail_unless(i == 2, + "sdb_store_iterate called callback %d times; expected: 1", (int)i); + + i = 0; + check = sdb_store_iterate(iter_error, &i); + fail_unless(check == -1, + "sdb_store_iterate(), error callback = %d; expected: -1", check); + fail_unless(i == 1, + "sdb_store_iterate called callback %d times " + "(callback returned error); expected: 1", (int)i); } END_TEST @@ -220,12 +492,14 @@ core_store_suite(void) TCase *tc; tc = tcase_create("core"); - /* test this first to ensure the store is empty - * even when using CK_NOFORK */ tcase_add_test(tc, test_store_tojson); tcase_add_test(tc, test_store_host); + tcase_add_test(tc, test_store_get_host); tcase_add_test(tc, test_store_attr); tcase_add_test(tc, test_store_service); + tcase_add_test(tc, test_interval); + tcase_add_test(tc, test_iterate); + tcase_add_unchecked_fixture(tc, NULL, sdb_store_clear); suite_add_tcase(s, tc); return s;