diff options
| -rw-r--r-- | tests/framework.h | 22 | ||||
| -rw-r--r-- | tests/test_checksum_sha256.c | 32 |
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 |