Spring/spring-batch
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

BATCH-760: added remaining database specific implementations and a delegating implementation that uses database metadata to determine which implementation to delegate to

Browse Source
This commit is contained in:
trisberg
2008-08-26 05:16:58 +00:00
parent 9798bfefe1
commit 4e39831c82
13 changed files with 658 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
spring-batch-infrastructure/src/test/java/org/springframework/batch/item/database/support/Db2PagingQueryProviderTests.java Normal file
View File

@@ -0,0 +1,36 @@
package org.springframework.batch.item.database.support;
import org.junit.Test;
import org.junit.Assert;
/**
* @author Thomas Risberg
*/
public class Db2PagingQueryProviderTests extends AbstractSqlPagingQueryProviderTests {
public Db2PagingQueryProviderTests() {
pagingQueryProvider = new Db2PagingQueryProvider();
}
@Test
@Override
public void testGenerateFirstPageQuery() {
String sql = "SELECT id, name, age FROM foo WHERE bar = 1 FETCH FIRST 100 ROWS ONLY";
String s = pagingQueryProvider.generateFirstPageQuery(pageSize);
Assert.assertEquals("", sql, s);
}
@Test @Override
public void testGenerateRemainingPagesQuery() {
String sql = "SELECT id, name, age FROM foo WHERE id > ? AND bar = 1 FETCH FIRST 100 ROWS ONLY";
String s = pagingQueryProvider.generateRemainingPagesQuery(pageSize);
Assert.assertEquals("", sql, s);
}
@Test @Override
public void testGenerateJumpToItemQuery() {
String sql = "SELECT SORT_KEY FROM ( SELECT id AS SORT_KEY, ROW_NUMBER() OVER (ORDER BY id ASC) AS ROW_NUMBER FROM foo WHERE bar = 1) WHERE ROW_NUMBER = 100";
String s = pagingQueryProvider.generateJumpToItemQuery(145, pageSize);
Assert.assertEquals("", sql, s);
}
}

36
spring-batch-infrastructure/src/test/java/org/springframework/batch/item/database/support/OraclePagingQueryProviderTests.java Normal file
View File

@@ -0,0 +1,36 @@
package org.springframework.batch.item.database.support;
import org.junit.Test;
import org.junit.Assert;
/**
* @author Thomas Risberg
*/
public class OraclePagingQueryProviderTests extends AbstractSqlPagingQueryProviderTests {
public OraclePagingQueryProviderTests() {
pagingQueryProvider = new OraclePagingQueryProvider();
}
@Test
@Override
public void testGenerateFirstPageQuery() {
String sql = "SELECT id, name, age FROM foo WHERE ROWNUM <= 100 AND bar = 1";
String s = pagingQueryProvider.generateFirstPageQuery(pageSize);
Assert.assertEquals("", sql, s);
}
@Test @Override
public void testGenerateRemainingPagesQuery() {
String sql = "SELECT id, name, age FROM foo WHERE id > ? AND ROWNUM <= 100 AND bar = 1";
String s = pagingQueryProvider.generateRemainingPagesQuery(pageSize);
Assert.assertEquals("", sql, s);
}
@Test @Override
public void testGenerateJumpToItemQuery() {
String sql = "SELECT SORT_KEY FROM ( SELECT id AS SORT_KEY, ROW_NUMBER() OVER (ORDER BY id ASC) AS ROW_NUMBER FROM foo WHERE bar = 1) WHERE ROW_NUMBER = 100";
String s = pagingQueryProvider.generateJumpToItemQuery(145, pageSize);
Assert.assertEquals("", sql, s);
}
}

234
spring-batch-infrastructure/src/test/java/org/springframework/batch/item/database/support/SimpleDelegatingPagingQueryProviderTests.java Normal file
View File

