store_test: Test various serialization flags.

diff --git a/t/core/store_test.c b/t/core/store_test.c
index 9a5b7e563200dfeae84872d29278bf7065ad2376..4a939b64f430ef8dd43d479fdc57ed0789188805 100644 (file)
--- a/t/core/store_test.c
+++ b/t/core/store_test.c
@@ -153,29 +153,84 @@ START_TEST(test_store_service)
 }
 END_TEST
 
+static void
+verify_json_output(sdb_strbuf_t *buf, const char *expected, int flags)
+{
+       int pos;
+       size_t len1, len2;
+       size_t i;
+
+       len1 = strlen(sdb_strbuf_string(buf));
+       len2 = strlen(expected);
+
+       pos = -1;
+       if (len1 != len2)
+               pos = (int)(len1 <= len2 ? len1 : len2);
+
+       for (i = 0; i < (len1 <= len2 ? len1 : len2); ++i) {
+               if (sdb_strbuf_string(buf)[i] != expected[i]) {
+                       pos = (int)i;
+                       break;
+               }
+       }
+
+       fail_unless(pos == -1,
+                       "sdb_store_tojson(%x) returned unexpected result\n"
+                       "         got: %s\n              %*s\n    expected: %s",
+                       flags, sdb_strbuf_string(buf), pos + 1, "^", expected);
+} /* verify_json_output */
+
 START_TEST(test_store_tojson)
 {
        sdb_strbuf_t *buf;
-
-       int status, 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\"}"
-                       "]}"
-       "]}";
+       struct {
+               int flags;
+               const char *expected;
+       } golden_data[] = {
+               { 0, "{\"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_SKIP_SERVICES,
+                       "{\"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\"}"
+                                       "]},"
+                               "{\"name\": \"h2\", \"last_update\": \"1970-01-01 00:00:00 +0000\", "
+                                       "\"attributes\": []}"
+                       "]}" },
+               { SDB_SKIP_ATTRIBUTES,
+                       "{\"hosts\":["
+                               "{\"name\": \"h1\", \"last_update\": \"1970-01-01 00:00:00 +0000\", "
+                                       "\"services\": []},"
+                               "{\"name\": \"h2\", \"last_update\": \"1970-01-01 00:00:00 +0000\", "
+                                       "\"services\": ["
+                                               "{\"name\": \"s1\", \"last_update\": \"1970-01-01 00:00:00 +0000\"},"
+                                               "{\"name\": \"s2\", \"last_update\": \"1970-01-01 00:00:00 +0000\"}"
+                                       "]}"
+                       "]}" },
+               { SDB_SKIP_SERVICES | SDB_SKIP_ATTRIBUTES,
+                       "{\"hosts\":["
+                               "{\"name\": \"h1\", \"last_update\": \"1970-01-01 00:00:00 +0000\"},"
+                               "{\"name\": \"h2\", \"last_update\": \"1970-01-01 00:00:00 +0000\"}"
+                       "]}" },
+       };
 
        sdb_store_host("h1", 1);
        sdb_store_host("h2", 1);
@@ -188,28 +243,20 @@ START_TEST(test_store_tojson)
        sdb_store_service("h2", "s2", 1);
 
        buf = sdb_strbuf_create(0);
-       status = sdb_store_tojson(buf, /* flags = */ 0);
-       fail_unless(status == 0,
-                       "sdb_store_tojson() = %d; expected: 0", status);
 
-       len1 = strlen(sdb_strbuf_string(buf));
-       len2 = strlen(expected);
+       for (i = 0; i < SDB_STATIC_ARRAY_LEN(golden_data); ++i) {
+               int status;
 
-       pos = -1;
-       if (len1 != len2)
-               pos = (int)(len1 <= len2 ? len1 : len2);
+               sdb_strbuf_clear(buf);
 
-       for (i = 0; i < (len1 <= len2 ? len1 : len2); ++i) {
-               if (sdb_strbuf_string(buf)[i] != expected[i]) {
-                       pos = (int)i;
-                       break;
-               }
-       }
+               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);
 
-       fail_unless(pos == -1,
-                       "sdb_store_tojson() returned unexpected result\n"
-                       "         got: %s\n              %*s\n    expected: %s",
-                       sdb_strbuf_string(buf), pos + 1, "^", expected);
+               verify_json_output(buf, golden_data[i].expected, golden_data[i].flags);
+       }
+       sdb_strbuf_destroy(buf);
 }
 END_TEST