package com.atlassian.activeobjects.internal;
   
   import com.atlassian.activeobjects.config.ActiveObjectsConfiguration;
   import com.atlassian.activeobjects.spi.DatabaseType;
   import net.java.ao.DatabaseProvider;
   import net.java.ao.SchemaConfiguration;
   import net.java.ao.schema.NameConverters;
   import net.java.ao.schema.TableNameConverter;
   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 javax.sql.DataSource;
  
  import static org.junit.Assert.assertNotNull;
  import static org.mockito.Mockito.mock;
  import static org.mockito.Mockito.verify;
  import static org.mockito.Mockito.when;
  
  /**
   * Testing {@link com.atlassian.activeobjects.internal.EntityManagerFactoryImpl}
   */
  @RunWith(MockitoJUnitRunner.class)
  public final class EntityManagerFactoryImplTest {
      private EntityManagerFactory entityManagerFactory;
  
      @Mock
      private DatabaseProviderFactory databaseProviderFactory;
  
      @Before
      public void setUp() {
          entityManagerFactory = new EntityManagerFactoryImpl(databaseProviderFactory);
      }
  
      @After
      public void tearDown() {
          entityManagerFactory = null;
          databaseProviderFactory = null;
      }
  
      @Test
      public void testGetEntityManager() {
          final String schema = null;
          final DataSource dataSource = mock(DataSource.class);
          final DatabaseType databaseType = DatabaseType.UNKNOWN;
          final DatabaseProvider databaseProvider = mock(DatabaseProvider.class);
  
          final ActiveObjectsConfiguration configuration = getMockConfiguration();
  
          when(databaseProviderFactory.getDatabaseProvider(dataSource, databaseType, schema)).thenReturn(databaseProvider);
          assertNotNull(entityManagerFactory.getEntityManager(dataSource, databaseType, schema, configuration));
  
          verify(databaseProviderFactory).getDatabaseProvider(dataSource, databaseType, schema);
      }
  
      private ActiveObjectsConfiguration getMockConfiguration() {
          final NameConverters nameConverters = mock(NameConverters.class);
          final TableNameConverter tableNameConverter = mock(TableNameConverter.class);
          final SchemaConfiguration schemaConfiguration = mock(SchemaConfiguration.class);
  
          final ActiveObjectsConfiguration configuration = mock(ActiveObjectsConfiguration.class);
          when(configuration.getNameConverters()).thenReturn(nameConverters);
          when(nameConverters.getTableNameConverter()).thenReturn(tableNameConverter);
          when(configuration.getSchemaConfiguration()).thenReturn(schemaConfiguration);
          return configuration;
      }
  }