Example usage for org.springframework.jdbc.datasource.embedded EmbeddedDatabase getConnection

Introduction

In this page you can find the example usage for org.springframework.jdbc.datasource.embedded EmbeddedDatabase getConnection.

Prototype

Connection getConnection() throws SQLException;

Source Link

Document

Attempts to establish a connection with the data source that this DataSource object represents.

Usage

From source file:com.jaxio.celerio.configuration.database.support.MetaDataExtractorTest.java

@Test
public void minimal() throws ClassNotFoundException, SQLException, JAXBException, IOException {
    EmbeddedDatabase embeddedDatabase = createMinimalEmbeddedDatabase(MINIMAL_SCRIPT);
    Metadata meta = extractor.extract(embeddedDatabase.getConnection());

    assertThat(countTables(meta)).isEqualTo(6);
    assertThat(countColumns(meta)).isEqualTo(29);
    assertThat(countPrimaryKeys(meta)).isEqualTo(9);
    assertThat(countImportedKeys(meta)).isEqualTo(4);
    assertThat(countIndexes(meta)).isEqualTo(13);
    assertThat(countEnums(meta)).isZero();

    embeddedDatabase.shutdown();//  ww  w.  ja  va 2s  .  c  o  m
}

---

From source file:com.jaxio.celerio.configuration.database.support.MetaDataExtractorTest.java

@Test
public void noTable() throws ClassNotFoundException, SQLException, JAXBException, IOException {
    EmbeddedDatabaseBuilder builder = new EmbeddedDatabaseBuilder();
    EmbeddedDatabase embeddedDatabase = builder.setType(EmbeddedDatabaseType.H2).build();

    Metadata meta = extractor.extract(embeddedDatabase.getConnection());

    assertThat(countTables(meta)).isZero();
    assertThat(countColumns(meta)).isZero();
    assertThat(countPrimaryKeys(meta)).isZero();
    assertThat(countImportedKeys(meta)).isZero();
    assertThat(countIndexes(meta)).isZero();
    assertThat(countEnums(meta)).isZero();

    embeddedDatabase.shutdown();/*from ww w. j a  v a2s  .  com*/
}

---

From source file:com.jaxio.celerio.configuration.database.support.MetaDataExtractorTest.java

@Test
public void outputIsSameAsInput()
        throws ClassNotFoundException, SQLException, XmlMappingException, IOException {
    EmbeddedDatabase embeddedDatabase = createMinimalEmbeddedDatabase(ALL_RELATIONS_SCRIPT);

    Metadata meta = extractor.extract(embeddedDatabase.getConnection());

    assertThat(countTables(meta)).isEqualTo(18);
    assertThat(countColumns(meta)).isEqualTo(49);
    assertThat(countPrimaryKeys(meta)).isEqualTo(18);
    assertThat(countIndexes(meta)).isEqualTo(21);
    assertThat(countEnums(meta)).isEqualTo(3);

    File tempFile = File.createTempFile(getClass().getName(), ".xml");
    tempFile.deleteOnExit();//from   w  w  w  . j a  v  a  2 s .  c o  m
    loader.write(meta, tempFile);

    Metadata loadedMeta = loader.load(tempFile);
    assertThat(countTables(meta)).isEqualTo(countTables(loadedMeta));
    assertThat(countColumns(meta)).isEqualTo(countColumns(loadedMeta));
    assertThat(countPrimaryKeys(meta)).isEqualTo(countPrimaryKeys(loadedMeta));
    assertThat(countIndexes(meta)).isEqualTo(countIndexes(loadedMeta));
    assertThat(countEnums(meta)).isEqualTo(countEnums(loadedMeta));

    embeddedDatabase.shutdown();
}

---

From source file:com.jaxio.celerio.main.CelerioProducerTest.java

private Metadata getMetaDataFromH2Script(String sqlResource) throws ClassNotFoundException, SQLException {
    EmbeddedDatabase embeddedDatabase = new EmbeddedDatabaseBuilder().setType(EmbeddedDatabaseType.H2)
            .addScript(sqlResource).build();
    return extractor.extract(embeddedDatabase.getConnection());
}

---

From source file:com.github.mjeanroy.dbunit.integration.spring.DbUnitEmbeddedDatabaseRuleTest.java

@SuppressWarnings({ "unchecked", "rawtypes" })
@Test//from   w ww  .j a  v a 2  s . com
public void it_should_start_database_and_load_data_set() throws Throwable {
    final Statement statement = mock(Statement.class);
    final Class testClass = TestClassWithDataSet.class;
    final Description description = mock(Description.class);
    when(description.getTestClass()).thenReturn(testClass);
    when(description.getMethodName()).thenReturn("method1");

    final EmbeddedDatabase db = spy(new EmbeddedDatabaseBuilder().setType(EmbeddedDatabaseType.HSQL)
            .addScript("classpath:/sql/init.sql").build());

    final DbUnitEmbeddedDatabaseRule rule = new DbUnitEmbeddedDatabaseRule(db);

    Statement result = rule.apply(statement, description);

    assertThat(result).isNotNull();
    verify(statement, never()).evaluate();
    verify(db, never()).shutdown();

    doAnswer(new Answer() {
        @Override
        public Void answer(InvocationOnMock invocationOnMock) throws Throwable {
            assertThat(countFrom(db.getConnection(), "foo")).isEqualTo(2);
            assertThat(countFrom(db.getConnection(), "bar")).isEqualTo(3);
            return null;
        }
    }).when(statement).evaluate();

    result.evaluate();

    InOrder inOrder = inOrder(db, statement);
    inOrder.verify(statement).evaluate();
    inOrder.verify(db).shutdown();
}

---