1 files changed, 139 insertions, 0 deletions
diff --git a/tests/test.c b/tests/test.c
new file mode 100644
index 0000000..a58602a
--- /dev/null
+++ b/tests/test.c
@@ -0,0 +1,139 @@
+#include <stdlib.h>
+#include <string.h>
+#include <stdio.h>
+#include <assert.h>
+#include <sys/types.h>
+#include "../src/extlib.h"
+
+int
+test_malloc_free_secure (size_t size)
+{
+  char *mall_test = malloc_secure (size);
+  if (!mall_test)
+    return 1;
+  if (memvcmp (mall_test, 0, size) != 0)
+    return 2;
+  free_secure ((void **)&mall_test, size);
+  if (mall_test)
+    return 3;
+  return 0;
+}
+
+int
+test_fcopy()
+{
+  char *buf_a;
+  char *buf_b;
+  size_t len_a;
+  size_t len_b;
+  FILE *file_test_a = open_memstream (&buf_a, &len_a);
+  FILE *file_test_b = open_memstream (&buf_b, &len_b);
+
+  fprintf (file_test_a, "hello");
+  fflush (file_test_a);
+  fclose (file_test_a);
+  file_test_a = fmemopen(buf_a, len_a, "r");
+
+  fcopy (file_test_a, file_test_b);
+  fclose (file_test_b);
+
+  if (memcmp(buf_a, buf_b, len_a) != 0) {
+    free_secure ((void**)&buf_a, len_a);
+    free_secure ((void **)&buf_b, len_b);
+    return 1;
+  }
+  free_secure ((void**)&buf_a, len_a);
+  free_secure ((void **)&buf_b, len_b);
+  return 0;
+}
+
+int
+test_strlwr_strupr ()
+{
+  char *test_string = strdup ("hello");
+  char *uppr_string = strupr (test_string);
+  char *lower_string = strlwr (uppr_string);
+
+  if (strcmp (uppr_string, "HELLO") != 0) {
+    free_secure ((void **)&test_string, strlen (test_string));
+    return 1;
+  }
+  if (strcmp (lower_string, "hello") != 0) {
+    free_secure ((void **)&test_string, strlen (test_string));
+    return 2;
+  }
+  free_secure ((void **)&test_string, strlen (test_string));
+  return 0;
+}
+
+int
+test_trim()
+{
+  char *test_string = strdup ("\t\thi\t\t");
+  char *trimmed_string = trim (test_string);
+  if (strstr (trimmed_string, "\t") != NULL) {
+    free_secure ((void**)&test_string, strlen (test_string));
+    return 1;
+  }
+  free_secure ((void**)&test_string, strlen (test_string));
+  return 0;
+}
+
+int
+test_replace_str ()
+{
+  char *test_string = strdup ("replace world!");
+  char *replaced_string = replace_str (test_string, "replace", "hello");
+  if (strcmp (replaced_string, "hello world!") != 0) {
+    free_secure ((void**)&test_string, strlen (test_string));
+    free_secure ((void**)&replaced_string, strlen (replaced_string));
+    return 1;
+  }
+  free_secure ((void**)&test_string, strlen (test_string));
+  free_secure ((void**)&replaced_string, strlen (replaced_string));
+  return 0;
+}
+
+
+int
+main (void)
+{
+  printf ("=== extlib unit tests ===\n");
+  printf ("\n");
+
+  printf ("Test Case 1: test_malloc_free_secure -- malloc_secure free_secure memvcmp\n");
+  int test_1_result = test_malloc_free_secure (12);
+  if (test_1_result == 0)
+    printf ("Test Case 1: test_malloc_free_secure -- SUCCESS\n\n");
+  else
+    printf ("Test Case 1: test_malloc_free_secure -- FAILED %d\n\n", test_1_result);
+
+  printf ("Test Case 2: test_fcopy -- fcopy\n");
+  int test_2_result = test_fcopy ();
+  if (test_2_result == 0)
+    printf ("Test Case 2: test_fcopy -- SUCCESS\n\n");
+  else
+    printf ("Test Case 2: test_fcopy -- FAILED %d\n\n", test_2_result);
+
+  printf ("Test Case 3: test_strlwr_strupr -- strlwr strupr\n");
+  int test_3_result = test_strlwr_strupr ();
+  if (test_2_result == 0)
+    printf ("Test Case 3: test_strlwr_strupr -- SUCCESS\n\n");
+  else
+    printf ("Test Case 3: test_strlwr_strupr -- FAILED %d\n\n", test_3_result);
+
+  printf ("Test Case 4: test_trim -- trim\n");
+  int test_4_result = test_trim ();
+  if (test_4_result == 0)
+    printf ("Test Case 4: test_trim -- SUCCESS\n\n");
+  else
+    printf ("Test Case 4: test_trim -- FAILED %d\n\n", test_4_result);
+
+  printf ("Test Case 5: test_replace_str -- replace_str\n");
+  int test_5_result = test_replace_str();
+  if (test_5_result == 0)
+    printf ("Test Case 5: test_replace_str -- SUCCESS\n\n");
+  else
+    printf ("Test Case 5: test_replace_str -- FAILED %d\n\n", test_5_result);
+  return 0;
+}