HibernatesTest.java 文件源码

@Test
public void testGetDialect() throws SQLException {
    DataSource mockDataSource = Mockito.mock(DataSource.class);
    Connection mockConnection = Mockito.mock(Connection.class);
    DatabaseMetaData mockMetaData = Mockito.mock(DatabaseMetaData.class);

    Mockito.when(mockDataSource.getConnection()).thenReturn(mockConnection);
    Mockito.when(mockConnection.getMetaData()).thenReturn(mockMetaData);

    Mockito.when(mockMetaData.getURL()).thenReturn("jdbc:h2:file:~/test;AUTO_SERVER=TRUE");
    String dialect = Hibernates.getDialect(mockDataSource);
    assertEquals(H2Dialect.class.getName(), dialect);

    Mockito.when(mockMetaData.getURL()).thenReturn("jdbc:mysql://localhost:3306/test");
    dialect = Hibernates.getDialect(mockDataSource);
    assertEquals(MySQL5InnoDBDialect.class.getName(), dialect);

    Mockito.when(mockMetaData.getURL()).thenReturn("jdbc:oracle:thin:@127.0.0.1:1521:XE");
    dialect = Hibernates.getDialect(mockDataSource);
    assertEquals(Oracle10gDialect.class.getName(), dialect);
}