Fix tests (due to new Bukkit/GeoIP updates)

AuthMe · Apr 27, 2024 · e133a37 · e133a37
1 parent 6eec2c8
commit e133a37
Show file tree

Hide file tree

Showing 6 changed files with 32 additions and 16 deletions.
diff --git a/src/main/java/fr/xephi/authme/process/join/AsynchronousJoin.java b/src/main/java/fr/xephi/authme/process/join/AsynchronousJoin.java
@@ -197,7 +197,7 @@ private void processJoinSync(Player player, boolean isAuthAvailable) {
             if (service.getProperty(RegistrationSettings.APPLY_BLIND_EFFECT)) {
                 // Allow infinite blindness effect
                 int blindTimeOut = (registrationTimeout <= 0) ? 99999 : registrationTimeout;
-                player.addPotionEffect(new PotionEffect(PotionEffectType.BLINDNESS, blindTimeOut, 2));
+                player.addPotionEffect(bukkitService.createBlindnessEffect(blindTimeOut));
             }
             commandManager.runCommandsOnJoin(player);
         });

diff --git a/src/main/java/fr/xephi/authme/process/logout/ProcessSyncPlayerLogout.java b/src/main/java/fr/xephi/authme/process/logout/ProcessSyncPlayerLogout.java
@@ -75,7 +75,7 @@ private void applyLogoutEffect(Player player) {
         // Apply Blindness effect
         if (service.getProperty(RegistrationSettings.APPLY_BLIND_EFFECT)) {
             int timeout = service.getProperty(RestrictionSettings.TIMEOUT) * TICKS_PER_SECOND;
-            player.addPotionEffect(new PotionEffect(PotionEffectType.BLINDNESS, timeout, 2));
+            player.addPotionEffect(bukkitService.createBlindnessEffect(timeout));
         }
 
         // Set player's data to unauthenticated

diff --git a/src/main/java/fr/xephi/authme/process/unregister/AsynchronousUnregister.java b/src/main/java/fr/xephi/authme/process/unregister/AsynchronousUnregister.java
@@ -144,7 +144,7 @@ private void performPostUnregisterActions(String name, Player player) {
     private void applyBlindEffect(Player player) {
         if (service.getProperty(RegistrationSettings.APPLY_BLIND_EFFECT)) {
             int timeout = service.getProperty(RestrictionSettings.TIMEOUT) * TICKS_PER_SECOND;
-            player.addPotionEffect(new PotionEffect(PotionEffectType.BLINDNESS, timeout, 2));
+            player.addPotionEffect(bukkitService.createBlindnessEffect(timeout));
         }
     }
 

diff --git a/src/main/java/fr/xephi/authme/service/BukkitService.java b/src/main/java/fr/xephi/authme/service/BukkitService.java
@@ -13,6 +13,8 @@
 import org.bukkit.configuration.file.YamlConfiguration;
 import org.bukkit.entity.Player;
 import org.bukkit.event.Event;
+import org.bukkit.potion.PotionEffect;
+import org.bukkit.potion.PotionEffectType;
 import org.bukkit.scheduler.BukkitRunnable;
 import org.bukkit.scheduler.BukkitTask;
 
@@ -338,6 +340,16 @@ public Optional<Boolean> isBungeeCordConfiguredForSpigot() {
         }
     }
 
+    /**
+     * Creates a PotionEffect with blindness for the given duration in ticks.
+     *
+     * @param timeoutInTicks duration of the effect in ticks
+     * @return blindness potion effect
+     */
+    public PotionEffect createBlindnessEffect(int timeoutInTicks) {
+        return new PotionEffect(PotionEffectType.BLINDNESS, timeoutInTicks, 2);
+    }
+
     /**
      * @return the IP string that this server is bound to, otherwise empty string
      */

diff --git a/src/test/java/fr/xephi/authme/process/unregister/AsynchronousUnregisterTest.java b/src/test/java/fr/xephi/authme/process/unregister/AsynchronousUnregisterTest.java
@@ -115,6 +115,7 @@ public void shouldPerformUnregister() {
         given(service.getProperty(RegistrationSettings.APPLY_BLIND_EFFECT)).willReturn(true);
         given(service.getProperty(RestrictionSettings.TIMEOUT)).willReturn(21);
         setBukkitServiceToScheduleSyncTaskFromOptionallyAsyncTask(bukkitService);
+        given(bukkitService.createBlindnessEffect(21 * 20)).willReturn(mock(PotionEffect.class));
 
         // when
         asynchronousUnregister.unregister(player, userPassword);
@@ -127,7 +128,6 @@ public void shouldPerformUnregister() {
         verify(teleportationService).teleportOnJoin(player);
         verifyCalledUnregisterEventFor(player);
         verify(commandManager).runCommandsOnUnregister(player);
-        verify(player).addPotionEffect(new PotionEffect(PotionEffectType.BLINDNESS, 21 * 20, 2));
     }
 
     @Test

diff --git a/src/test/java/fr/xephi/authme/service/GeoIpServiceTest.java b/src/test/java/fr/xephi/authme/service/GeoIpServiceTest.java
@@ -3,9 +3,14 @@
 import java.io.File;
 import java.io.IOException;
 import java.net.InetAddress;
+import java.util.Collections;
+import java.util.List;
+import java.util.Map;
 
+import com.google.common.collect.ImmutableMap;
 import com.maxmind.geoip2.DatabaseReader;
 import com.maxmind.geoip2.model.CountryResponse;
+import com.maxmind.geoip2.record.Continent;
 import com.maxmind.geoip2.record.Country;
 import fr.xephi.authme.settings.Settings;
 import fr.xephi.authme.settings.properties.ProtectionSettings;
@@ -58,12 +63,7 @@ public void shouldGetCountry() throws Exception {
         // given
         InetAddress ip = InetAddress.getByName("123.45.67.89");
         String countryCode = "XX";
-
-        Country country = mock(Country.class);
-        given(country.getIsoCode()).willReturn(countryCode);
-
-        CountryResponse response = mock(CountryResponse.class);
-        given(response.getCountry()).willReturn(country);
+        CountryResponse response = createCountryResponse(countryCode, "Unknown");
         given(lookupService.country(ip)).willReturn(response);
         given(settings.getProperty(ProtectionSettings.ENABLE_GEOIP)).willReturn(true);
 
@@ -93,12 +93,7 @@ public void shouldLookUpCountryName() throws Exception {
         // given
         InetAddress ip = InetAddress.getByName("24.45.167.89");
         String countryName = "Ecuador";
-
-        Country country = mock(Country.class);
-        given(country.getName()).willReturn(countryName);
-
-        CountryResponse response = mock(CountryResponse.class);
-        given(response.getCountry()).willReturn(country);
+        CountryResponse response = createCountryResponse("EC", countryName);
         given(lookupService.country(ip)).willReturn(response);
         given(settings.getProperty(ProtectionSettings.ENABLE_GEOIP)).willReturn(true);
 
@@ -136,4 +131,13 @@ public void shouldNotLookUpCountryNameIfDisabled() throws Exception {
         assertThat(result, equalTo("N/A"));
         verifyNoInteractions(lookupService);
     }
+
+    private static CountryResponse createCountryResponse(String countryCode, String countryName) {
+        List<String> locales = Collections.singletonList("en");
+        Continent continent = new Continent(locales, "XX", 1L, Collections.emptyMap());
+
+        Map<String, String> countryNames = ImmutableMap.of("en", countryName);
+        Country country = new Country(locales, 100, 3L, false, countryCode, countryNames);
+        return new CountryResponse(continent, country, null, country, null, null);
+    }
 }