Merge pull request #58 from ChenKangg/master

add JUnit test for tagging and untagging
AY2324S2-CS2113-T12-4 · Mar 20, 2024 · a2d5a93 · a2d5a93
2 parents 91d336f + 38bbbac
commit a2d5a93
Showing 1 changed file with 24 additions and 3 deletions.
diff --git a/src/test/java/seedu/duke/DukeTest.java b/src/test/java/seedu/duke/DukeTest.java
@@ -78,7 +78,7 @@ public void trueTest(){
         assertTrue(true);
     }
     @Test
-    public void testStringCoversion(){
+    public void testStringConversion(){
         assertEquals("visit museum", new TravelActivity("visit museum").toString());
     }
     @Test
@@ -88,8 +88,29 @@ public void searchTest () {
         assertTrue(true);
     }
 
-
-
+    @Test
+    public void testTagActivity() throws OmniException {
+        TravelActivityList list = new TravelActivityList();
+        list.addTravelActivity(new TravelActivity("visit museum"));
+        assertEquals("visit museum", list.getDescription("visit museum"));
+        // Tagging an existing task
+        list.tagActivity(1, "activity 1");
+        TravelActivity travelActivity = list.getTravelActivities().get(0);
+        assertEquals("activity 1", travelActivity.getTag());
+    }
 
+    @Test
+    public void testRemoveTagFromActivity() throws OmniException {
+        TravelActivityList list = new TravelActivityList();
+        list.addTravelActivity(new TravelActivity("visit museum"));
+        assertEquals("visit museum", list.getDescription("visit museum"));
+        // Tagging an existing task
+        list.tagActivity(1, "activity 1");
+        TravelActivity travelActivity = list.getTravelActivities().get(0);
+        assertEquals("activity 1", travelActivity.getTag());
+        // Remove an existing tag
+        list.removeTag(1);
+        assertEquals("visit museum", list.getDescription("visit museum"));
+    }
 
 }