diff --git a/src/main/java/fr/xephi/authme/cache/TempbanManager.java b/src/main/java/fr/xephi/authme/cache/TempbanManager.java index aa41c473..6372d257 100644 --- a/src/main/java/fr/xephi/authme/cache/TempbanManager.java +++ b/src/main/java/fr/xephi/authme/cache/TempbanManager.java @@ -34,8 +34,10 @@ public class TempbanManager implements SettingsDependent { private int length; @Inject - TempbanManager(NewSetting settings) { - playerCounts = new ConcurrentHashMap<>(); + TempbanManager(BukkitService bukkitService, Messages messages, NewSetting settings) { + this.playerCounts = new ConcurrentHashMap<>(); + this.bukkitService = bukkitService; + this.messages = messages; loadSettings(settings); } diff --git a/src/test/java/fr/xephi/authme/cache/TempbanManagerTest.java b/src/test/java/fr/xephi/authme/cache/TempbanManagerTest.java index ab6fd50d..bf02253b 100644 --- a/src/test/java/fr/xephi/authme/cache/TempbanManagerTest.java +++ b/src/test/java/fr/xephi/authme/cache/TempbanManagerTest.java @@ -1,9 +1,14 @@ package fr.xephi.authme.cache; import fr.xephi.authme.ReflectionTestUtils; +import fr.xephi.authme.output.Messages; import fr.xephi.authme.settings.NewSetting; import fr.xephi.authme.settings.properties.SecuritySettings; +import fr.xephi.authme.util.BukkitService; import org.junit.Test; +import org.junit.runner.RunWith; +import org.mockito.Mock; +import org.mockito.runners.MockitoJUnitRunner; import java.util.Map; @@ -15,13 +20,20 @@ import static org.mockito.Mockito.mock; /** * Test for {@link TempbanManager}. */ +@RunWith(MockitoJUnitRunner.class) public class TempbanManagerTest { + @Mock + BukkitService bukkitService; + + @Mock + Messages messages; + @Test public void shouldAddCounts() { // given NewSetting settings = mockSettings(3, 60); - TempbanManager manager = new TempbanManager(settings); + TempbanManager manager = new TempbanManager(bukkitService, messages, settings); String player = "Tester"; // when @@ -41,7 +53,7 @@ public class TempbanManagerTest { // given String player = "plaYah"; NewSetting settings = mockSettings(3, 60); - TempbanManager manager = new TempbanManager(settings); + TempbanManager manager = new TempbanManager(bukkitService, messages, settings); // when manager.increaseCount(player); @@ -66,7 +78,7 @@ public class TempbanManagerTest { String player = "playah"; NewSetting settings = mockSettings(1, 5); given(settings.getProperty(SecuritySettings.TEMPBAN_ON_MAX_LOGINS)).willReturn(false); - TempbanManager manager = new TempbanManager(settings); + TempbanManager manager = new TempbanManager(bukkitService, messages, settings); // when manager.increaseCount(player); @@ -81,7 +93,7 @@ public class TempbanManagerTest { // given String player = "playah"; NewSetting settings = mockSettings(1, 5); - TempbanManager manager = new TempbanManager(settings); + TempbanManager manager = new TempbanManager(bukkitService, messages, settings); given(settings.getProperty(SecuritySettings.TEMPBAN_ON_MAX_LOGINS)).willReturn(false); // when