Merge pull request #13 from vibes-863/add-Junit-tests

Create JUnit test case for TransactionList class
AY2324S2-CS2113-T15-2 · Mar 14, 2024 · bfe0515 · bfe0515
2 parents 4f6de35 + 4b2df36
commit bfe0515
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/src/main/java/financemanager/TransactionList.java b/src/main/java/financemanager/TransactionList.java
@@ -41,7 +41,7 @@ public void removeTransaction(String input){
         }
     }
 
-    private void addTransaction(Transaction t) {
+    void addTransaction(Transaction t) {
         transactions.add(t);
     }
 

diff --git a/src/test/java/financemanager/TransactionListTest.java b/src/test/java/financemanager/TransactionListTest.java
@@ -5,6 +5,7 @@
 import transactions.Transaction;
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertThrows;
 
 public class TransactionListTest {
 
@@ -32,4 +33,23 @@ public void processTransaction_addsTransaction() {
         assertEquals(testTransaction.getDate(), transactionList.getTransactions().get(0).getDate());
     }
 
+    public void removeTransaction_removesCorrectTransaction() {
+        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);
+
+        transactionList.removeTransaction("delete 1");
+
+        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"));
+    }
 }