Allow storing void* on Stack

1 files changed, 22 insertions, 8 deletions

diff --git a/src/02/05/stack_test.c b/src/02/05/stack_test.c
index 5c74290..f53fe58 100644
--- a/src/02/05/stack_test.c
+++ b/src/02/05/stack_test.c
@@ -9,7 +9,7 @@ AfterEach(Stack) {}
 Ensure(Stack, when_pushing_an_item_on_to_a_stack) {
   Stack *stack = stack_init();
 
-  stack_push(stack, 10);
+  stack_push(stack, (void *)10);
 
   assert_that(stack_size(stack), is_equal_to(1));
   assert_that(stack_peek(stack), is_equal_to(10));
@@ -18,22 +18,36 @@ Ensure(Stack, when_pushing_an_item_on_to_a_stack) {
 Ensure(Stack, when_pushing_multiple_items_on_to_a_stack) {
   Stack *stack = stack_init();
 
-  stack_push(stack, 20);
-  stack_push(stack, 10);
-  stack_push(stack, 50);
+  stack_push(stack, (void *)20);
+  stack_push(stack, (void *)10);
+  stack_push(stack, (void *)50);
 
   assert_that(stack_size(stack), is_equal_to(3));
   assert_that(stack_peek(stack), is_equal_to(50));
 }
 
+Ensure(Stack, when_pushing_a_custom_type_on_to_a_stack) {
+  typedef struct {
+  } Item;
+
+  Stack *stack = stack_init();
+  Item *item = malloc(sizeof(Item));
+
+  stack_push(stack, item);
+
+  assert_that(stack_size(stack), is_equal_to(1));
+  assert_that(stack_peek(stack), is_equal_to(item));
+  assert_that(stack_pop(stack), is_equal_to(item));
+}
+
 Ensure(Stack, when_popping_an_item_off_of_a_stack) {
   Stack *stack = stack_init();
 
-  stack_push(stack, 20);
-  stack_push(stack, 10);
-  stack_push(stack, 50);
+  stack_push(stack, (void *)20);
+  stack_push(stack, (void *)10);
+  stack_push(stack, (void *)50);
 
-  int result = stack_pop(stack);
+  void *result = stack_pop(stack);
 
   assert_that(stack_size(stack), is_equal_to(2));
   assert_that(stack_peek(stack), is_equal_to(10));