From d5d1cc7d580b1749b9f29df644ea34cbddeee8ac Mon Sep 17 00:00:00 2001
From: Joao Esperancinha <joao.esperancinha@eonics.nl>
Date: Tue, 31 Mar 2020 12:02:24 +0200
Subject: [PATCH] [BAEL-2749] Adds DataSourceBaseDBTestCase

---
 .../baeldung/dbunit/DataSourceDBUnitTest.java | 46 +++++++++++++++++++
 ...{SampleDbUnitTest.java => DbUnitTest.java} | 12 ++---
 2 files changed, 52 insertions(+), 6 deletions(-)
 create mode 100644 dbunit/src/test/java/com/baeldung/dbunit/DataSourceDBUnitTest.java
 rename dbunit/src/test/java/com/baeldung/dbunit/{SampleDbUnitTest.java => DbUnitTest.java} (85%)

diff --git a/dbunit/src/test/java/com/baeldung/dbunit/DataSourceDBUnitTest.java b/dbunit/src/test/java/com/baeldung/dbunit/DataSourceDBUnitTest.java
new file mode 100644
index 0000000000..154d93e0af
--- /dev/null
+++ b/dbunit/src/test/java/com/baeldung/dbunit/DataSourceDBUnitTest.java
@@ -0,0 +1,46 @@
+package com.baeldung.dbunit;
+
+import org.dbunit.DataSourceBasedDBTestCase;
+import org.dbunit.dataset.IDataSet;
+import org.dbunit.dataset.xml.FlatXmlDataSetBuilder;
+import org.h2.jdbcx.JdbcDataSource;
+import org.junit.Test;
+
+import javax.sql.DataSource;
+import java.sql.Connection;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+public class DataSourceDBUnitTest extends DataSourceBasedDBTestCase {
+    @Override
+    protected DataSource getDataSource() {
+        JdbcDataSource dataSource = new JdbcDataSource();
+        dataSource.setURL(
+                "jdbc:h2:mem:default;DB_CLOSE_DELAY=-1;init=runscript from 'classpath:schema.sql'");
+        dataSource.setUser("sa");
+        dataSource.setPassword("sa");
+        return dataSource;
+    }
+
+    @Override
+    protected IDataSet getDataSet() throws Exception {
+        return new FlatXmlDataSetBuilder().build(getClass()
+                .getClassLoader()
+                .getResourceAsStream("data.xml"));
+    }
+
+    @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");
+    }
+}
diff --git a/dbunit/src/test/java/com/baeldung/dbunit/SampleDbUnitTest.java b/dbunit/src/test/java/com/baeldung/dbunit/DbUnitTest.java
similarity index 85%
rename from dbunit/src/test/java/com/baeldung/dbunit/SampleDbUnitTest.java
rename to dbunit/src/test/java/com/baeldung/dbunit/DbUnitTest.java
index f7970754e7..8d6a94279a 100644
--- a/dbunit/src/test/java/com/baeldung/dbunit/SampleDbUnitTest.java
+++ b/dbunit/src/test/java/com/baeldung/dbunit/DbUnitTest.java
@@ -19,9 +19,9 @@ import static com.baeldung.dbunit.ConnectionSettings.PASSWORD;
 import static com.baeldung.dbunit.ConnectionSettings.USER;
 import static org.assertj.core.api.Assertions.assertThat;
 
-public class SampleDbUnitTest extends DBTestCase {
+public class DbUnitTest extends DBTestCase {
 
-    public SampleDbUnitTest(String name) {
+    public DbUnitTest(String name) {
         super(name);
         System.setProperty(PropertiesBasedJdbcDatabaseTester.DBUNIT_DRIVER_CLASS, JDBC_DRIVER);
         System.setProperty(PropertiesBasedJdbcDatabaseTester.DBUNIT_CONNECTION_URL, JDBC_URL);
@@ -32,7 +32,7 @@ public class SampleDbUnitTest extends DBTestCase {
 
     @Override
     protected IDataSet getDataSet() throws Exception {
-        final InputStream is = SampleDbUnitTest.class.getClassLoader().getResourceAsStream("data.xml");
+        final InputStream is = DbUnitTest.class.getClassLoader().getResourceAsStream("data.xml");
         return new FlatXmlDataSetBuilder().build(is);
     }
 
@@ -53,14 +53,14 @@ public class SampleDbUnitTest extends DBTestCase {
         final ResultSet rs = connection.createStatement().executeQuery("select * from iTEMS where id = 1");
 
         assertThat(rs.next()).isTrue();
-        assertThat(rs.getString("title")).isEqualTo("Grey T-Shirt");
+        assertThat(rs.getString( "title")).isEqualTo("Grey T-Shirt");
     }
 
     @Test
     public void testDelete() throws Exception {
         final Connection connection = getConnection().getConnection();
 
-        final InputStream is = SampleDbUnitTest.class.getClassLoader().getResourceAsStream("items_exp_delete.xml");
+        final InputStream is = DbUnitTest.class.getClassLoader().getResourceAsStream("items_exp_delete.xml");
         ITable expectedTable = (new FlatXmlDataSetBuilder().build(is)).getTable("items");
 
         connection.createStatement().executeUpdate("delete from ITEMS where id = 2");
@@ -75,7 +75,7 @@ public class SampleDbUnitTest extends DBTestCase {
     public void testUpdate() throws Exception {
         final Connection connection = getConnection().getConnection();
 
-        final InputStream is = SampleDbUnitTest.class.getClassLoader().getResourceAsStream("items_exp_rename.xml");
+        final InputStream is = DbUnitTest.class.getClassLoader().getResourceAsStream("items_exp_rename.xml");
         ITable expectedTable = (new FlatXmlDataSetBuilder().build(is)).getTable("items");
 
         connection.createStatement().executeUpdate("update ITEMS set title='new name' where id = 1");