[AB-270] fixing test

This commit is contained in:
Sheldan
2021-05-26 11:51:45 +02:00
parent 08e2a31f15
commit 04b3e073aa

View File

@@ -5,6 +5,7 @@ import dev.sheldan.abstracto.core.models.listener.MessageReceivedModel;
import dev.sheldan.abstracto.core.service.management.UserInServerManagementService; import dev.sheldan.abstracto.core.service.management.UserInServerManagementService;
import dev.sheldan.abstracto.experience.service.AUserExperienceService; import dev.sheldan.abstracto.experience.service.AUserExperienceService;
import net.dv8tion.jda.api.entities.Message; 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.MessageType;
import net.dv8tion.jda.api.entities.User; import net.dv8tion.jda.api.entities.User;
import org.junit.Test; import org.junit.Test;
@@ -43,10 +44,13 @@ public class ExperienceTrackerListenerTest {
Message mockedMessage = Mockito.mock(Message.class); Message mockedMessage = Mockito.mock(Message.class);
when(mockedMessage.isFromGuild()).thenReturn(true); when(mockedMessage.isFromGuild()).thenReturn(true);
when(mockedMessage.isWebhookMessage()).thenReturn(false); when(mockedMessage.isWebhookMessage()).thenReturn(false);
MessageChannel channel = Mockito.mock(MessageChannel.class);
MessageType type = MessageType.DEFAULT; MessageType type = MessageType.DEFAULT;
when(mockedMessage.getType()).thenReturn(type); when(mockedMessage.getType()).thenReturn(type);
when(userInServerManagementService.loadOrCreateUser(SERVER_ID, USER_ID)).thenReturn(userInAServer); when(userInServerManagementService.loadOrCreateUser(SERVER_ID, USER_ID)).thenReturn(userInAServer);
when(model.getMessage()).thenReturn(mockedMessage); when(model.getMessage()).thenReturn(mockedMessage);
when(userExperienceService.experienceGainEnabledInChannel(channel)).thenReturn(true);
when(mockedMessage.getChannel()).thenReturn(channel);
when(model.getServerId()).thenReturn(SERVER_ID); when(model.getServerId()).thenReturn(SERVER_ID);
when(mockedMessage.getAuthor()).thenReturn(user); when(mockedMessage.getAuthor()).thenReturn(user);
when(user.getIdLong()).thenReturn(USER_ID); when(user.getIdLong()).thenReturn(USER_ID);