initialize head of treeHEADmain

3 files changed, 14 insertions, 4 deletions

diff --git a/merkle_tree.c b/merkle_tree.c
index 2b0ac0d..c3da3a1 100644
--- a/merkle_tree.c
+++ b/merkle_tree.c
@@ -4,5 +4,6 @@
 MTree *mtree_initialize(int size) {
   MTree *tree = malloc(sizeof(MTree));
   tree->max_size = size;
+  tree->head = NULL;
   return tree;
 }
diff --git a/merkle_tree.h b/merkle_tree.h
index 8f83a03..85a93d8 100644
--- a/merkle_tree.h
+++ b/merkle_tree.h
@@ -1,5 +1,9 @@
+struct node {
+};
+
 typedef struct {
   int max_size;
+  struct node *head;
 } MTree;
 
 MTree* mtree_initialize(int size);
diff --git a/merkle_tree_test.c b/merkle_tree_test.c
index f4623c2..ee13e2d 100644
--- a/merkle_tree_test.c
+++ b/merkle_tree_test.c
@@ -1,19 +1,24 @@
 #include "merkle_tree.h"
 #include <cgreen/cgreen.h>
 
-Ensure(initialize_returns_new_tree) {
+Ensure(initialize__returns_new_tree) {
   MTree *tree = mtree_initialize(256);
   assert_that(tree, is_not_equal_to(NULL));
 }
 
-Ensure(returns_max_size_of_tree) {
+Ensure(initialize__assigns_max_size_of_tree) {
   MTree *tree = mtree_initialize(256);
   assert_that(tree->max_size, is_equal_to(256));
 };
 
+Ensure(initialize__assign_null_head) {
+  MTree *tree = mtree_initialize(256);
+  assert_that(tree->head, is_equal_to(NULL));
+};
+
 TestSuite *merkle_tree_tests() {
   TestSuite *x = create_test_suite();
-  add_test(x, initialize_returns_new_tree);
-  add_test(x, returns_max_size_of_tree);
+  add_test(x, initialize__returns_new_tree);
+  add_test(x, initialize__assigns_max_size_of_tree);
   return x;
 }