41 lines
1.3 KiB
Java
41 lines
1.3 KiB
Java
package com.moparisthebest.jdbc.codegen;
|
|
|
|
import com.moparisthebest.jdbc.*;
|
|
import com.moparisthebest.jdbc.dto.Person;
|
|
import org.junit.Test;
|
|
|
|
import java.sql.Connection;
|
|
import java.sql.SQLException;
|
|
|
|
import static com.moparisthebest.jdbc.QueryMapperTest.fieldPerson1;
|
|
import static org.junit.Assert.assertEquals;
|
|
|
|
/**
|
|
* Created by mopar on 7/3/17.
|
|
*/
|
|
public class PersonDAOQueryRunnerTest {
|
|
|
|
public static final QueryRunner<PersonDAO> pqr = QueryRunner.withRetry(JdbcMapperFactory.of(PersonDAO.class, new Factory<Connection>() {
|
|
@Override
|
|
public Connection create() throws SQLException {
|
|
return QueryMapperTest.getConnection();
|
|
}
|
|
}), 10, QueryRunner.fixedDelay(5).withJitter(5));
|
|
|
|
@Test
|
|
public void testPerson() throws Exception {
|
|
//final QueryRunner<ListQueryMapper> lqr = pqr.withFactory(() -> new ListQueryMapper(QueryMapperTest::getConnection));
|
|
final int[] failCount = new int[]{0};
|
|
assertEquals(fieldPerson1, pqr.runRetryFuture(new QueryRunner.Runner<PersonDAO, Person>() {
|
|
@Override
|
|
public Person run(final PersonDAO dao) throws SQLException {
|
|
if(++failCount[0] < 5) {
|
|
System.out.println("fake fail");
|
|
throw new SQLException("fake 50% failure rate");
|
|
}
|
|
return dao.getPerson(fieldPerson1.getPersonNo());
|
|
}
|
|
}).get());
|
|
}
|
|
}
|