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.*;
import static com.moparisthebest.jdbc.codegen.QueryMapperQmDao.*;

/**
 * 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);
	}
}