1 files changed, 21 insertions, 0 deletions
diff --git a/strnset.c b/strnset.c
index 909222f..b2e1da4 100644
--- a/strnset.c
+++ b/strnset.c
@@ -12,6 +12,27 @@ extern inline char *libsimple_strnset(char *, int, size_t);
 int
 main(void)
 {
+	char buf[100];
+
+	stpcpy(mempcpy(buf, "hello world", 12), "goodbye world");
+	assert(libsimple_strnset(buf, 'x', SIZE_MAX) == buf);
+	assert(!strcmp(buf, "xxxxxxxxxxx"));
+	assert(!strcmp(&buf[12], "goodbye world"));
+
+	stpcpy(mempcpy(buf, "hello world", 12), "goodbye world");
+	assert(libsimple_strnset(buf, '\0', SIZE_MAX) == buf);
+	assert(!memcmp(buf, "\0\0\0\0\0\0\0\0\0\0\0\0goodbye world", 26));
+
+
+	stpcpy(mempcpy(buf, "hello world", 12), "goodbye world");
+	assert(libsimple_strnset(buf, 'x', 5) == buf);
+	assert(!strcmp(buf, "xxxxx world"));
+	assert(!strcmp(&buf[12], "goodbye world"));
+
+	stpcpy(mempcpy(buf, "hello world", 12), "goodbye world");
+	assert(libsimple_strnset(buf, '\0', 5) == buf);
+	assert(!memcmp(buf, "\0\0\0\0\0 world\0goodbye world", 26));
+
 	return 0;
 }