Insert into root of tree

1 files changed, 17 insertions, 1 deletions

diff --git a/src/03/avl_tree_test.c b/src/03/avl_tree_test.c
index 75902d3..5f8c327 100644
--- a/src/03/avl_tree_test.c
+++ b/src/03/avl_tree_test.c
@@ -3,12 +3,28 @@
 #include <string.h>
 
 Ensure(initialize_returns_new_tree) {
-  assert_that(NULL, is_equal_to(NULL));
+  AVLTree *tree = avl_tree_init();
+  assert_that(tree, is_not_equal_to(NULL));
+}
+
+Ensure(size_returns_zero) {
+  AVLTree *tree = avl_tree_init();
+
+  assert_that(avl_tree_size(tree), is_equal_to(0));
+}
+
+Ensure(insert_changes_size) {
+  AVLTree *tree = avl_tree_init();
+  avl_tree_insert(tree, 33);
+
+  assert_that(avl_tree_size(tree), is_equal_to(1));
 }
 
 TestSuite *avl_tree_tests() {
   TestSuite *suite = create_test_suite();
   add_test(suite, initialize_returns_new_tree);
+  add_test(suite, size_returns_zero);
+  add_test(suite, insert_changes_size);
   return suite;
 }