Move some test code to newline created .c files

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

1 files changed, 11 insertions, 0 deletions

diff --git a/strset.c b/strset.c
index 2c62c0d..8d92f31 100644
--- a/strset.c
+++ b/strset.c
@@ -12,6 +12,17 @@ extern inline char *libsimple_strset(char *, int);
 int
 main(void)
 {
+	char buf[100];
+
+	stpcpy(mempcpy(buf, "hello world", 12), "goodbye world");
+	assert(libsimple_strset(buf, 'x') == buf);
+	assert(!strcmp(buf, "xxxxxxxxxxx"));
+	assert(!strcmp(&buf[12], "goodbye world"));
+
+	stpcpy(mempcpy(buf, "hello world", 12), "goodbye world");
+	assert(libsimple_strset(buf, '\0') == buf);
+	assert(!memcmp(buf, "\0\0\0\0\0\0\0\0\0\0\0\0goodbye world", 26));
+
 	return 0;
 }