Add test_checksum_sha256 test

2 files changed, 54 insertions, 0 deletions

diff --git a/tests/framework.h b/tests/framework.h
index f1ce15c..7770ed7 100644
--- a/tests/framework.h
+++ b/tests/framework.h
@@ -51,6 +51,28 @@ char *array_to_string(char **array, const char *sep) {
     return buffer;
 }
 
+/**
+ * Write data to a file
+ * @param filename name of file
+ * @param data data to write
+ * @param size type of data
+ * @param nelem number of elements of `size` to write
+ * @return bytes written (0 = failure)
+ */
+size_t mock(const char *filename, void *data, size_t size, size_t nelem) {
+    unsigned long written = 0;
+    FILE *fp = fopen(filename, "w+b");
+    if (fp == NULL) {
+        perror(filename);
+        exit(errno);
+    }
+    if ((written = fwrite(data, size, nelem, fp)) == 0) {
+        fprintf(stderr, "warning: %s: no data written\n", filename);
+    }
+    fclose(fp);
+    return written;
+}
+
 #define myassert(condition, ...) \
     do { \
         if (!(condition)) { \
diff --git a/tests/test_checksum_sha256.c b/tests/test_checksum_sha256.c
new file mode 100644
index 0000000..24b1a3e
--- /dev/null
+++ b/tests/test_checksum_sha256.c
@@ -0,0 +1,32 @@
+#include "spm.h"
+#include "shell.h"
+#include "framework.h"
+
+const char *testFmt = "returned '%s', expected '%s'\n";
+struct TestCase testCase[] = {
+    {.caseValue.sptr = "check this hash", .truthValue.sptr = "72a9728d7b00d42fe573d91d023e2ebe4fa2c3c5d38c67306cbcd4f5e410aa33"},
+    {.caseValue.sptr = "check this hash", .truthValue.sptr = "72a9728d7b00d42fe573d91d023e2ebe4fa2c3c5d38c67306cbcd4f5e410aa33"},
+    {.caseValue.sptr = "check this hash", .truthValue.sptr = "72a9728d7b00d42fe573d91d023e2ebe4fa2c3c5d38c67306cbcd4f5e410aa33"},
+    {.caseValue.sptr = "check this hash", .truthValue.sptr = "72a9728d7b00d42fe573d91d023e2ebe4fa2c3c5d38c67306cbcd4f5e410aa33"},
+};
+size_t numCases = sizeof(testCase) / sizeof(struct TestCase);
+
+int main(int argc, char *argv[]) {
+    char filename[NAME_MAX] = {0};
+    sprintf(filename, "%s.%s.mock", basename(__FILE__), __FUNCTION__);
+
+    for (size_t i = 0; i < numCases; i++) {
+        char *result = NULL;
+        char *data = strdup(testCase[i].caseValue.sptr);
+
+        mock(filename, data, sizeof(char), strlen(testCase[i].caseValue.sptr));
+        result = sha256sum(filename);
+
+        myassert(strcmp(result, testCase[i].truthValue.sptr) == 0, testFmt, result, testCase[i].truthValue.sptr);
+
+        free(result);
+        free(data);
+        unlink(filename);
+    }
+    return 0;
+}
\ No newline at end of file