From: Sebastian Harl <sh@tokkee.org>
Date: Fri, 3 Jan 2014 13:36:54 +0000 (+0100)
Subject: store_test: Test various serialization flags.
X-Git-Tag: sysdb-0.1.0~254
X-Git-Url: https://git.tokkee.org/?a=commitdiff_plain;h=249fdc86f92076078418834335e22f684fafc9d9;p=sysdb.git

store_test: Test various serialization flags.
---

diff --git a/t/core/store_test.c b/t/core/store_test.c
index 9a5b7e5..4a939b6 100644
--- 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