RoundEnvironmentTester Class — spring-boot Architecture

Source Code

configuration-metadata/spring-boot-configuration-processor/src/test/java/org/springframework/boot/configurationprocessor/test/RoundEnvironmentTester.java lines 27–49

public class RoundEnvironmentTester {

	private final RoundEnvironment roundEnvironment;

	RoundEnvironmentTester(RoundEnvironment roundEnvironment) {
		this.roundEnvironment = roundEnvironment;
	}

	/**
	 * Return the root {@link TypeElement} for the specified {@code type}.
	 * @param type the type of the class
	 * @return the {@link TypeElement}
	 */
	public TypeElement getRootElement(Class<?> type) {
		return (TypeElement) this.roundEnvironment.getRootElements()
			.stream()
			.filter((element) -> element.toString().equals(type.getName()))
			.findFirst()
			.orElseThrow(() -> new IllegalStateException(
					"No element found for " + type + " make sure it is included in the list of classes to compile"));
	}

}