Add error handling for negative indices

Asthenosphere · Oct 3, 2020 · 088e3b4 · 088e3b4
1 parent b05a3c0
commit 088e3b4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/main/java/duke/command/DeleteCommand.java b/src/main/java/duke/command/DeleteCommand.java
@@ -41,7 +41,7 @@ public DeleteCommand(int index) {
     @Override
     public String execute(TaskList tasks, Storage storage) throws DukeException {
         try {
-            if (index > tasks.getSize()) {
+            if (index > tasks.getSize() || index < 0) {
                 throw new InvalidArgumentException(ARGUMENT_EXCEPTION);
             }
             String output = DELETE_MESSAGE;

diff --git a/src/main/java/duke/command/DoneCommand.java b/src/main/java/duke/command/DoneCommand.java
@@ -36,7 +36,7 @@ public DoneCommand(int index) {
     @Override
     public String execute(TaskList tasks, Storage storage) {
         try {
-            if (index > tasks.getSize()) {
+            if (index > tasks.getSize() || index < 0) {
                 throw new InvalidArgumentException("☹ OOPS!!! The task index you give is not found.");
             }
             if (tasks.markDone(index)) {