From 93049f98ee391993327484236f3064845de34a73 Mon Sep 17 00:00:00 2001
From: Sam Brannen <sbrannen@vmware.com>
Date: Fri, 9 Sep 2022 17:04:52 +0200
Subject: [PATCH] Support TestNG with @CompileWithTargetClassAccess

---
 .../compile/CompileWithTargetClassAccessClassLoader.java        | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/spring-core-test/src/main/java/org/springframework/aot/test/generate/compile/CompileWithTargetClassAccessClassLoader.java b/spring-core-test/src/main/java/org/springframework/aot/test/generate/compile/CompileWithTargetClassAccessClassLoader.java
index 03e3b7cd0ea..01e2085b51e 100644
--- a/spring-core-test/src/main/java/org/springframework/aot/test/generate/compile/CompileWithTargetClassAccessClassLoader.java
+++ b/spring-core-test/src/main/java/org/springframework/aot/test/generate/compile/CompileWithTargetClassAccessClassLoader.java
@@ -41,7 +41,7 @@ final class CompileWithTargetClassAccessClassLoader extends ClassLoader {
 
 	@Override
 	public Class<?> loadClass(String name) throws ClassNotFoundException {
-		if (name.startsWith("org.junit")) {
+		if (name.startsWith("org.junit") || name.startsWith("org.testng")) {
 			return Class.forName(name, false, this.testClassLoader);
 		}
 		return super.loadClass(name);