Merge pull request #27 from annnniexu/master

Added 1 test
AY2324S2-CS2113-T12-4 · Mar 15, 2024 · 1054e69 · 1054e69
2 parents 1c54ca9 + 65fcad0
commit 1054e69
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 2 deletions.
diff --git a/src/main/java/seedu/duke/TravelActivityList.java b/src/main/java/seedu/duke/TravelActivityList.java
@@ -17,6 +17,7 @@ public TravelActivityList() {
      */
     public void addTask(TravelActivity travelActivity){
         travelActivities.add(travelActivity);
+        noOfTasks += 1;
     }
 
     /**
@@ -52,15 +53,14 @@ public void removeTask(int taskNumber){
         travelActivities.remove(indexOfTask);
         System.out.println("I have removed this task:");
         System.out.println(removedTask);
+        noOfTasks -=1;
 
     }
 
     public String getDescription(String plan){
         for(TravelActivity travelActivity: travelActivities){
             if(travelActivity.getPlan().equals(plan)){
                 return travelActivity.getPlan();
-            } else {
-                break;
             }
         }
         return "cant be found";

diff --git a/src/test/java/seedu/duke/DukeTest.java b/src/test/java/seedu/duke/DukeTest.java
@@ -25,6 +25,29 @@ public void deleteTest() {
         assertEquals("cant be found", travelActivityList.getDescription("visit museum"));
     }
 
+    @Test
+    public void getNoTasksTest() {
+        //add the first plan
+        TravelActivityList travelActivityList = new TravelActivityList();
+        travelActivityList.addTask(new TravelActivity("visit museum"));
+        assertEquals("visit museum", travelActivityList.getDescription("visit museum"));
+        //add the second and third plan
+        travelActivityList.addTask(new TravelActivity("go to beach"));
+        assertEquals("go to beach", travelActivityList.getDescription("go to beach"));
+        travelActivityList.addTask(new TravelActivity("shopping"));
+        assertEquals("shopping", travelActivityList.getDescription("shopping"));
+        //check number of tasks
+        assertEquals(3, travelActivityList.getNoOfTasks());
+        //delete the first plan
+        travelActivityList.removeTask(1);
+        assertEquals("cant be found", travelActivityList.getDescription("visit museum"));
+        //check number of tasks
+        assertEquals(2, travelActivityList.getNoOfTasks());
+
+
+
+    }
+
     @Test
     public void trueTest(){
         assertTrue(true);