Add test for reading chunks

2 files changed, 115 insertions, 1 deletions

diff --git a/tests/read_chunk_test.c b/tests/read_chunk_test.c
new file mode 100644
index 00000000..75750b8b
--- /dev/null
+++ b/tests/read_chunk_test.c
@@ -0,0 +1,112 @@
+/*
+  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.
+*/
+
+#undef NDEBUG
+
+#include "serd/serd.h"
+
+#include <assert.h>
+#include <stddef.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)
+{
+	(void)handle;
+	(void)uri;
+
+	++n_base;
+	return SERD_SUCCESS;
+}
+
+static SerdStatus
+on_prefix(void* handle, const SerdNode* name, const SerdNode* uri)
+{
+	(void)handle;
+	(void)name;
+	(void)uri;
+
+	++n_prefix;
+	return SERD_SUCCESS;
+}
+
+static SerdStatus
+on_statement(void*                handle,
+             SerdStatementFlags   flags,
+             const SerdStatement* statement)
+{
+	(void)handle;
+	(void)flags;
+	(void)statement;
+
+	++n_statement;
+	return SERD_SUCCESS;
+}
+
+static SerdStatus
+on_end(void* handle, const SerdNode* node)
+{
+	(void)handle;
+	(void)node;
+
+	++n_end;
+	return SERD_SUCCESS;
+}
+
+int
+main(void)
+{
+	SerdWorld* world = serd_world_new();
+	SerdSink*  sink  = serd_sink_new(NULL);
+	serd_sink_set_base_func(sink, on_base);
+	serd_sink_set_prefix_func(sink, on_prefix);
+	serd_sink_set_statement_func(sink, on_statement);
+	serd_sink_set_end_func(sink, on_end);
+
+	SerdReader* reader = serd_reader_new(world, SERD_TURTLE, sink, 4096);
+	assert(reader);
+
+	assert(!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",
+	                                NULL));
+
+	assert(!serd_reader_read_chunk(reader) && n_prefix == 1);
+	assert(!serd_reader_read_chunk(reader) && n_base == 1);
+	assert(!serd_reader_read_chunk(reader) && n_statement == 3);
+	assert(!serd_reader_read_chunk(reader) && n_statement == 5);
+	assert(!serd_reader_read_chunk(reader) && n_statement == 6);
+	assert(!serd_reader_read_chunk(reader) && n_statement == 8);
+	assert(serd_reader_read_chunk(reader) == SERD_FAILURE);
+	assert(n_end == 1);
+	assert(serd_reader_read_chunk(reader) == SERD_FAILURE);
+
+	serd_reader_free(reader);
+	serd_sink_free(sink);
+	serd_world_free(world);
+	return 0;
+}
diff --git a/wscript b/wscript
index 38e5e00d..a6101773 100644
--- a/wscript
+++ b/wscript
@@ -145,7 +145,8 @@ def build(bld):
         # Test programs
         for prog in [('serdi_static', 'src/serdi.c'),
                      ('cursor_test', 'tests/cursor_test.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',
@@ -417,6 +418,7 @@ def test(tst):
     with tst.group('Unit') as check:
         check(['./cursor_test'])
         check(['./serd_test'])
+        check(['./read_chunk_test'])
 
     def test_syntax_io(check, in_name, check_name, lang):
         in_path = 'tests/good/%s' % in_name