diff --git a/src/main/java/MyStack.java b/src/main/java/MyStack.java index d1a485d781842c310fa3769b2f39dbbc8e7ea7d2..64b81c4e544372b95972dbb36c5b7b4f74dac6f8 100644 --- a/src/main/java/MyStack.java +++ b/src/main/java/MyStack.java @@ -15,6 +15,10 @@ public class MyStack { this.top = new Node(element, top); } + public Object pop() { + return null; + } + private class Node { private Object payload; private int height; diff --git a/src/test/java/MyStackTest.java b/src/test/java/MyStackTest.java index 562c9d052e58e121c7fa44ee2317f5e339d28292..850509e4bdb5c290d5d169931bedd10eacb30fed 100644 --- a/src/test/java/MyStackTest.java +++ b/src/test/java/MyStackTest.java @@ -19,7 +19,7 @@ public class MyStackTest { private void pushDummies(int numberOfDummies) { for (int i=0; i<numberOfDummies; i++) { - stack.push(new Object()); + stack.push(i+1); } } @@ -39,4 +39,10 @@ public class MyStackTest { pushDummies(3); assertEquals(3, stack.size()); } + + @Test + public void testPopFromHeightOneValue() { + pushDummies(1); + assertEquals(1, stack.pop()); + } } \ No newline at end of file