added unit test for checkPlurality and IAE for negative values

Slimefun · Aug 30, 2023 · 5b2e959 · 5b2e959
1 parent 1581671
commit 5b2e959
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/src/main/java/io/github/thebusybiscuit/slimefun4/utils/ChatUtils.java b/src/main/java/io/github/thebusybiscuit/slimefun4/utils/ChatUtils.java
@@ -89,8 +89,13 @@ public static void awaitInput(@Nonnull Player p, @Nonnull Consumer<String> callb
      *      The amount of things
      * @return
      *      {@code string} if {@code count} is 1 else {@code string + "s"}
+     * @throws IllegalArgumentException
+     *      if count is less than 0
      */
     public static @Nonnull String checkPlurality(@Nonnull String string, int count) {
+        if (count < 0) {
+            throw new IllegalArgumentException("Argument count cannot be negative.");
+        }
         if (count == 1) {
             return string;
         }

diff --git a/src/test/java/io/github/thebusybiscuit/slimefun4/utils/TestChatUtils.java b/src/test/java/io/github/thebusybiscuit/slimefun4/utils/TestChatUtils.java
@@ -33,4 +33,14 @@ void testColorCodeRemoval() {
         Assertions.assertEquals(expected, ChatUtils.removeColorCodes(ChatColor.GREEN + "Hello " + ChatColor.RED + "world"));
     }
 
+    @Test
+    @DisplayName("Test ChatUtils.checkPlurality(...)")
+    void testPluralization() {
+        String input = "Banana";
+        Assertions.assertThrows(IllegalArgumentException.class, () -> ChatUtils.checkPlurality(input, -1));
+        Assertions.assertEquals("Bananas", ChatUtils.checkPlurality(input, 0));
+        Assertions.assertEquals("Banana", ChatUtils.checkPlurality(input, 1));
+        Assertions.assertEquals("Bananas", ChatUtils.checkPlurality(input, 2));
+    }
+
 }