Add test for reading chunks

2 files changed, 112 insertions, 2 deletions

diff --git a/tests/read_chunk_test.c b/tests/read_chunk_test.c
new file mode 100644
index 00000000..aadbe2c8
--- /dev/null
+++ b/tests/read_chunk_test.c
@@ -0,0 +1,108 @@
+/*
+  Copyright 2018 David Robillard <http://drobilla.net>
+
+  Permission to use, copy, modify, and/or distribute this software for any
+  purpose with or without fee is hereby granted, provided that the above
+  copyright notice and this permission notice appear in all copies.
+
+  THIS SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+  WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+  MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+  ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+  WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+  ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+  OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+*/
+
+#include "serd/serd.h"
+
+#include "test_utils.h"
+
+static size_t n_base      = 0;
+static size_t n_prefix    = 0;
+static size_t n_statement = 0;
+static size_t n_end       = 0;
+
+static SerdStatus
+on_base(void* handle, const SerdNode* uri)
+{
+	++n_base;
+	return SERD_SUCCESS;
+}
+
+static SerdStatus
+on_prefix(void* handle, const SerdNode* name, const SerdNode* uri)
+{
+	++n_prefix;
+	return SERD_SUCCESS;
+}
+
+static SerdStatus
+on_statement(void*              handle,
+             SerdStatementFlags flags,
+             const SerdNode*    graph,
+             const SerdNode*    subject,
+             const SerdNode*    predicate,
+             const SerdNode*    object)
+{
+	++n_statement;
+	return SERD_SUCCESS;
+}
+
+static SerdStatus
+on_end(void* handle, const SerdNode* node)
+{
+	++n_end;
+	return SERD_SUCCESS;
+}
+
+int
+main(int argc, char** argv)
+{
+	SerdReader* reader = serd_reader_new(SERD_TURTLE,
+	                                     NULL,
+	                                     NULL,
+	                                     on_base,
+	                                     on_prefix,
+	                                     on_statement,
+	                                     on_end);
+	if (!reader) {
+		FAIL("Failed to create reader\n");
+	}
+
+	if (serd_reader_start_string(reader,
+	                             "@prefix eg: <http://example.org/> .\n"
+	                             "@base <http://example.org/base> .\n"
+	                             "eg:s1 eg:p1 eg:o1 ;\n"
+	                             "      eg:p2 eg:o2 ,\n"
+	                             "            eg:o3 .\n"
+	                             "eg:s2 eg:p1 eg:o1 ;\n"
+	                             "      eg:p2 eg:o2 .\n"
+	                             "eg:s3 eg:p1 eg:o1 .\n"
+	                             "eg:s4 eg:p1 [ eg:p3 eg:o1 ] .\n")) {
+		FAIL("Failed to start string read\n");
+	}
+
+	if (serd_reader_read_chunk(reader) || n_prefix != 1) {
+		FAILF("%zu prefixes after chunk, expected 1\n", n_prefix);
+	} else if (serd_reader_read_chunk(reader) || n_base != 1) {
+		FAILF("%zu bases after chunk, expected 1\n", n_base);
+	} else if (serd_reader_read_chunk(reader) || n_statement != 3) {
+		FAILF("%zu statements after chunk, expected 3\n", n_statement);
+	} else if (serd_reader_read_chunk(reader) || n_statement != 5) {
+		FAILF("%zu statements after chunk, expected 5\n", n_statement);
+	} else if (serd_reader_read_chunk(reader) || n_statement != 6) {
+		FAILF("%zu statements after chunk, expected 6\n", n_statement);
+	} else if (serd_reader_read_chunk(reader) || n_statement != 8) {
+		FAILF("%zu statements after chunk, expected 6\n", n_statement);
+	} else if (serd_reader_read_chunk(reader) || n_end != 1) {
+		FAILF("Expected end after %zu statements\n", n_statement);
+	} else if (serd_reader_read_chunk(reader) != SERD_FAILURE) {
+		FAIL("Successfully read past end of input\n");
+	} else if (serd_reader_read_chunk(reader) != SERD_FAILURE) {
+		FAIL("Successfully read past end of input\n");
+	}
+
+	serd_reader_free(reader);
+	return 0;
+}
diff --git a/wscript b/wscript
index 3849543d..26efaa35 100644
--- a/wscript
+++ b/wscript
@@ -140,7 +140,8 @@ def build(bld):
 
         # Test programs
         for prog in [('serdi_static', 'src/serdi.c'),
-                     ('serd_test', 'tests/serd_test.c')]:
+                     ('serd_test', 'tests/serd_test.c'),
+                     ('read_chunk_test', 'tests/read_chunk_test.c')]:
             bld(features     = 'c cprogram',
                 source       = prog[1],
                 use          = 'libserd_profiled',
@@ -426,7 +427,8 @@ def test(ctx):
     os.environ['PATH'] = '.' + os.pathsep + os.getenv('PATH')
 
     autowaf.pre_test(ctx, APPNAME)
-    autowaf.run_test(ctx, APPNAME, 'serd_test', dirs=['.'])
+    for test in ['serd_test', 'read_chunk_test']:
+        autowaf.run_test(ctx, APPNAME, test, dirs=['.'])
 
     def test_ttl(in_name, expected_name):
         in_path = 'tests/good/%s.ttl' % in_name