Fix tests to match expected behaviour of a min stack

1 files changed, 12 insertions, 3 deletions

diff --git a/src/01/06/min_stack_test.c b/src/01/06/min_stack_test.c
index a02c163..b892d5f 100644
--- a/src/01/06/min_stack_test.c
+++ b/src/01/06/min_stack_test.c
@@ -40,12 +40,15 @@ Ensure(MinStack, when_pushing_multiple_integers_out_of_order) {
 
   assert_that(pop(stack), is_equal_to(1));
   assert_that(size(stack), is_equal_to(2));
+  assert_that(min(stack), is_equal_to(2));
 
-  assert_that(pop(stack), is_equal_to(2));
+  assert_that(pop(stack), is_equal_to(3));
   assert_that(size(stack), is_equal_to(1));
+  assert_that(min(stack), is_equal_to(3));
 
-  assert_that(pop(stack), is_equal_to(3));
+  assert_that(pop(stack), is_equal_to(2));
   assert_that(size(stack), is_equal_to(0));
+  assert_that(min(stack), is_equal_to(NULL));
 
   assert_that(pop(stack), is_equal_to(NULL));
   assert_that(size(stack), is_equal_to(0));
@@ -63,9 +66,15 @@ Ensure(MinStack, when_pushing_duplicate_values_on_to_the_stack) {
   assert_that(size(stack), is_equal_to(3));
   assert_that(min(stack), is_equal_to(1));
 
-  assert_that(pop(stack), is_equal_to(1));
   assert_that(pop(stack), is_equal_to(2));
+  assert_that(min(stack), is_equal_to(1));
+
+  assert_that(pop(stack), is_equal_to(1));
+  assert_that(min(stack), is_equal_to(2));
+
   assert_that(pop(stack), is_equal_to(2));
+  assert_that(min(stack), is_equal_to(NULL));
+
   assert_that(pop(stack), is_equal_to(NULL));
 
   free(stack);