1 files changed, 263 insertions, 0 deletions
diff --git a/fluent-bit/lib/avro/tests/test_data_structures.c b/fluent-bit/lib/avro/tests/test_data_structures.c
new file mode 100644
index 00000000..2986f5cc
--- /dev/null
+++ b/fluent-bit/lib/avro/tests/test_data_structures.c
@@ -0,0 +1,263 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements.  See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to you under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License.  You may obtain a copy of the License at
+ *
+ * https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+ * implied.  See the License for the specific language governing
+ * permissions and limitations under the License.
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include "avro_private.h"
+#include "avro/data.h"
+
+static int  result = EXIT_SUCCESS;
+
+typedef int (*avro_test) (void);
+
+
+static int
+test_array(void)
+{
+	avro_raw_array_t  array;
+	long  *element;
+
+	/* Test once on a fresh array */
+
+	avro_raw_array_init(&array, sizeof(long));
+	element = (long *) avro_raw_array_append(&array);
+	*element = 1;
+	element = (long *) avro_raw_array_append(&array);
+	*element = 3;
+
+	if (avro_raw_array_size(&array) != 2) {
+		fprintf(stderr, "Incorrect array size: got %lu, expected %lu.\n",
+			(unsigned long) avro_raw_array_size(&array),
+			(unsigned long) 2);
+		return EXIT_FAILURE;
+	}
+
+	if (avro_raw_array_get(&array, long, 0) != 1) {
+		fprintf(stderr, "Unexpected array element %u: got %ld, expected %ld.\n",
+			(unsigned int) 0, avro_raw_array_get(&array, long, 0),
+			(long) 1);
+		return EXIT_FAILURE;
+	}
+
+	/* And test again after clearing the array */
+
+	avro_raw_array_clear(&array);
+	element = (long *) avro_raw_array_append(&array);
+	*element = 1;
+	element = (long *) avro_raw_array_append(&array);
+	*element = 3;
+
+	if (avro_raw_array_size(&array) != 2) {
+		fprintf(stderr, "Incorrect array size: got %" PRIsz ", expected %" PRIsz ".\n",
+			(size_t) avro_raw_array_size(&array),
+			(size_t) 2);
+		return EXIT_FAILURE;
+	}
+
+	if (avro_raw_array_get(&array, long, 0) != 1) {
+		fprintf(stderr, "Unexpected array element %u: got %ld, expected %ld.\n",
+			(unsigned int) 0, avro_raw_array_get(&array, long, 0),
+			(long) 1);
+		return EXIT_FAILURE;
+	}
+
+	avro_raw_array_done(&array);
+	return EXIT_SUCCESS;
+}
+
+
+static int
+test_map(void)
+{
+	avro_raw_map_t  map;
+	long  *element;
+	size_t  index;
+
+	/* Test once on a fresh map */
+
+	avro_raw_map_init(&map, sizeof(long));
+	avro_raw_map_get_or_create(&map, "x", (void **) &element, NULL);
+	*element = 1;
+	avro_raw_map_get_or_create(&map, "y", (void **) &element, NULL);
+	*element = 3;
+
+	if (avro_raw_map_size(&map) != 2) {
+		fprintf(stderr, "Incorrect map size: got %" PRIsz ", expected %" PRIsz ".\n",
+			(size_t) avro_raw_map_size(&map),
+			(size_t) 2);
+		return EXIT_FAILURE;
+	}
+
+	if (avro_raw_map_get_by_index(&map, long, 0) != 1) {
+		fprintf(stderr, "Unexpected map element %u: got %ld, expected %ld.\n",
+			(unsigned int) 0,
+			avro_raw_map_get_by_index(&map, long, 0),
+			(long) 1);
+		return EXIT_FAILURE;
+	}
+
+	if (strcmp(avro_raw_map_get_key(&map, 0), "x") != 0) {
+		fprintf(stderr, "Unexpected key for map element 0: "
+			"got \"%s\", expected \"%s\".\n",
+			avro_raw_map_get_key(&map, 0), "x");
+		return EXIT_FAILURE;
+	}
+
+	element = (long *) avro_raw_map_get(&map, "y", &index);
+	if (index != 1) {
+		fprintf(stderr, "Unexpected index for map element \"%s\": "
+			"got %" PRIsz ", expected %u.\n",
+			"y", index, 1);
+		return EXIT_FAILURE;
+	}
+
+	if (*element != 3) {
+		fprintf(stderr, "Unexpected map element %s: got %ld, expected %ld.\n",
+			"y",
+			*element, (long) 3);
+		return EXIT_FAILURE;
+	}
+
+	/* And test again after clearing the map */
+
+	avro_raw_map_clear(&map);
+	avro_raw_map_get_or_create(&map, "x", (void **) &element, NULL);
+	*element = 1;
+	avro_raw_map_get_or_create(&map, "y", (void **) &element, NULL);
+	*element = 3;
+
+	if (avro_raw_map_size(&map) != 2) {
+		fprintf(stderr, "Incorrect map size: got %" PRIsz ", expected %" PRIsz ".\n",
+			(size_t) avro_raw_map_size(&map),
+			(size_t) 2);
+		return EXIT_FAILURE;
+	}
+
+	if (avro_raw_map_get_by_index(&map, long, 0) != 1) {
+		fprintf(stderr, "Unexpected map element %u: got %ld, expected %ld.\n",
+			(unsigned int) 0,
+			avro_raw_map_get_by_index(&map, long, 0),
+			(long) 1);
+		return EXIT_FAILURE;
+	}
+
+	element = (long *) avro_raw_map_get(&map, "y", &index);
+	if (index != 1) {
+		fprintf(stderr, "Unexpected index for map element \"%s\": "
+			"got %" PRIsz ", expected %u.\n",
+			"y", index, 1);
+		return EXIT_FAILURE;
+	}
+
+	if (*element != 3) {
+		fprintf(stderr, "Unexpected map element %s: got %ld, expected %ld.\n",
+			"y",
+			*element, (long) 3);
+		return EXIT_FAILURE;
+	}
+
+	avro_raw_map_done(&map);
+	return EXIT_SUCCESS;
+}
+
+
+static int
+test_string(void)
+{
+	avro_raw_string_t  str;
+
+	avro_raw_string_init(&str);
+
+	avro_raw_string_set(&str, "a");
+	avro_raw_string_set(&str, "abcdefgh");
+	avro_raw_string_set(&str, "abcd");
+
+	if (avro_raw_string_length(&str) != 5) {
+		fprintf(stderr, "Incorrect string size: got %" PRIsz ", expected %" PRIsz ".\n",
+			(size_t) avro_raw_string_length(&str),
+			(size_t) 5);
+		return EXIT_FAILURE;
+	}
+
+	if (strcmp((const char *) str.wrapped.buf, "abcd") != 0) {
+		fprintf(stderr, "Incorrect string contents: "
+				"got \"%s\", expected \"%s\".\n",
+			(char *) avro_raw_string_get(&str),
+			"abcd");
+		return EXIT_FAILURE;
+	}
+
+	avro_wrapped_buffer_t  wbuf;
+	avro_wrapped_buffer_new_string(&wbuf, "abcd");
+	avro_raw_string_give(&str, &wbuf);
+
+	if (avro_raw_string_length(&str) != 5) {
+		fprintf(stderr, "Incorrect string size: got %" PRIsz ", expected %" PRIsz ".\n",
+			(size_t) avro_raw_string_length(&str),
+			(size_t) 5);
+		return EXIT_FAILURE;
+	}
+
+	if (strcmp((const char *) str.wrapped.buf, "abcd") != 0) {
+		fprintf(stderr, "Incorrect string contents: "
+				"got \"%s\", expected \"%s\".\n",
+			(char *) avro_raw_string_get(&str),
+			"abcd");
+		return EXIT_FAILURE;
+	}
+
+	avro_raw_string_t  str2;
+	avro_raw_string_init(&str2);
+	avro_raw_string_set(&str2, "abcd");
+
+	if (!avro_raw_string_equals(&str, &str2)) {
+		fprintf(stderr, "Strings should be equal.\n");
+		return EXIT_FAILURE;
+	}
+
+	avro_raw_string_done(&str);
+	avro_raw_string_done(&str2);
+	return EXIT_SUCCESS;
+}
+
+
+int main(int argc, char *argv[])
+{
+	AVRO_UNUSED(argc);
+	AVRO_UNUSED(argv);
+
+	unsigned int i;
+	struct avro_tests {
+		char *name;
+		avro_test func;
+	} tests[] = {
+		{ "array", test_array },
+		{ "map", test_map },
+		{ "string", test_string }
+	};
+
+	for (i = 0; i < sizeof(tests) / sizeof(tests[0]); i++) {
+		struct avro_tests *test = tests + i;
+		fprintf(stderr, "**** Running %s tests ****\n", test->name);
+		if (test->func() != 0) {
+			result = EXIT_FAILURE;
+		}
+	}
+	return result;
+}