Move some test code to newline created .c files

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

1 files changed, 26 insertions, 0 deletions

diff --git a/strnreplace.c b/strnreplace.c
index f8a2036..eb26ec9 100644
--- a/strnreplace.c
+++ b/strnreplace.c
@@ -12,6 +12,32 @@ extern inline char *libsimple_strnreplace(char *, int, int, size_t);
 int
 main(void)
 {
+	char buf[100];
+
+	stpcpy(mempcpy(buf, "hello world", 12), "goodbye world");
+	assert(libsimple_strnreplace(buf, 'o', 'x', SIZE_MAX) == &buf[11]);
+	assert(!memcmp(buf, "hellx wxrld\0goodbye world", 26));
+
+	stpcpy(mempcpy(buf, "hello world", 12), "goodbye world");
+	assert(libsimple_strnreplace(buf, 'o', 'x', 12) == &buf[11]);
+	assert(!memcmp(buf, "hellx wxrld\0goodbye world", 26));
+
+	stpcpy(mempcpy(buf, "hello world", 12), "goodbye world");
+	assert(libsimple_strnreplace(buf, 'o', 'x', 11) == &buf[11]);
+	assert(!memcmp(buf, "hellx wxrld\0goodbye world", 26));
+
+	stpcpy(mempcpy(buf, "hello world", 12), "goodbye world");
+	assert(libsimple_strnreplace(buf, 'o', 'x', 6) == &buf[6]);
+	assert(!memcmp(buf, "hellx world\0goodbye world", 26));
+
+	stpcpy(mempcpy(buf, "hello world", 12), "goodbye world");
+	assert(libsimple_strnreplace(buf, 'o', 'x', 5) == &buf[5]);
+	assert(!memcmp(buf, "hellx world\0goodbye world", 26));
+
+	stpcpy(mempcpy(buf, "hello world", 12), "goodbye world");
+	assert(libsimple_strnreplace(buf, 'o', 'x', 4) == &buf[4]);
+	assert(!memcmp(buf, "hello world\0goodbye world", 26));
+
 	return 0;
 }