Add tests

Signed-off-by: Mattias Andrée <m@maandree.se>

1 files changed, 15 insertions, 1 deletions

diff --git a/librecrypt_concat_str_.c b/librecrypt_concat_str_.c
index 6ad7e48..d89f4dd 100644
--- a/librecrypt_concat_str_.c
+++ b/librecrypt_concat_str_.c
@@ -12,10 +12,24 @@ extern inline void librecrypt_concat_str_(struct concat_state *state, const char
 int
 main(void)
 {
+	char buf[5];
+	struct concat_state state = {buf, sizeof(buf), 0u};
+
 	SET_UP_ALARM();
 	INIT_RESOURCE_TEST();
 
-	/* TODO test */
+	memset(buf, 'x', sizeof(buf));
+
+	librecrypt_concat_str_(&state, "ab");
+	librecrypt_concat_str_(&state, "cdef");
+	assert(!memcmp(buf, "abcd\0", sizeof(buf)));
+	assert(state.buf == &buf[4u]);
+	assert(state.size == 1u);
+	assert(state.len == 6u);
+
+	state = (struct concat_state){NULL, 0u, 0u};
+	librecrypt_concat_str_(&state, "abc");
+	assert(state.len == 3u);
 
 	STOP_RESOURCE_TEST();
 	return 0;