@@ -0,0 +1,234 @@
package org.springframework.batch.item.database.support;
import static org.junit.Assert.fail;
import static org.easymock.EasyMock.createMock;
import static org.easymock.EasyMock.expect;
import org.junit.Test;
import org.junit.Assert;
import org.junit.Before;
import org.easymock.EasyMock;
import org.springframework.dao.InvalidDataAccessResourceUsageException;
import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
/**
* @author Thomas Risberg
*/
public class SimpleDelegatingPagingQueryProviderTests {
protected AbstractSqlPagingQueryProvider pagingQueryProvider;
protected int pageSize;
DataSource ds;
Connection con;
DatabaseMetaData dmd;
@Before
public void onSetUp() throws Exception {
ds = createMock(DataSource.class);
con = createMock(Connection.class);
dmd = createMock(DatabaseMetaData.class);
expect(con.getMetaData()).andReturn(dmd);
expect(ds.getConnection()).andReturn(con);
pagingQueryProvider = new SimpleDelegatingPagingQueryProvider();
initializeQueryProvider(pagingQueryProvider);
}
@Test
public void testApacheDerby() throws Exception {
AbstractSqlPagingQueryProvider queryProviderToBeUsed = new DerbyPagingQueryProvider();
initializeQueryProvider(queryProviderToBeUsed);
expect(dmd.getDatabaseProductName()).andReturn("Apache Derby");
DatabaseMetaData dmd2 = createMock(DatabaseMetaData.class);
expect(dmd2.getDatabaseProductVersion()).andReturn("10.4.1.3");
expect(con.getMetaData()).andReturn(dmd2);
expect(ds.getConnection()).andReturn(con);
EasyMock.replay(dmd2);
EasyMock.replay(con);
EasyMock.replay(dmd);
EasyMock.replay(ds);
pagingQueryProvider.init(ds);
String sql = queryProviderToBeUsed.generateFirstPageQuery(pageSize);
String s = pagingQueryProvider.generateFirstPageQuery(pageSize);
Assert.assertEquals("", sql, s);
EasyMock.verify(con);
EasyMock.verify(dmd);
EasyMock.verify(dmd2);
}
@Test
public void testDb2() throws Exception {
AbstractSqlPagingQueryProvider queryProviderToBeUsed = new Db2PagingQueryProvider();
initializeQueryProvider(queryProviderToBeUsed);
EasyMock.expect(dmd.getDatabaseProductName()).andReturn("DB2");
EasyMock.replay(dmd);
EasyMock.replay(con);
EasyMock.replay(ds);
pagingQueryProvider.init(ds);
String sql = queryProviderToBeUsed.generateFirstPageQuery(pageSize);
String s = pagingQueryProvider.generateFirstPageQuery(pageSize);
Assert.assertEquals("", sql, s);
EasyMock.verify(ds);
EasyMock.verify(con);
EasyMock.verify(dmd);
}
@Test
public void testHsql() throws Exception {
AbstractSqlPagingQueryProvider queryProviderToBeUsed = new HsqlPagingQueryProvider();
initializeQueryProvider(queryProviderToBeUsed);
EasyMock.expect(dmd.getDatabaseProductName()).andReturn("HSQL Database Engine");
EasyMock.replay(dmd);
EasyMock.replay(con);
EasyMock.replay(ds);
pagingQueryProvider.init(ds);
String sql = queryProviderToBeUsed.generateFirstPageQuery(pageSize);
String s = pagingQueryProvider.generateFirstPageQuery(pageSize);
Assert.assertEquals("", sql, s);
EasyMock.verify(ds);
EasyMock.verify(con);
EasyMock.verify(dmd);
}
@Test
public void testSqlServer() throws Exception {
AbstractSqlPagingQueryProvider queryProviderToBeUsed = new SqlServerPagingQueryProvider();
initializeQueryProvider(queryProviderToBeUsed);
EasyMock.expect(dmd.getDatabaseProductName()).andReturn("Microsoft SQL Server");
EasyMock.replay(dmd);
EasyMock.replay(con);
EasyMock.replay(ds);
pagingQueryProvider.init(ds);
String sql = queryProviderToBeUsed.generateFirstPageQuery(pageSize);
String s = pagingQueryProvider.generateFirstPageQuery(pageSize);
Assert.assertEquals("", sql, s);
EasyMock.verify(ds);
EasyMock.verify(con);
EasyMock.verify(dmd);
}
@Test
public void testMySql() throws Exception {
AbstractSqlPagingQueryProvider queryProviderToBeUsed = new MySqlPagingQueryProvider();
initializeQueryProvider(queryProviderToBeUsed);
EasyMock.expect(dmd.getDatabaseProductName()).andReturn("MySQL");
EasyMock.replay(dmd);
EasyMock.replay(con);
EasyMock.replay(ds);
pagingQueryProvider.init(ds);
String sql = queryProviderToBeUsed.generateFirstPageQuery(pageSize);
String s = pagingQueryProvider.generateFirstPageQuery(pageSize);
Assert.assertEquals("", sql, s);
EasyMock.verify(ds);
EasyMock.verify(con);
EasyMock.verify(dmd);
}
@Test
public void testOracle() throws Exception {
AbstractSqlPagingQueryProvider queryProviderToBeUsed = new OraclePagingQueryProvider();
initializeQueryProvider(queryProviderToBeUsed);
EasyMock.expect(dmd.getDatabaseProductName()).andReturn("Oracle");
EasyMock.replay(dmd);
EasyMock.replay(con);
EasyMock.replay(ds);
pagingQueryProvider.init(ds);
String sql = queryProviderToBeUsed.generateFirstPageQuery(pageSize);
String s = pagingQueryProvider.generateFirstPageQuery(pageSize);
Assert.assertEquals("", sql, s);
EasyMock.verify(ds);
EasyMock.verify(con);
EasyMock.verify(dmd);
}
@Test
public void testPostgres() throws Exception {
AbstractSqlPagingQueryProvider queryProviderToBeUsed = new PostgresPagingQueryProvider();
initializeQueryProvider(queryProviderToBeUsed);
EasyMock.expect(dmd.getDatabaseProductName()).andReturn("PostgreSQL");
EasyMock.replay(dmd);
EasyMock.replay(con);
EasyMock.replay(ds);
pagingQueryProvider.init(ds);
String sql = queryProviderToBeUsed.generateFirstPageQuery(pageSize);
String s = pagingQueryProvider.generateFirstPageQuery(pageSize);
Assert.assertEquals("", sql, s);
EasyMock.verify(ds);
EasyMock.verify(con);
EasyMock.verify(dmd);
}
@Test
public void testSybase() throws Exception {
AbstractSqlPagingQueryProvider queryProviderToBeUsed = new SybasePagingQueryProvider();
initializeQueryProvider(queryProviderToBeUsed);
EasyMock.expect(dmd.getDatabaseProductName()).andReturn("Sybase");
EasyMock.replay(dmd);
EasyMock.replay(con);
EasyMock.replay(ds);
pagingQueryProvider.init(ds);
String sql = queryProviderToBeUsed.generateFirstPageQuery(pageSize);
String s = pagingQueryProvider.generateFirstPageQuery(pageSize);
Assert.assertEquals("", sql, s);
EasyMock.verify(ds);
EasyMock.verify(con);
EasyMock.verify(dmd);
}
@Test
public void testUnsupportedDatabase() throws Exception {
EasyMock.expect(dmd.getDatabaseProductName()).andReturn("MyDB");
EasyMock.replay(dmd);
EasyMock.replay(con);
EasyMock.replay(ds);
try {
pagingQueryProvider.init(ds);
fail("Expected an InvalidDataAccessResourceUsageException since the MyDB database is not supported");
} catch (InvalidDataAccessResourceUsageException e) {
// expected
}
EasyMock.verify(ds);
EasyMock.verify(con);
EasyMock.verify(dmd);
}
private void initializeQueryProvider(AbstractSqlPagingQueryProvider queryProvider) throws Exception {
queryProvider.setSelectClause("id, name, age");
queryProvider.setFromClause("foo");
queryProvider.setWhereClause("bar = 1");
queryProvider.setSortKey("id");
}
}

