package com.atlassian.activeobjects.confluence.hibernate;
   
   import com.atlassian.hibernate.PluginHibernateSessionFactory;
   import com.atlassian.sal.api.transaction.TransactionCallback;
   import com.atlassian.sal.api.transaction.TransactionTemplate;
   import net.sf.hibernate.Session;
   import net.sf.hibernate.SessionFactory;
   import net.sf.hibernate.dialect.Dialect;
   import net.sf.hibernate.engine.SessionFactoryImplementor;
  import org.junit.After;
  import org.junit.Before;
  import org.junit.Test;
  import org.junit.runner.RunWith;
  import org.mockito.Mock;
  import org.mockito.junit.MockitoJUnitRunner;
  
  import static org.junit.Assert.assertEquals;
  import static org.junit.Assert.assertNull;
  import static org.mockito.ArgumentMatchers.any;
  import static org.mockito.Mockito.mock;
  import static org.mockito.Mockito.when;
  
  /**
   * Testing {@link com.atlassian.activeobjects.confluence.hibernate.HibernateSessionDialectExtractor}
   */
  @RunWith(MockitoJUnitRunner.class)
  public class HibernateSessionDialectExtractorTest {
      private HibernateSessionDialectExtractor dialectExtractor;
  
      @Mock
      private PluginHibernateSessionFactory pluginSessionFactory;
  
      @Mock
      private TransactionTemplate transactionTemplate;
  
      @Before
      public void setUp() {
          dialectExtractor = new HibernateSessionDialectExtractor(pluginSessionFactory, transactionTemplate);
          when(transactionTemplate.execute(any())).thenAnswer(invocation ->
                  ((TransactionCallback<?>) invocation.getArgument(0)).doInTransaction());
      }
  
      @After
      public void tearDown() {
          dialectExtractor = null;
      }
  
      @Test
      public void testGetDialectReturnsNullIfSessionFactoryDoesNotImplementSessionFactoryImplementor() {
          mockPluginSessionFactory(SessionFactory.class);
          assertNull(dialectExtractor.getDialect());
      }
  
      @Test
      public void testGetDialectReturnsCorrectDialectIfSessionFactoryDoesImplementSessionFactoryImplementor() {
          final Dialect dialect = mock(Dialect.class);
          final SessionFactoryImplementor sessionFactory = mockPluginSessionFactory(SessionFactoryImplementor.class);
          when(sessionFactory.getDialect()).thenReturn(dialect);
  
          assertEquals(dialect.getClass(), dialectExtractor.getDialect());
      }
  
      private <S extends SessionFactory> S mockPluginSessionFactory(Class<S> sessionFactoryClass) {
          final Session session = mock(Session.class);
          final S sessionFactory = mock(sessionFactoryClass);
          when(pluginSessionFactory.getSession()).thenReturn(session);
          when(session.getSessionFactory()).thenReturn(sessionFactory);
          return sessionFactory;
      }
  }