1 files changed, 97 insertions, 0 deletions
diff --git a/test/inputstream.c b/test/inputstream.c
new file mode 100644
index 0000000..bad3127
--- /dev/null
+++ b/test/inputstream.c
@@ -0,0 +1,97 @@
+#include <inttypes.h>
+#include <stdio.h>
+
+#include <parserutils/parserutils.h>
+#include <parserutils/charset/utf8.h>
+#include <parserutils/input/inputstream.h>
+
+#include "utils/utils.h"
+
+#include "testutils.h"
+
+static void *myrealloc(void *ptr, size_t len, void *pw)
+{
+	UNUSED(pw);
+
+	return realloc(ptr, len);
+}
+
+int main(int argc, char **argv)
+{
+	parserutils_inputstream *stream;
+	FILE *fp;
+	size_t len, origlen;
+#define CHUNK_SIZE (4096)
+	uint8_t buf[CHUNK_SIZE];
+	uintptr_t c;
+	size_t clen;
+
+	if (argc != 3) {
+		printf("Usage: %s <aliases_file> <filename>\n", argv[0]);
+		return 1;
+	}
+
+	/* Initialise library */
+	assert(parserutils_initialise(argv[1], myrealloc, NULL) == 
+			PARSERUTILS_OK);
+
+	stream = parserutils_inputstream_create("UTF-8", 1, NULL, 
+			myrealloc, NULL);
+	assert(stream != NULL);
+
+	fp = fopen(argv[2], "rb");
+	if (fp == NULL) {
+		printf("Failed opening %s\n", argv[2]);
+		return 1;
+	}
+
+	fseek(fp, 0, SEEK_END);
+	origlen = len = ftell(fp);
+	fseek(fp, 0, SEEK_SET);
+
+	while (len >= CHUNK_SIZE) {
+		fread(buf, 1, CHUNK_SIZE, fp);
+
+		assert(parserutils_inputstream_append(stream,
+				buf, CHUNK_SIZE) == PARSERUTILS_OK);
+
+		len -= CHUNK_SIZE;
+
+		while ((c = parserutils_inputstream_peek(stream, 0, &clen)) !=
+				PARSERUTILS_INPUTSTREAM_OOD) {
+			parserutils_inputstream_advance(stream, clen);
+		}
+	}
+
+	if (len > 0) {
+		fread(buf, 1, len, fp);
+
+		assert(parserutils_inputstream_append(stream,
+				buf, len) == PARSERUTILS_OK);
+
+		len = 0;
+	}
+
+	fclose(fp);
+
+	assert(parserutils_inputstream_insert(stream,
+			(const uint8_t *) "hello!!!",
+			SLEN("hello!!!")) == PARSERUTILS_OK);
+
+	assert(parserutils_inputstream_append(stream, NULL, 0) == 
+			PARSERUTILS_OK);
+
+	while ((c = parserutils_inputstream_peek(stream, 0, &clen)) !=
+			PARSERUTILS_INPUTSTREAM_EOF) {
+		parserutils_inputstream_advance(stream, clen);
+	}
+
+	parserutils_inputstream_destroy(stream);
+
+	assert(parserutils_finalise(myrealloc, NULL) == PARSERUTILS_OK);
+
+	printf("PASS\n");
+
+	return 0;
+}
+