Add module to support testing of generated code

Closes gh-28120

Author: snicoll

build.gradle

@@ -74,6 +74,7 @@ configure(allprojects) { project ->
 			dependency "com.google.code.gson:gson:2.8.9"
 			dependency "com.google.protobuf:protobuf-java-util:3.19.3"
 			dependency "com.googlecode.protobuf-java-format:protobuf-java-format:1.4"
+			dependency "com.thoughtworks.qdox:qdox:2.0.1"
 			dependency("com.thoughtworks.xstream:xstream:1.4.18") {
 				exclude group: "xpp3", name: "xpp3_min"
 				exclude group: "xmlpull", name: "xmlpull"

framework-bom/framework-bom.gradle

@@ -7,7 +7,7 @@ group = "org.springframework"
 
 dependencies {
 	constraints {
-		parent.moduleProjects.sort { "$it.name" }.each {
+		parent.moduleProjects.findAll{ it.name != 'spring-core-test' }.sort{ "$it.name" }.each {
 			api it
 		}
 	}

settings.gradle

@@ -18,6 +18,7 @@ include "spring-context"
 include "spring-context-indexer"
 include "spring-context-support"
 include "spring-core"
+include "spring-core-test"
 include "spring-expression"
 include "spring-instrument"
 include "spring-jcl"

spring-core-test/spring-core-test.gradle

@@ -0,0 +1,11 @@
+description = "Spring Core Test"
+
+dependencies {
+	api(project(":spring-core"))
+	api("org.assertj:assertj-core")
+	api("com.thoughtworks.qdox:qdox")
+}
+
+tasks.withType(PublishToMavenRepository).configureEach {
+	it.enabled = false
+}

spring-core-test/src/main/java/org/springframework/aot/test/generator/compile/CompilationException.java

@@ -0,0 +1,32 @@
+/*
+ * Copyright 2002-2022 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.springframework.aot.test.generator.compile;
+
+/**
+ * Exception thrown when code cannot compile.
+ *
+ * @author Phillip Webb
+ * @since 6.0
+ */
+@SuppressWarnings("serial")
+public class CompilationException extends RuntimeException {
+
+	CompilationException(String message) {
+		super(message);
+	}
+
+}

spring-core-test/src/main/java/org/springframework/aot/test/generator/compile/Compiled.java

@@ -0,0 +1,170 @@
+/*
+ * Copyright 2002-2022 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.springframework.aot.test.generator.compile;
+
+import java.lang.reflect.Constructor;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+
+import org.springframework.aot.test.generator.file.ResourceFile;
+import org.springframework.aot.test.generator.file.ResourceFiles;
+import org.springframework.aot.test.generator.file.SourceFile;
+import org.springframework.aot.test.generator.file.SourceFiles;
+import org.springframework.lang.Nullable;
+import org.springframework.util.Assert;
+
+/**
+ * Fully compiled results provided from a {@link TestCompiler}.
+ *
+ * @author Phillip Webb
+ * @since 6.0
+ */
+public class Compiled {
+
+
+	private final ClassLoader classLoader;
+
+	private final SourceFiles sourceFiles;
+
+	private final ResourceFiles resourceFiles;
+
+	@Nullable
+	private List<Class<?>> compiledClasses;
+
+
+	Compiled(ClassLoader classLoader, SourceFiles sourceFiles,
+			ResourceFiles resourceFiles) {
+		this.classLoader = classLoader;
+		this.sourceFiles = sourceFiles;
+		this.resourceFiles = resourceFiles;
+	}
+
+
+	/**
+	 * Return the classloader containing the compiled content and access to the
+	 * resources.
+	 * @return the classLoader
+	 */
+	public ClassLoader getClassLoader() {
+		return this.classLoader;
+	}
+
+	/**
+	 * Return the single source file that was compiled.
+	 * @return the single source file
+	 * @throws IllegalStateException if the compiler wasn't passed exactly one
+	 * file
+	 */
+	public SourceFile getSourceFile() {
+		return this.sourceFiles.getSingle();
+	}
+
+	/**
+	 * Return all source files that were compiled.
+	 * @return the source files used by the compiler
+	 */
+	public SourceFiles getSourceFiles() {
+		return this.sourceFiles;
+	}
+
+	/**
+	 * Return the single resource file that was used when compiled.
+	 * @return the single resource file
+	 * @throws IllegalStateException if the compiler wasn't passed exactly one
+	 * file
+	 */
+	public ResourceFile getResourceFile() {
+		return this.resourceFiles.getSingle();
+	}
+
+	/**
+	 * Return all resource files that were compiled.
+	 * @return the resource files used by the compiler
+	 */
+	public ResourceFiles getResourceFiles() {
+		return this.resourceFiles;
+	}
+
+	/**
+	 * Return a new instance of a compiled class of the given type. There must
+	 * be only a single instance and it must have a default constructor.
+	 * @param <T> the required type
+	 * @param type the required type
+	 * @return an instance of type created from the compiled classes
+	 * @throws IllegalStateException if no instance can be found or instantiated
+	 */
+	public <T> T getInstance(Class<T> type) {
+		List<Class<?>> matching = getAllCompiledClasses().stream().filter(type::isAssignableFrom).toList();
+		Assert.state(!matching.isEmpty(), () -> "No instance found of type " + type.getName());
+		Assert.state(matching.size() == 1, () -> "Multiple instances found of type " + type.getName());
+		return newInstance(matching.get(0));
+	}
+
+	/**
+	 * Return an instance of a compiled class identified by its class name. The
+	 * class must have a default constructor.
+	 * @param <T> the type to return
+	 * @param type the type to return
+	 * @param className the class name to load
+	 * @return an instance of the class
+	 * @throws IllegalStateException if no instance can be found or instantiated
+	 */
+	public <T> T getInstance(Class<T> type, String className) {
+		Class<?> loaded = loadClass(className);
+		return newInstance(loaded);
+	}
+
+	/**
+	 * Return all compiled classes.
+	 * @return a list of all compiled classes
+	 */
+	public List<Class<?>> getAllCompiledClasses() {
+		List<Class<?>> compiledClasses = this.compiledClasses;
+		if (compiledClasses == null) {
+			compiledClasses = new ArrayList<>();
+			this.sourceFiles.stream().map(this::loadClass).forEach(compiledClasses::add);
+			this.compiledClasses = Collections.unmodifiableList(compiledClasses);
+		}
+		return compiledClasses;
+	}
+
+	@SuppressWarnings("unchecked")
+	private <T> T newInstance(Class<?> loaded) {
+		try {
+			Constructor<?> constructor = loaded.getDeclaredConstructor();
+			return (T) constructor.newInstance();
+		}
+		catch (Exception ex) {
+			throw new IllegalStateException(ex);
+		}
+	}
+
+	private Class<?> loadClass(SourceFile sourceFile) {
+		return loadClass(sourceFile.getClassName());
+	}
+
+	private Class<?> loadClass(String className) {
+		try {
+			return this.classLoader.loadClass(className);
+		}
+		catch (ClassNotFoundException ex) {
+			throw new IllegalStateException(ex);
+		}
+	}
+
+}

spring-core-test/src/main/java/org/springframework/aot/test/generator/compile/DynamicClassFileObject.java

@@ -0,0 +1,62 @@
+/*
+ * Copyright 2002-2022 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.springframework.aot.test.generator.compile;
+
+import java.io.ByteArrayOutputStream;
+import java.io.OutputStream;
+import java.net.URI;
+
+import javax.tools.JavaFileObject;
+import javax.tools.SimpleJavaFileObject;
+
+/**
+ * In-memory {@link JavaFileObject} used to hold class bytecode.
+ *
+ * @author Phillip Webb
+ * @since 6.0
+ */
+class DynamicClassFileObject extends SimpleJavaFileObject {
+
+	private volatile byte[] bytes = new byte[0];
+
+
+	DynamicClassFileObject(String className) {
+		super(URI.create("class:///" + className.replace('.', '/') + ".class"),
+				Kind.CLASS);
+	}
+
+
+	@Override
+	public OutputStream openOutputStream() {
+		return new JavaClassOutputStream();
+	}
+
+	byte[] getBytes() {
+		return this.bytes;
+	}
+
+
+	class JavaClassOutputStream extends ByteArrayOutputStream {
+
+		@Override
+		public void close() {
+			DynamicClassFileObject.this.bytes = toByteArray();
+		}
+
+	}
+
+}