36
spring-batch-infrastructure/src/test/java/org/springframework/batch/item/database/support/SqlServerPagingQueryProviderTests.java Normal file
View File

@@ -0,0 +1,36 @@
package org.springframework.batch.item.database.support;
import org.junit.Test;
import org.junit.Assert;
/**
* @author Thomas Risberg
*/
public class SqlServerPagingQueryProviderTests extends AbstractSqlPagingQueryProviderTests {
public SqlServerPagingQueryProviderTests() {
pagingQueryProvider = new SqlServerPagingQueryProvider();
}
@Test
@Override
public void testGenerateFirstPageQuery() {
String sql = "SELECT TOP 100 id, name, age FROM foo WHERE bar = 1";
String s = pagingQueryProvider.generateFirstPageQuery(pageSize);
Assert.assertEquals("", sql, s);
}
@Test @Override
public void testGenerateRemainingPagesQuery() {
String sql = "SELECT TOP 100 id, name, age FROM foo WHERE id > ? AND bar = 1";
String s = pagingQueryProvider.generateRemainingPagesQuery(pageSize);
Assert.assertEquals("", sql, s);
}
@Test @Override
public void testGenerateJumpToItemQuery() {
String sql = "SELECT SORT_KEY FROM ( SELECT id AS SORT_KEY, ROW_NUMBER() OVER (ORDER BY id ASC) AS ROW_NUMBER FROM foo WHERE bar = 1) WHERE ROW_NUMBER = 100";
String s = pagingQueryProvider.generateJumpToItemQuery(145, pageSize);
Assert.assertEquals("", sql, s);
}
}

36
spring-batch-infrastructure/src/test/java/org/springframework/batch/item/database/support/SybasePagingQueryProviderTests.java Normal file
View File

@@ -0,0 +1,36 @@
package org.springframework.batch.item.database.support;
import org.junit.Test;
import org.junit.Assert;
/**
* @author Thomas Risberg
*/
public class SybasePagingQueryProviderTests extends AbstractSqlPagingQueryProviderTests {
public SybasePagingQueryProviderTests() {
pagingQueryProvider = new SybasePagingQueryProvider();
}
@Test
@Override
public void testGenerateFirstPageQuery() {
String sql = "SELECT TOP 100 id, name, age FROM foo WHERE bar = 1";
String s = pagingQueryProvider.generateFirstPageQuery(pageSize);
Assert.assertEquals("", sql, s);
}
@Test @Override
public void testGenerateRemainingPagesQuery() {
String sql = "SELECT TOP 100 id, name, age FROM foo WHERE id > ? AND bar = 1";
String s = pagingQueryProvider.generateRemainingPagesQuery(pageSize);
Assert.assertEquals("", sql, s);
}
@Test @Override
public void testGenerateJumpToItemQuery() {
String sql = "SELECT SORT_KEY FROM ( SELECT id AS SORT_KEY, ROW_NUMBER() OVER (ORDER BY id ASC) AS ROW_NUMBER FROM foo WHERE bar = 1) WHERE ROW_NUMBER = 100";
String s = pagingQueryProvider.generateJumpToItemQuery(145, pageSize);
Assert.assertEquals("", sql, s);
}
}