1 files changed, 61 insertions, 0 deletions
diff --git a/nserver/tests/ncmd_tests.c b/nserver/tests/ncmd_tests.c
new file mode 100644
index 0000000..19e1d26
--- /dev/null
+++ b/nserver/tests/ncmd_tests.c
@@ -0,0 +1,61 @@
+#include "minunit.h"
+#include <ncmd.h>
+
+char *test_sanitize()
+{
+    char *cmd = calloc(128, sizeof(char));
+    mu_assert(cmd != NULL, "calloc failed");
+
+    char *c_cmd = "/create api/beef\n";
+    size_t c_cmd_sz = strlen(c_cmd);
+
+    strncpy(cmd, c_cmd, c_cmd_sz + 1);
+    mu_assert(strlen(cmd) == c_cmd_sz, "strncpy failed");
+
+    sanitize(cmd);
+    mu_assert(strlen(cmd) == c_cmd_sz - 1, "sanitize failed");
+    mu_assert(strcmp(cmd, "/create api/beef") == 0, "sanitize failed");
+
+
+    // Clear cmd.
+    memset(cmd, '\0', 128);
+
+
+    c_cmd = "/create api/ham\n\n\n";
+    c_cmd_sz = strlen(c_cmd);
+
+    strncpy(cmd, c_cmd, c_cmd_sz + 1);
+    mu_assert(strlen(cmd) == c_cmd_sz, "strncpy failed");
+
+    sanitize(cmd);
+    mu_assert(strlen(cmd) == c_cmd_sz - 3, "sanitize failed");
+    mu_assert(strcmp(cmd, "/create api/ham") == 0, "sanitize failed");
+
+
+    // Clear cmd.
+    memset(cmd, '\0', 128);
+
+
+    c_cmd = "/create api/bacon\n/create api/turkey\n";
+    c_cmd_sz = strlen(c_cmd);
+
+    strncpy(cmd, c_cmd, c_cmd_sz + 1);
+    mu_assert(strlen(cmd) == c_cmd_sz, "strncpy failed");
+
+    sanitize(cmd);
+    mu_assert(strlen(cmd) == c_cmd_sz - 20, "sanitize failed");
+    mu_assert(strcmp(cmd, "/create api/bacon") == 0, "sanitize failed");
+
+    return NULL;
+}
+
+char *all_tests()
+{
+    mu_suite_start();
+
+    mu_run_test(test_sanitize);
+
+    return NULL;
+}
+
+RUN_TESTS(all_tests);