78 lines
2.2 KiB
Java
78 lines
2.2 KiB
Java
package com.moparisthebest.jdbc;
|
|
|
|
import com.moparisthebest.jdbc.codegen.JdbcMapperFactory;
|
|
import com.moparisthebest.jdbc.dto.Person;
|
|
import org.junit.Assert;
|
|
import org.junit.Test;
|
|
|
|
import java.sql.Connection;
|
|
import java.sql.SQLException;
|
|
|
|
import static com.moparisthebest.jdbc.QueryMapperTest.*;
|
|
|
|
/**
|
|
* Created by mopar on 7/1/17.
|
|
*/
|
|
public class QueryRunnerTest {
|
|
public static final QueryRunner<QueryMapper> qr = QueryRunner.withRetry(JdbcMapperFactory.of(QueryMapper.class, new Factory<Connection>() {
|
|
@Override
|
|
public Connection create() throws SQLException {
|
|
return QueryMapperTest.getConnection();
|
|
}
|
|
}), 10, QueryRunner.fixedDelay(5).withJitter(5));
|
|
|
|
private void testPerson(final Person expected, final String query) throws Throwable {
|
|
//final QueryRunner<ListQueryMapper> lqr = qr.withFactory(() -> new ListQueryMapper(QueryMapperTest::getConnection));
|
|
final int[] failCount = new int[]{0};
|
|
final Person actual =
|
|
//qr.run(
|
|
//qr.runRetry(
|
|
qr.runRetryFuture(
|
|
//qr.runRetryCompletableFuture(
|
|
new QueryRunner.Runner<QueryMapper, Person>() {
|
|
@Override
|
|
public Person run(final QueryMapper qm) throws SQLException {
|
|
if(++failCount[0] < 5) {
|
|
//System.out.println("fake fail");
|
|
throw new SQLException("fake 50% failure rate");
|
|
}
|
|
return qm.toObject(query, expected.getClass(), expected.getPersonNo());
|
|
}
|
|
})
|
|
.get()
|
|
//.join()
|
|
//.thenAccept(actual -> Assert.assertEquals(expected, actual))
|
|
;
|
|
/*
|
|
System.out.println("expected: " + expected);
|
|
System.out.println("actual: " + actual);
|
|
*/
|
|
Assert.assertEquals(expected, actual);
|
|
}
|
|
|
|
@Test
|
|
public void testFieldRegularPerson() throws Throwable {
|
|
testPerson(fieldPerson1, personRegular);
|
|
}
|
|
|
|
@Test
|
|
public void testFieldRegularAndUnderscore() throws Throwable {
|
|
testPerson(fieldBoss1, bossRegularAndUnderscore);
|
|
}
|
|
|
|
@Test
|
|
public void testFieldRegularAndUnderscoreReverse() throws Throwable {
|
|
testPerson(fieldBoss1, bossRegularAndUnderscoreReverse);
|
|
}
|
|
|
|
@Test
|
|
public void testFieldRegular() throws Throwable {
|
|
testPerson(fieldBoss2, bossRegular);
|
|
}
|
|
|
|
@Test
|
|
public void testFieldUnderscore() throws Throwable {
|
|
testPerson(fieldBoss3, bossUnderscore);
|
|
}
|
|
}
|