diff --git a/test/src/test/java/com/moparisthebest/jdbc/codegen/PersonDAOQueryRunnerTest.java b/test/src/test/java/com/moparisthebest/jdbc/codegen/PersonDAOQueryRunnerTest.java index d9e5cfd..5b87323 100644 --- a/test/src/test/java/com/moparisthebest/jdbc/codegen/PersonDAOQueryRunnerTest.java +++ b/test/src/test/java/com/moparisthebest/jdbc/codegen/PersonDAOQueryRunnerTest.java @@ -25,10 +25,11 @@ public class PersonDAOQueryRunnerTest { @Test public void testPerson() throws Exception { //final QueryRunner lqr = pqr.withFactory(() -> new ListQueryMapper(QueryMapperTest::getConnection)); + final int[] failCount = new int[]{0}; assertEquals(fieldPerson1, pqr.runRetryFuture(new QueryRunner.Runner() { @Override public Person run(final PersonDAO dao) throws SQLException { - if(Math.random() < 0.5) { + if(++failCount[0] < 5) { System.out.println("fake fail"); throw new SQLException("fake 50% failure rate"); }