diff --git a/abstracto-application/abstracto-modules/experience-tracking/experience-tracking-impl/src/test/java/dev/sheldan/abstracto/experience/listener/ExperienceTrackerListenerTest.java b/abstracto-application/abstracto-modules/experience-tracking/experience-tracking-impl/src/test/java/dev/sheldan/abstracto/experience/listener/ExperienceTrackerListenerTest.java index 1810d4490..de2785af4 100644 --- a/abstracto-application/abstracto-modules/experience-tracking/experience-tracking-impl/src/test/java/dev/sheldan/abstracto/experience/listener/ExperienceTrackerListenerTest.java +++ b/abstracto-application/abstracto-modules/experience-tracking/experience-tracking-impl/src/test/java/dev/sheldan/abstracto/experience/listener/ExperienceTrackerListenerTest.java @@ -5,6 +5,7 @@ import dev.sheldan.abstracto.core.models.listener.MessageReceivedModel; import dev.sheldan.abstracto.core.service.management.UserInServerManagementService; import dev.sheldan.abstracto.experience.service.AUserExperienceService; import net.dv8tion.jda.api.entities.Message; +import net.dv8tion.jda.api.entities.MessageChannel; import net.dv8tion.jda.api.entities.MessageType; import net.dv8tion.jda.api.entities.User; import org.junit.Test; @@ -43,10 +44,13 @@ public class ExperienceTrackerListenerTest { Message mockedMessage = Mockito.mock(Message.class); when(mockedMessage.isFromGuild()).thenReturn(true); when(mockedMessage.isWebhookMessage()).thenReturn(false); + MessageChannel channel = Mockito.mock(MessageChannel.class); MessageType type = MessageType.DEFAULT; when(mockedMessage.getType()).thenReturn(type); when(userInServerManagementService.loadOrCreateUser(SERVER_ID, USER_ID)).thenReturn(userInAServer); when(model.getMessage()).thenReturn(mockedMessage); + when(userExperienceService.experienceGainEnabledInChannel(channel)).thenReturn(true); + when(mockedMessage.getChannel()).thenReturn(channel); when(model.getServerId()).thenReturn(SERVER_ID); when(mockedMessage.getAuthor()).thenReturn(user); when(user.getIdLong()).thenReturn(USER_ID);