package com.jaregu.database.queries;

import static org.assertj.core.api.Assertions.assertThat;

import java.lang.annotation.Annotation;
import java.lang.annotation.Retention;
import java.util.HashMap;
import java.util.Map;

import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnitRunner;

import com.jaregu.database.queries.building.ParameterBinder;
import com.jaregu.database.queries.dialect.Dialect;
import com.jaregu.database.queries.proxy.QueryConverterFactory;
import com.jaregu.database.queries.proxy.QueryMapperFactory;

import groovy.lang.DelegatesTo.Target;

@RunWith(MockitoJUnitRunner.class)
public class QueriesConfigImplTest {

	private QueriesConfigImpl config;

	@Mock
	private Dialect dialect;

	@Mock
	private ParameterBinder binder;

	@Mock
	private QueryMapperFactory mapper1;

	@Mock
	private QueryMapperFactory mapper2;

	private Map<Class<? extends Annotation>, QueryMapperFactory> mappers;

	@Mock
	private QueryConverterFactory converter1;

	@Mock
	private QueryConverterFactory converter2;

	private Map<Class<? extends Annotation>, QueryConverterFactory> converters;

	@Before
	public void setUp() {
		mappers = new HashMap<>();
		mappers.put(Retention.class, mapper1);
		mappers.put(Target.class, mapper2);

		converters = new HashMap<>();
		converters.put(Retention.class, converter1);
		converters.put(Target.class, converter2);

		config = new QueriesConfigImpl(dialect, binder, mappers, converters);
	}

	@Test
	public void testGetters() {
		testGetters(config);
	}

	public void testGetters(QueriesConfig config) {
		assertThat(config.getDialect()).isSameAs(dialect);
		assertThat(config.getParameterBinder()).isSameAs(binder);
		assertThat(config.getQueryMapperFactories().values().stream().map(v -> (QueryMapperFactory) v))
				.containsOnly(mapper1, mapper2);
		assertThat(config.getQueryMapperFactories()).containsAllEntriesOf(mappers);
		assertThat(config.getQueryConverterFactories()).containsAllEntriesOf(converters);
	}
}

@@ -165,24 +165,60 @@
 .Retention;%0A
+import java.lang.annotation.Target;%0A
 import java.
@@ -649,48 +649,8 @@
 y;%0A%0A
-import groovy.lang.DelegatesTo.Target;%0A%0A
 @Run

Fix wrong annotation import