A-JUnit

xyzhangg · Aug 26, 2020 · 856928b · 856928b
1 parent bb2c223
commit 856928b
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 7 deletions.
diff --git a/src/main/java/Storage.java b/src/main/java/Storage.java
@@ -17,6 +17,13 @@ public class Storage {
 
     ArrayList<Task> load() throws DukeException {
         ArrayList<Task> taskList = new ArrayList<>();
+
+        try {
+            new File(filePath).createNewFile();
+        } catch (IOException e) {
+            throw new DukeException("☹ OOPS!!! There was an error creating a save file.");
+        }
+
         try {
             Scanner fileReader = new Scanner(new File(filePath));
             while (fileReader.hasNextLine()) {

diff --git a/src/main/java/TaskList.java b/src/main/java/TaskList.java
@@ -46,11 +46,15 @@ public Task deleteTask(int taskIndex) throws DukeException {
 
     @Override
     public String toString() {
-        StringBuilder taskListString = new StringBuilder();
-        for (int i = 1; i < myTaskList.size() + 1; i++) {
-            taskListString.append("     ").append(i).append(".").append(myTaskList.get(i - 1)).append("\n");
+        if (myTaskList.isEmpty()) {
+            return "";
+        } else {
+            StringBuilder taskListString = new StringBuilder();
+            for (int i = 1; i < myTaskList.size() + 1; i++) {
+                taskListString.append("     ").append(i).append(".").append(myTaskList.get(i - 1)).append("\n");
+            }
+            taskListString.delete(taskListString.length() - 1, taskListString.length());
+            return taskListString.toString();
         }
-        taskListString.delete(taskListString.length() - 1, taskListString.length());
-        return taskListString.toString();
     }
 }
diff --git a/src/main/java/Ui.java b/src/main/java/Ui.java
@@ -42,8 +42,7 @@ void showGoodbye() {
     }
 
     void showTaskList(String taskListString) {
-        System.out.println("     Here are the tasks in your list:\n");
-        System.out.println(taskListString);
+        System.out.println("     Here are the tasks in your list:\n" +  taskListString);
     }
 
     void showDoneTask(Task task) {