package com.atlassian.activeobjects.bamboo;

import com.atlassian.activeobjects.bamboo.hibernate.DialectExtractor;
import com.atlassian.activeobjects.spi.DatabaseType;
import com.atlassian.bamboo.persistence3.PluginHibernateSessionFactory;
import net.sf.hibernate.dialect.DB2390Dialect;
import net.sf.hibernate.dialect.DB2400Dialect;
import net.sf.hibernate.dialect.DB2Dialect;
import net.sf.hibernate.dialect.Dialect;
import net.sf.hibernate.dialect.HSQLDialect;
import net.sf.hibernate.dialect.MySQLDialect;
import net.sf.hibernate.dialect.Oracle9Dialect;
import net.sf.hibernate.dialect.OracleDialect;
import net.sf.hibernate.dialect.PostgreSQLDialect;
import net.sf.hibernate.dialect.SQLServerIntlDialect;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.runners.MockitoJUnitRunner;

import static org.junit.Assert.assertEquals;
import static org.mockito.Mockito.doReturn;

/**
 * Testing {@link com.atlassian.activeobjects.bamboo.BambooDataSourceProvider}
 */
@RunWith(MockitoJUnitRunner.class)
public class BambooDataSourceProviderTest
{
    private BambooDataSourceProvider dataSourceProvider;

    @Mock
    private PluginHibernateSessionFactory sessionFactory;

    @Mock
    private DialectExtractor dialectExtractor;

    @Before
    public void setUp() throws Exception
    {
        dataSourceProvider = new BambooDataSourceProvider(sessionFactory, dialectExtractor);
    }

    @After
    public void tearDown() throws Exception
    {
        dataSourceProvider = null;
    }

    @Test
    public void testGetUnknownDatabaseTypeWithNullDialect() throws Exception
    {
        assertDatabaseTypeForDialect(DatabaseType.UNKNOWN, null);
    }

    @Test
    public void testGetHsqlDatabaseTypeWithHsqlDialect()
    {
        assertDatabaseTypeForDialect(DatabaseType.HSQL, HSQLDialect.class);
    }

    @Test
    public void testGetMySqlDatabaseTypeWithMySqlDialect()
    {
        assertDatabaseTypeForDialect(DatabaseType.MYSQL, com.atlassian.hibernate.dialect.MySQLDialect.class);
        assertDatabaseTypeForDialect(DatabaseType.MYSQL, MySQLDialect.class);
    }

    @Test
    public void testGetPostgresDatabaseTypeWithPostgresDialect()
    {
        assertDatabaseTypeForDialect(DatabaseType.POSTGRESQL, PostgreSQLDialect.class);
    }

    @Test
    public void testGetOracleDatabaseTypeWithOracleDialect()
    {
        assertDatabaseTypeForDialect(DatabaseType.ORACLE, OracleDialect.class);
        assertDatabaseTypeForDialect(DatabaseType.ORACLE, Oracle9Dialect.class);
    }

    @Test
    public void testGetMsSqlDatabaseTypeWithMsSqlDialect()
    {
        assertDatabaseTypeForDialect(DatabaseType.MS_SQL, SQLServerIntlDialect.class);
    }

    @Test
    public void testGetDB2DatabaseTypeWithDB2Dialect()
    {
        assertDatabaseTypeForDialect(DatabaseType.DB2, com.atlassian.hibernate.dialect.DB2Dialect.class);
        assertDatabaseTypeForDialect(DatabaseType.DB2, DB2Dialect.class);
        assertDatabaseTypeForDialect(DatabaseType.DB2, DB2390Dialect.class);
        assertDatabaseTypeForDialect(DatabaseType.DB2, DB2400Dialect.class);
    }

    private void assertDatabaseTypeForDialect(DatabaseType databaseType, Class<? extends Dialect> dialect)
    {
        doReturn(dialect).when(dialectExtractor).getDialect();
        assertEquals(databaseType, dataSourceProvider.getDatabaseType());
    }
}