From dfe6e2fc645f613a09d54bf58d60090ec207cdaa Mon Sep 17 00:00:00 2001
From: mo khan <mo.khan@gmail.com>
Date: Sun, 28 Jun 2020 13:26:56 -0600
Subject: Pop item from stack

---
 assignments/01/min_stack_test.c | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

(limited to 'assignments')

diff --git a/assignments/01/min_stack_test.c b/assignments/01/min_stack_test.c
index ef02bd6..580e3e7 100644
--- a/assignments/01/min_stack_test.c
+++ b/assignments/01/min_stack_test.c
@@ -55,11 +55,21 @@ static int min(Stack *stack) {
   return (int)NULL;
 }
 
+static int pop(Stack *stack) {
+  if (!stack->head)
+    return (int)NULL;
+
+  Node *current = stack->head;
+  stack->head = current->next;
+  return current->data;
+}
+
 Ensure(MinStack, when_empty) {
   Stack *stack = initialize();
 
   assert_that(size(stack), is_equal_to(0));
   assert_that(min(stack), is_equal_to(NULL));
+  assert_that(pop(stack), is_equal_to(NULL));
 
   free(stack);
 }
@@ -71,7 +81,8 @@ Ensure(MinStack, when_pushing_a_single_integer) {
 
   assert_that(size(stack), is_equal_to(1));
   assert_that(min(stack), is_equal_to(1));
-  /*assert_that(pop(stack), is_equal(1));*/
+  assert_that(pop(stack), is_equal_to(1));
+  assert_that(size(stack), is_equal_to(0));
 
   free(stack);
 }
-- 
cgit v1.2.3