Add a test for the printbuf functions.

diff --git a/tests/Makefile.am b/tests/Makefile.am
index c54144c..1de7723 100644 (file)
--- a/tests/Makefile.am
+++ b/tests/Makefile.am
@@ -20,6 +20,11 @@ test_cast_LDADD = $(LIBJSON_LA)
 test_parse_LDADD = $(LIBJSON_LA)
 
 TESTS = test1.test test2.test test4.test parse_int64.test test_null.test test_cast.test test_parse.test
+
+TESTS+= test_printbuf.test
+check_PROGRAMS+=test_printbuf
+test_printbuf_LDADD = $(LIBJSON_LA)
+
 EXTRA_DIST=
 EXTRA_DIST += $(TESTS)
 

diff --git a/tests/test_printbuf.c b/tests/test_printbuf.c
new file mode 100644 (file)
index 0000000..3676b54
--- /dev/null
+++ b/tests/test_printbuf.c
@@ -0,0 +1,165 @@
+#include <assert.h>
+#include <stddef.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <limits.h>
+
+#include "debug.h"
+#include "printbuf.h"
+
+static void test_basic_printbuf_memset(void);
+static void test_printbuf_memset_length(void);
+
+static void test_basic_printbuf_memset()
+{
+       struct printbuf *pb;
+       
+       printf("%s: starting test\n", __func__);
+       pb = printbuf_new();
+       sprintbuf(pb, "blue:%d", 1);
+       printbuf_memset(pb, -1, 'x', 52);
+       printf("Buffer contents:%.*s\n", printbuf_length(pb), pb->buf);
+       printbuf_free(pb);
+       printf("%s: end test\n", __func__);
+}
+
+static void test_printbuf_memset_length()
+{
+       struct printbuf *pb;
+
+       printf("%s: starting test\n", __func__);
+       pb = printbuf_new();
+       printbuf_memset(pb, -1, ' ', 0);
+       printbuf_memset(pb, -1, ' ', 0);
+       printbuf_memset(pb, -1, ' ', 0);
+       printbuf_memset(pb, -1, ' ', 0);
+       printbuf_memset(pb, -1, ' ', 0);
+       printf("Buffer length: %d\n", printbuf_length(pb));
+       printbuf_memset(pb, -1, ' ', 2);
+       printbuf_memset(pb, -1, ' ', 4);
+       printbuf_memset(pb, -1, ' ', 6);
+       printf("Buffer length: %d\n", printbuf_length(pb));
+       printbuf_memset(pb, -1, ' ', 6);
+       printf("Buffer length: %d\n", printbuf_length(pb));
+       printbuf_memset(pb, -1, ' ', 8);
+       printbuf_memset(pb, -1, ' ', 10);
+       printbuf_memset(pb, -1, ' ', 10);
+       printbuf_memset(pb, -1, ' ', 10);
+       printbuf_memset(pb, -1, ' ', 20);
+       printf("Buffer length: %d\n", printbuf_length(pb));
+
+       // No length change should occur
+       printbuf_memset(pb, 0, 'x', 30);
+       printf("Buffer length: %d\n", printbuf_length(pb));
+
+       // This should extend it by one.
+       printbuf_memset(pb, 0, 'x', printbuf_length(pb) + 1);
+       printf("Buffer length: %d\n", printbuf_length(pb));
+
+       printbuf_free(pb);
+       printf("%s: end test\n", __func__);
+}
+
+static void test_printbuf_memappend(int *before_resize);
+static void test_printbuf_memappend(int *before_resize)
+{
+       struct printbuf *pb;
+       int initial_size;
+
+       printf("%s: starting test\n", __func__);
+       pb = printbuf_new();
+       printf("Buffer length: %d\n", printbuf_length(pb));
+
+       initial_size = pb->size;
+
+       while(pb->size == initial_size)
+       {
+               printbuf_memappend_fast(pb, "x", 1);
+       }
+       *before_resize = printbuf_length(pb) - 1;
+       printf("Appended %d bytes for resize: [%s]\n", *before_resize + 1, pb->buf);
+
+       printbuf_reset(pb);
+       printbuf_memappend_fast(pb, "bluexyz123", 3);
+       printf("Partial append: %d, [%s]\n", printbuf_length(pb), pb->buf);
+
+       char with_nulls[] = { 'a', 'b', '\0', 'c' };
+       printbuf_reset(pb);
+       printbuf_memappend_fast(pb, with_nulls, sizeof(with_nulls));
+       printf("With embedded \\0 character: %d, [%s]\n", printbuf_length(pb), pb->buf);
+
+       printbuf_free(pb);
+       pb = printbuf_new();
+       char *data = malloc(*before_resize);
+       memset(data, 'X', *before_resize);
+       printbuf_memappend_fast(pb, data, *before_resize);
+       printf("Append to just before resize: %d, [%s]\n", printbuf_length(pb), pb->buf);
+
+       free(data);
+       printbuf_free(pb);
+
+       pb = printbuf_new();
+       data = malloc(*before_resize + 1);
+       memset(data, 'X', *before_resize + 1);
+       printbuf_memappend_fast(pb, data, *before_resize + 1);
+       printf("Append to just after resize: %d, [%s]\n", printbuf_length(pb), pb->buf);
+       
+       free(data);
+
+       printbuf_free(pb);
+       printf("%s: end test\n", __func__);
+}
+
+static void test_sprintbuf(int before_resize);
+static void test_sprintbuf(int before_resize)
+{
+       struct printbuf *pb;
+
+       printf("%s: starting test\n", __func__);
+       pb = printbuf_new();
+       printf("Buffer length: %d\n", printbuf_length(pb));
+
+       char *data = malloc(before_resize + 1 + 1);
+       memset(data, 'X', before_resize + 1 + 1);
+       data[before_resize + 1] = '\0';
+       sprintbuf(pb, "%s", data);
+       printf("sprintbuf to just after resize(%d+1): %d, [%s], strlen(buf)=%d\n", before_resize, printbuf_length(pb), pb->buf, strlen(pb->buf));
+
+       printbuf_reset(pb);
+       sprintbuf(pb, "plain");
+       printf("%d, [%s]\n", printbuf_length(pb), pb->buf);
+
+       sprintbuf(pb, "%d", 1);
+       printf("%d, [%s]\n", printbuf_length(pb), pb->buf);
+
+       sprintbuf(pb, "%d", INT_MAX);
+       printf("%d, [%s]\n", printbuf_length(pb), pb->buf);
+
+       sprintbuf(pb, "%d", INT_MIN);
+       printf("%d, [%s]\n", printbuf_length(pb), pb->buf);
+
+       sprintbuf(pb, "%s", "%s");
+       printf("%d, [%s]\n", printbuf_length(pb), pb->buf);
+
+       printbuf_free(pb);
+       printf("%s: end test\n", __func__);
+}
+
+int main(int argc, char **argv)
+{
+       int before_resize = 0;
+
+       mc_set_debug(1);
+
+       test_basic_printbuf_memset();
+       printf("========================================\n");
+       test_printbuf_memset_length();
+       printf("========================================\n");
+       test_printbuf_memappend(&before_resize);
+       printf("========================================\n");
+       test_sprintbuf(before_resize);
+       printf("========================================\n");
+
+       return 0;
+}

diff --git a/tests/test_printbuf.test b/tests/test_printbuf.test
new file mode 100755 (executable)
index 0000000..09d27c2
--- /dev/null
+++ b/tests/test_printbuf.test
@@ -0,0 +1,12 @@
+#!/bin/sh
+
+# Common definitions
+if test -z "$srcdir"; then
+    srcdir="${0%/*}"
+    test "$srcdir" = "$0" && srcdir=.
+    test -z "$srcdir" && srcdir=.
+fi
+. "$srcdir/test-defs.sh"
+
+run_output_test test_printbuf
+exit $?