com.jupitertools.springtestkafka.customizer.KafkaContextCustomizerFactory Maven / Gradle / Ivy

Go to download
package com.jupitertools.springtestkafka.customizer;


import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

import com.jupitertools.springtestkafka.KafkaTestContainer;
import com.jupitertools.springtestkafka.KafkaTestContainers;

import org.springframework.core.annotation.AnnotationUtils;
import org.springframework.test.context.ContextConfigurationAttributes;
import org.springframework.test.context.ContextCustomizer;
import org.springframework.test.context.ContextCustomizerFactory;


/**
 * ContextCustomizerFactory to find a {@link KafkaTestContainer} annotation
 * in tests and make a different context configuration for different set of
 * kafka instances in tests.
 *
 * @author Anatoliy Korovin
 */
public class KafkaContextCustomizerFactory implements ContextCustomizerFactory {

	@Override
	public ContextCustomizer createContextCustomizer(Class testClass,
	                                                 List list) {
		Set annotations =
				AnnotationUtils.getRepeatableAnnotations(testClass,
				                                         KafkaTestContainer.class,
				                                         KafkaTestContainers.class);

		Set descriptions = annotations.stream()
		                                      .map(KafkaTestContainer::bootstrapServersPropertyName)
		                                      .collect(Collectors.toSet());

		return new KafkaContainerContextCustomizer(descriptions);
	}
}