Add removeTransaction IndexOutOfBounds test case

AY2324S2-CS2113-T15-2 · Mar 14, 2024 · 325180d · 325180d
1 parent 91704cc
commit 325180d
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/src/test/java/financemanager/TransactionListTest.java b/src/test/java/financemanager/TransactionListTest.java
@@ -15,11 +15,11 @@ public class TransactionListTest {
     public void setUp() {
         transactionList = new TransactionList();
     }
-    
+
     @Test
     public void removeTransaction_removesCorrectTransaction() {
-        Transaction testTransaction1 = new Transaction("Test1", 100,"Category1", "14-03-2024");
-        Transaction testTransaction2 = new Transaction("Test2", 200,"Category2", "16-03-2024");
+        Transaction testTransaction1 = new Transaction("Test1", 100, "Category1", "14-03-2024");
+        Transaction testTransaction2 = new Transaction("Test2", 200, "Category2", "16-03-2024");
         transactionList.addTransaction(testTransaction1);
         transactionList.addTransaction(testTransaction2);
 
@@ -28,4 +28,12 @@ public void removeTransaction_removesCorrectTransaction() {
         assertEquals(1, transactionList.getTransactions().size());
         assertEquals(testTransaction2, transactionList.getTransactions().get(0));
     }
+
+    @Test
+    public void removeTransaction_withInvalidIndex_throwsIndexOutOfBoundsException() {
+        Transaction testTransaction = new Transaction("Test", 200, "Personal", "14-03-2024");
+        transactionList.addTransaction(testTransaction);
+
+        assertThrows(IndexOutOfBoundsException.class, () -> transactionList.removeTransaction("delete 2"));
+    }
 }