crypto/shachain: new module

[ccan] / ccan / crypto / shachain / test / run.c

diff --git a/ccan/crypto/shachain/test/run.c b/ccan/crypto/shachain/test/run.c
new file mode 100644 (file)
index 0000000..3939168
--- /dev/null
+++ b/ccan/crypto/shachain/test/run.c
@@ -0,0 +1,42 @@
+#include <ccan/crypto/shachain/shachain.h>
+/* Include the C files directly. */
+#include <ccan/crypto/shachain/shachain.c>
+#include <ccan/tap/tap.h>
+
+#define NUM_TESTS 50
+
+int main(void)
+{
+       struct sha256 seed;
+       struct shachain chain;
+       struct sha256 expect[NUM_TESTS];
+       size_t i, j;
+
+       /* This is how many tests you plan to run */
+       plan_tests(NUM_TESTS + NUM_TESTS * (NUM_TESTS + 1) + NUM_TESTS);
+
+       memset(&seed, 0, sizeof(seed));
+       /* Generate a whole heap. */
+       for (i = 0; i < NUM_TESTS; i++) {
+               shachain_from_seed(&seed, i, &expect[i]);
+               if (i == 0)
+                       ok1(memcmp(&expect[i], &seed, sizeof(expect[i])));
+               else
+                       ok1(memcmp(&expect[i], &expect[i-1], sizeof(expect[i])));
+       }
+
+       shachain_init(&chain);
+
+       for (i = 0; i < NUM_TESTS; i++) {
+               struct sha256 hash;
+
+               shachain_add_hash(&chain, i, &expect[i]);
+               for (j = 0; j <= i; j++) {
+                       ok1(shachain_get_hash(&chain, j, &hash));
+                       ok1(memcmp(&hash, &expect[j], sizeof(hash)) == 0);
+               }
+               ok1(!shachain_get_hash(&chain, i+1, &hash));
+       }
+
+       return exit_status();
+}