Fix ThemeCommandParser

ruishanteo · Nov 4, 2023 · dd1d951 · dd1d951
1 parent cd69484
commit dd1d951
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/src/main/java/seedu/address/logic/parser/ThemeCommandParser.java b/src/main/java/seedu/address/logic/parser/ThemeCommandParser.java
@@ -23,7 +23,13 @@ public ThemeCommand parse(String args) throws ParseException {
                 String.format(MESSAGE_INVALID_COMMAND_FORMAT, ThemeCommand.MESSAGE_USAGE));
         }
 
-        String theme = trimmedArgs.split("\\s+")[0];
+        String[] splitArgs = trimmedArgs.split("\\s+");
+        if (splitArgs.length > 1) {
+            throw new ParseException(
+                String.format(MESSAGE_UNKNOWN_THEME, ThemeCommand.MESSAGE_USAGE));
+        }
+
+        String theme = splitArgs[0];
 
         switch(theme) {
 

diff --git a/src/test/java/seedu/address/logic/parser/ThemeCommandParserTest.java b/src/test/java/seedu/address/logic/parser/ThemeCommandParserTest.java
@@ -24,6 +24,12 @@ public void parse_unknownTheme_throwsParseException() {
             ThemeCommand.MESSAGE_USAGE));
     }
 
+    @Test
+    public void parse_twoThemes_throwsParseException() {
+        assertParseFailure(parser, "blue green", String.format(MESSAGE_UNKNOWN_THEME,
+            ThemeCommand.MESSAGE_USAGE));
+    }
+
     @Test
     public void parse_validArgs_returnsThemeCommand() {
         ThemeCommand expectedThemeCommand1 =