Add shortcut in TestGenerationContext

sbrannen · Sep 28, 2022 · c5fc0a5 · c5fc0a5
1 parent 737967f
commit c5fc0a5
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/...-core-test/src/main/java/org/springframework/aot/test/generate/TestGenerationContext.java b/...-core-test/src/main/java/org/springframework/aot/test/generate/TestGenerationContext.java
@@ -51,12 +51,21 @@ public TestGenerationContext(ClassNameGenerator classNameGenerator) {
 
 	/**
 	 * Create an instance using the specified {@code target}.
-	 * @param target the default target class to use
+	 * @param target the default target class name to use
 	 */
 	public TestGenerationContext(ClassName target) {
 		this(new ClassNameGenerator(target));
 	}
 
+	/**
+	 * Create an instance using the specified {@code target}.
+	 * @param target the default target class to use
+	 */
+	public TestGenerationContext(Class<?> target) {
+		this(ClassName.get(target));
+	}
+
+
 	/**
 	 * Create an instance using {@link #TEST_TARGET} as the {@code target}.
 	 */