AK: Implement AK::Stack

2 files changed, 46 insertions, 0 deletions

diff --git a/Tests/AK/CMakeLists.txt b/Tests/AK/CMakeLists.txt
index b74893f0f8..5f2b371495 100644
--- a/Tests/AK/CMakeLists.txt
+++ b/Tests/AK/CMakeLists.txt
@@ -47,6 +47,7 @@ set(AK_TEST_SOURCES
     TestSourceGenerator.cpp
     TestSourceLocation.cpp
     TestSpan.cpp
+    TestStack.cpp
     TestString.cpp
     TestStringUtils.cpp
     TestStringView.cpp
diff --git a/Tests/AK/TestStack.cpp b/Tests/AK/TestStack.cpp
new file mode 100644
index 0000000000..ddbe2765d7
--- /dev/null
+++ b/Tests/AK/TestStack.cpp
@@ -0,0 +1,45 @@
+/*
+ * Copyright (c) 2021, Jesse Buhagiar <jooster669@gmail.com>
+ *
+ * SPDX-License-Identifier: BSD-2-Clause
+ */
+
+#include <LibTest/TestCase.h>
+
+#include <AK/Stack.h>
+
+TEST_CASE(basic)
+{
+    AK::Stack<int, 3> stack;
+
+    EXPECT(stack.is_empty() == true);
+    stack.push(2);
+    stack.push(4);
+    stack.push(17);
+    EXPECT(stack.size() == 3);
+    EXPECT(stack.top() == 17);
+    EXPECT_EQ(stack.pop(), true);
+    EXPECT_EQ(stack.pop(), true);
+    EXPECT_EQ(stack.pop(), true);
+    EXPECT(stack.is_empty());
+}
+
+TEST_CASE(complex_type)
+{
+    AK::Stack<String, 4> stack;
+
+    EXPECT_EQ(stack.is_empty(), true);
+    EXPECT(stack.push("Well"));
+    EXPECT(stack.push("Hello"));
+    EXPECT(stack.push("Friends"));
+    EXPECT(stack.push(":^)"));
+    EXPECT_EQ(stack.top(), ":^)");
+    EXPECT_EQ(stack.pop(), true);
+    EXPECT_EQ(stack.top(), "Friends");
+    EXPECT_EQ(stack.pop(), true);
+    EXPECT_EQ(stack.top(), "Hello");
+    EXPECT_EQ(stack.pop(), true);
+    EXPECT_EQ(stack.top(), "Well");
+    EXPECT_EQ(stack.pop(), true);
+    EXPECT(stack.is_empty());
+}