1 files changed, 145 insertions, 0 deletions
diff --git a/test/testutils.h b/test/testutils.h
new file mode 100644
index 0000000..ab3b29d
--- /dev/null
+++ b/test/testutils.h
@@ -0,0 +1,145 @@
+#ifndef dom_test_testutils_h_
+#define dom_test_testutils_h_
+
+#include <stdbool.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <inttypes.h>
+
+#include "xmlbinding.h"
+#include "xmlparser.h"
+
+#ifndef UNUSED
+#define UNUSED(x) ((x) = (x))
+#endif
+
+/* Redefine assert, so we can simply use the standard assert mechanism
+ * within testcases and exit with the right output for the testrunner
+ * to do the right thing. */
+void __assert2(const char *expr, const char *function,
+		const char *file, int line);
+
+void __assert2(const char *expr, const char *function,
+		const char *file, int line)
+{
+	UNUSED(function);
+	UNUSED(file);
+
+	printf("FAIL - %s at line %d\n", expr, line);
+
+	exit(EXIT_FAILURE);
+}
+
+#define assert(expr) \
+  ((void) ((expr) || (__assert2 (#expr, __func__, __FILE__, __LINE__), 0)))
+
+static void *myrealloc(void *ptr, size_t len, void *pw)
+{
+	UNUSED(pw);
+
+	return realloc(ptr, len);
+}
+
+typedef struct TestObject {
+	xml_parser *parser;
+	struct dom_document *doc;
+} TestObject;
+
+TestObject *test_object_create(int argc, char **argv,
+		const char *uri, bool will_be_modified);
+struct dom_document *test_object_get_doc(TestObject *obj);
+const char *test_object_get_mimetype(TestObject *obj);
+
+TestObject *test_object_create(int argc, char **argv,
+		const char *uri, bool will_be_modified)
+{
+	static bool xml_parser_initialised;
+
+	char fnbuf[1024];
+#define CHUNK_SIZE 4096
+	uint8_t buf[CHUNK_SIZE];
+	FILE *fp;
+	size_t len;
+	TestObject *ret;
+
+	UNUSED(will_be_modified);
+
+	if (argc != 2) {
+		printf("Usage: %s <datapath>\n", argv[0]);
+		exit(EXIT_FAILURE);
+	}
+
+	if (xml_parser_initialised == false) {
+		assert(xml_dom_binding_initialise(myrealloc, NULL) == XML_OK);
+
+		xml_parser_initialised = true;
+	}
+
+	snprintf(fnbuf, sizeof fnbuf, "%s/%s.xml", argv[1], uri);
+
+	ret = malloc(sizeof(TestObject));
+	if (ret == NULL)
+		return NULL;
+
+	ret->parser = xml_parser_create(NULL, "UTF-8", myrealloc, NULL);
+	if (ret->parser == NULL) {
+		free(ret);
+		return NULL;
+	}
+
+	fp = fopen(fnbuf, "r");
+	if (fp == NULL) {
+		xml_parser_destroy(ret->parser);
+		free(ret);
+		return NULL;
+	}
+
+	fseek(fp, 0, SEEK_END);
+	len = ftell(fp);
+	fseek(fp, 0, SEEK_SET);
+
+	while (len > CHUNK_SIZE) {
+		fread(buf, 1, CHUNK_SIZE, fp);
+
+		assert(xml_parser_parse_chunk(ret->parser, buf,
+				CHUNK_SIZE) == XML_OK);
+
+		len -= CHUNK_SIZE;
+	}
+
+	if (len > 0) {
+		fread(buf, 1, len, fp);
+
+		assert(xml_parser_parse_chunk(ret->parser, buf,
+				len) == XML_OK);
+
+		len = 0;
+
+		assert(xml_parser_completed(ret->parser) == XML_OK);
+	}
+
+	fclose(fp);
+
+	ret->doc = xml_parser_get_document(ret->parser);
+
+	xml_parser_destroy(ret->parser);
+	ret->parser = NULL;
+
+	return ret;
+
+#undef CHUNK_SIZE
+}
+
+struct dom_document *test_object_get_doc(TestObject *obj)
+{
+	return obj->doc;
+}
+
+const char *test_object_get_mimetype(TestObject *obj)
+{
+	UNUSED(obj);
+
+	return "text/xml";
+}
+
+#endif