[BAEL-2749] Improves code and makes a copy in libraries-testing

2020-03-31 20:18:11 +02:00
parent 906d72859f
commit 0b1c5e01a5
25 changed files with 758 additions and 95 deletions
@@ -0,0 +1,136 @@
+package com.baeldung.dbunit;
+
+import org.dbunit.Assertion;
+import org.dbunit.DataSourceBasedDBTestCase;
+import org.dbunit.assertion.DiffCollectingFailureHandler;
+import org.dbunit.assertion.Difference;
+import org.dbunit.dataset.IDataSet;
+import org.dbunit.dataset.ITable;
+import org.dbunit.dataset.xml.FlatXmlDataSetBuilder;
+import org.dbunit.operation.DatabaseOperation;
+import org.h2.jdbcx.JdbcDataSource;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.platform.commons.logging.Logger;
+import org.junit.platform.commons.logging.LoggerFactory;
+
+import java.sql.Connection;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+
+import static com.baeldung.dbunit.ConnectionSettings.JDBC_URL;
+import static java.util.stream.Collectors.joining;
+import static org.assertj.core.api.Assertions.assertThat;
+
+public class DataSourceDBUnitTest extends DataSourceBasedDBTestCase {
+
+    private static Logger logger = LoggerFactory.getLogger(DataSourceDBUnitTest.class);
+
+    @Override
+    protected javax.sql.DataSource getDataSource() {
+        JdbcDataSource dataSource = new JdbcDataSource();
+        dataSource.setURL(JDBC_URL);
+        dataSource.setUser("sa");
+        dataSource.setPassword("");
+        return dataSource;
+    }
+
+    @Override
+    protected IDataSet getDataSet() throws Exception {
+        try (java.io.InputStream resourceAsStream = getClass()
+                .getClassLoader()
+                .getResourceAsStream("dbunit/data.xml")) {
+            return new FlatXmlDataSetBuilder().build(resourceAsStream);
+        }
+    }
+
+    @Override
+    protected DatabaseOperation getSetUpOperation() {
+        return DatabaseOperation.REFRESH;
+    }
+
+    @Override
+    protected DatabaseOperation getTearDownOperation() {
+        return DatabaseOperation.DELETE_ALL;
+    }
+
+    @Before
+    public void setUp() throws Exception {
+        super.setUp();
+    }
+
+    @After
+    public void tearDown() throws Exception {
+        super.tearDown();
+    }
+
+
+    @Test
+    public void testSimpleDataSet() throws SQLException {
+        final Connection connection = getDataSource()
+                .getConnection();
+
+        final ResultSet rs = connection
+                .createStatement()
+                .executeQuery("select * from iTEMS where id = 1");
+
+        assertThat(rs.next()).isTrue();
+        assertThat(rs.getString("title")).isEqualTo("Grey T-Shirt");
+    }
+
+    @Test
+    public void testEmptySchema() throws Exception {
+        final IDataSet expectedDataSet = getDataSet();
+        final ITable expectedTable = expectedDataSet.getTable("CLIENTS");
+        final IDataSet databaseDataSet = getConnection().createDataSet();
+        final ITable actualTable = databaseDataSet.getTable("CLIENTS");
+        Assertion.assertEquals(expectedTable, actualTable);
+    }
+
+    @Test
+    public void testAssertByQuery() throws Exception {
+        try (final java.io.InputStream is = getClass()
+                .getClassLoader()
+                .getResourceAsStream("dbunit/expected-user.xml")) {
+            final IDataSet expectedDataSet = new FlatXmlDataSetBuilder().build(is);
+            final ITable expectedTable = expectedDataSet.getTable("CLIENTS");
+            final Connection conn = getDataSource().getConnection();
+
+            conn.createStatement()
+                    .executeUpdate(
+                            "INSERT INTO CLIENTS (first_name, last_name) VALUES ('John', 'Jansen')");
+            final ITable actualData = getConnection()
+                    .createQueryTable(
+                            "result_name",
+                            "SELECT * FROM CLIENTS WHERE last_name='Jansen'");
+
+            Assertion.assertEqualsIgnoreCols(expectedTable, actualData, new String[]{"id"});
+        }
+    }
+
+    @Test
+    public void testMultipleFailures() throws Exception {
+        try (final java.io.InputStream is = getClass()
+                .getClassLoader()
+                .getResourceAsStream("dbunit/expected-multiple-failures.xml")) {
+            final IDataSet expectedDataSet = new FlatXmlDataSetBuilder().build(is);
+            final ITable expectedTable = expectedDataSet.getTable("ITEMS");
+            final Connection conn = getDataSource().getConnection();
+            final DiffCollectingFailureHandler collectingHandler = new DiffCollectingFailureHandler();
+
+            conn.createStatement().executeUpdate("INSERT INTO ITEMS (title, price) VALUES ('Battery', '1000000')");
+            final ITable actualData = getConnection().createDataSet().getTable("ITEMS");
+
+            Assertion.assertEquals(expectedTable, actualData, collectingHandler);
+            if (!collectingHandler.getDiffList().isEmpty()) {
+                String message = (String) collectingHandler.getDiffList().stream().map(d -> formatDifference((Difference) d)).collect(joining("\n"));
+                logger.error(() -> message);
+            }
+        }
+    }
+
+    private static String formatDifference(Difference diff) {
+        return "expected value in " + diff.getExpectedTable().getTableMetaData().getTableName() + "." + diff.getColumnName() + " row " + diff.getRowIndex() + ":" + diff.getExpectedValue() + ", but was: " + diff.getActualValue();
+    }
+}