Fix NPEs when running tests.

Note that there are still test failures, but these are assertions and not errors.
BentoBoxWorld · Apr 16, 2022 · 2263dae · 2263dae
1 parent f851a96
commit 2263dae
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/test/java/world/bentobox/challenges/tasks/TryToCompleteTest.java b/src/test/java/world/bentobox/challenges/tasks/TryToCompleteTest.java
@@ -21,6 +21,7 @@
 import java.util.UUID;
 
 import org.bukkit.Bukkit;
+import org.bukkit.ChatColor;
 import org.bukkit.GameMode;
 import org.bukkit.Location;
 import org.bukkit.Material;
@@ -76,7 +77,7 @@
  *
  */
 @RunWith(PowerMockRunner.class)
-@PrepareForTest({Bukkit.class, BentoBox.class, Util.class, Utils.class})
+@PrepareForTest({Bukkit.class, BentoBox.class, Util.class, Utils.class, ChatColor.class})
 public class TryToCompleteTest {
 
     // Constants
@@ -204,8 +205,9 @@ public void setUp() throws Exception {
         when(user.getPlayer()).thenReturn(player);
         UUID uniqueId = UUID.randomUUID();
         when(user.getUniqueId()).thenReturn(uniqueId);
-        when(user.getTranslation(Mockito.anyString())).thenAnswer((Answer<String>) invocation -> invocation.getArgument(0, String.class));
-        when(user.getTranslationOrNothing(Mockito.anyString())).thenAnswer((Answer<String>) invocation -> invocation.getArgument(0, String.class));
+        when(user.getTranslation(anyString())).thenAnswer((Answer<String>) invocation -> invocation.getArgument(0, String.class));
+        when(user.getTranslation(anyString(), anyString())).thenAnswer((Answer<String>) invocation -> invocation.getArgument(0, String.class));
+        when(user.getTranslationOrNothing(anyString())).thenAnswer((Answer<String>) invocation -> invocation.getArgument(0, String.class));
         when(user.getName()).thenReturn("tastybento");
         @Nullable
         Location userLoc = mock(Location.class);
@@ -256,6 +258,10 @@ public void setUp() throws Exception {
         // ItemFactory
         ItemFactory itemFactory = mock(ItemFactory.class);
         when(Bukkit.getItemFactory()).thenReturn(itemFactory);
+
+        // ChatColor
+        PowerMockito.mockStatic(ChatColor.class, Mockito.RETURNS_MOCKS);
+        when(ChatColor.stripColor(anyString())).thenAnswer((Answer<String>) invocation -> invocation.getArgument(0, String.class));
     }
 
     /**