LoginSystem/src/test/java/fr/xephi/authme/converter/AbstractDataSourceConverterTest.java

package fr.xephi.authme.converter;

import fr.xephi.authme.TestHelper;
import fr.xephi.authme.cache.auth.PlayerAuth;
import fr.xephi.authme.datasource.DataSource;
import fr.xephi.authme.datasource.DataSourceType;
import org.bukkit.command.CommandSender;
import org.junit.BeforeClass;
import org.junit.Test;
import org.mockito.Mockito;

import java.util.Arrays;
import java.util.List;

import static org.hamcrest.Matchers.containsString;
import static org.mockito.BDDMockito.given;
import static org.mockito.Matchers.any;
import static org.mockito.Matchers.anyString;
import static org.mockito.Matchers.argThat;
import static org.mockito.Mockito.doThrow;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.only;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.verifyNoMoreInteractions;
import static org.mockito.Mockito.verifyZeroInteractions;

/**
 * Test for {@link AbstractDataSourceConverter}.
 */
public class AbstractDataSourceConverterTest {

    @BeforeClass
    public static void initLogger() {
        TestHelper.setupLogger();
    }

    @Test
    public void shouldThrowForDestinationTypeMismatch() {
        // given
        DataSource destination = mock(DataSource.class);
        given(destination.getType()).willReturn(DataSourceType.MYSQL);
        DataSourceType destinationType = DataSourceType.SQLITE;
        DataSource source = mock(DataSource.class);
        Converter converter = new DataSourceConverterTestImpl<>(source, destination, destinationType);
        CommandSender sender = mock(CommandSender.class);

        // when
        converter.execute(sender);

        // then
        verify(sender).sendMessage(argThat(containsString("Please configure your connection to SQLITE")));
        verify(destination, only()).getType();
        verifyZeroInteractions(source);
    }

    @Test
    public void shouldHandleSourceThrowingException() {
        // given
        DataSource source = mock(DataSource.class);
        DataSource destination = mock(DataSource.class);
        DataSourceType destinationType = DataSourceType.SQLITE;
        given(destination.getType()).willReturn(destinationType);
        DataSourceConverterTestImpl converter =
            Mockito.spy(new DataSourceConverterTestImpl<>(source, destination, destinationType));
        doThrow(IllegalStateException.class).when(converter).getSource();
        CommandSender sender = mock(CommandSender.class);

        // when
        converter.execute(sender);

        // then
        verify(sender).sendMessage("The data source to convert from could not be initialized");
        verify(destination, only()).getType();
        verifyZeroInteractions(source);
    }

    @Test
    public void shouldConvertAndSkipExistingPlayers() {
        // given
        DataSource source = mock(DataSource.class);
        DataSource destination = mock(DataSource.class);
        DataSourceType destinationType = DataSourceType.MYSQL;
        given(destination.getType()).willReturn(destinationType);

        List<PlayerAuth> auths =
            Arrays.asList(mockAuthWithName("Steven"), mockAuthWithName("bobby"), mockAuthWithName("Jack"));
        given(source.getAllAuths()).willReturn(auths);
        given(destination.isAuthAvailable(auths.get(0).getNickname())).willReturn(true);

        Converter converter = new DataSourceConverterTestImpl<>(source, destination, destinationType);
        CommandSender sender = mock(CommandSender.class);

        // when
        converter.execute(sender);

        // then
        verify(destination).getType();
        verify(destination, times(3)).isAuthAvailable(anyString());
        verify(destination, times(2)).saveAuth(any(PlayerAuth.class));
        verify(destination, times(2)).updateQuitLoc(any(PlayerAuth.class));
        verifyNoMoreInteractions(destination);
        verify(sender).sendMessage(argThat(containsString(auths.get(0).getNickname())));
        verify(sender).sendMessage(argThat(containsString("successfully converted")));
    }

    private static PlayerAuth mockAuthWithName(String name) {
        PlayerAuth auth = mock(PlayerAuth.class);
        given(auth.getNickname()).willReturn(name);
        return auth;
    }

    private static class DataSourceConverterTestImpl<S extends DataSource> extends AbstractDataSourceConverter<S> {
        private final S source;

        DataSourceConverterTestImpl(S source, DataSource destination, DataSourceType destinationType) {
            super(destination, destinationType);
            this.source = source;
        }

        @Override
        protected S getSource() {
          return source;
        }
    }
}