Update JUnit tests

ronnnnnnnnn · Feb 23, 2024 · d7cb44d · d7cb44d
1 parent e32b0ae
commit d7cb44d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 9 deletions.
diff --git a/src/test/java/badgpt/util/ParserTest.java b/src/test/java/badgpt/util/ParserTest.java
@@ -26,20 +26,21 @@ public void parse_validCommandFormat_success() throws Exception {
         BadGpt bot = new BadGpt(new Gui());
         TaskList taskList = new TaskList(new TasksUi(out, new ByteArrayOutputStream()));
         taskList.store(new Task("return book"));
+        out.reset();
 
         Parser.parse("mark 1", bot, taskList);
-        assertEquals("____________________________________________________________\r\n" +
+        assertEquals("_____________________________________________________\r\n" +
                 "Nice! I've marked this task as done:\n" +
                 "[X] return book\r\n" +
-                "____________________________________________________________\r\n", out.toString());
+                "_____________________________________________________\r\n", out.toString());
 
         out.reset();
 
         Parser.parse("unmark 1", bot, taskList);
-        assertEquals("____________________________________________________________\r\n" +
+        assertEquals("_____________________________________________________\r\n" +
                 "wyd bro why undo\n" +
                 "[ ] return book\r\n" +
-                "____________________________________________________________\r\n", out.toString());
+                "_____________________________________________________\r\n", out.toString());
     }
 
     @Test
@@ -53,15 +54,15 @@ public void parse_invalidCommandFormat_notSuccess() {
             fail();
         } catch (BadException e) {
             assertEquals("Please type in the command as follows: mark taskNum\n" +
-                    "Example: mark 2", e.toString());
+                    "Example usage: mark 2", e.toString());
         }
 
         // Try unmark command without a number
         try {
             Parser.parse("unmark", bot, taskList);
         } catch (BadException e) {
             assertEquals("Please type in the command as follows: unmark taskNum\n" +
-                    "Example: unmark 2", e.toString());
+                    "Example usage: unmark 2", e.toString());
         }
     }
 }
diff --git a/src/test/java/badgpt/util/TaskListTest.java b/src/test/java/badgpt/util/TaskListTest.java
@@ -51,14 +51,15 @@ public void listTasks_nonEmpty_success() throws Exception {
 
         TaskList taskList = new TaskList(new TasksUi(out, new ByteArrayOutputStream()));
         taskList.store(new Task("return book"));
+        out.reset();
 
         taskList.listTasks();
 
         // Lists the newly added task
-        assertEquals("____________________________________________________________\r\n" +
+        assertEquals("_____________________________________________________\r\n" +
                 "Here are the tasks in your list:\r\n" +
-                "1. [ ] return book\r\n" +
-                "____________________________________________________________\r\n", out.toString());
+                "[ ] return book\r\n" +
+                "_____________________________________________________\r\n", out.toString());
     }
 
     @Test