1 files changed, 52 insertions, 0 deletions
diff --git a/CST 126/UnitTests/UnitTests.cpp b/CST 126/UnitTests/UnitTests.cpp
index 27ad6af..4b17624 100644
--- a/CST 126/UnitTests/UnitTests.cpp
+++ b/CST 126/UnitTests/UnitTests.cpp
@@ -66,7 +66,59 @@ namespace LinkedListUnitTests
 			}
 		}
 
+		TEST_METHOD(RemoveFirstLinkedList)
+		{
+			//Arrange
+			SinglyLinkedList linkedList{};
+			ListNode* node = new ListNode{ 2, nullptr };
+			ListNode* node2 = new ListNode{ 1, nullptr };
+			ListNode* node3 = new ListNode{ 3, nullptr };
+
+			//Act
+			bool success = Append(&linkedList, node);
+			success = Prepend(&linkedList, node2);
+			success = Append(&linkedList, node3);
+			success = RemoveFirst(&linkedList);
+
+			ListNode* travel = linkedList._head;
+
+			//Assert
 
+			for (auto i = 2; i <= 3; i++)
+			{
+				Assert::AreEqual(i, travel->_data);
+				travel = travel->_next;
+			}
+		}
 
+
+		TEST_METHOD(RemoveLastLinkedList)
+		{
+			//Arrange
+			SinglyLinkedList linkedList{};
+			ListNode* node = new ListNode{ 1, nullptr };
+			ListNode* node2 = new ListNode{ 2, nullptr };
+			ListNode* node3 = new ListNode{ 3, nullptr };
+			ListNode* node4 = new ListNode{ 4, nullptr };
+			ListNode* node5 = new ListNode{ 5, nullptr };
+
+			//Act
+			bool success = Append(&linkedList, node);
+			success = Append(&linkedList, node2);
+			success = Append(&linkedList, node3);
+			success = Append(&linkedList, node4);
+			success = Append(&linkedList, node5);
+			success = RemoveLast(&linkedList);
+
+			ListNode* travel = linkedList._head;
+
+			//Assert
+
+			for (auto i = 1; i <= 4; i++)
+			{
+				Assert::AreEqual(i, travel->_data);
+				travel = travel->_next;
+			}
+		}
 	};
 }