fix tests

apache · Mar 24, 2024 · 2eb1228 · 2eb1228
1 parent b9bf895
commit 2eb1228
Showing 1 changed file with 13 additions and 7 deletions.
diff --git a/java/fury-core/src/test/java/org/apache/fury/util/unsafe/DefineClassTest.java b/java/fury-core/src/test/java/org/apache/fury/util/unsafe/DefineClassTest.java
@@ -23,6 +23,7 @@
 import org.apache.fury.codegen.CompileUnit;
 import org.apache.fury.codegen.JaninoUtils;
 import org.apache.fury.util.ClassLoaderUtils;
+import org.apache.fury.util.Platform;
 import org.testng.Assert;
 import org.testng.annotations.Test;
 
@@ -55,12 +56,17 @@ public void testDefineClass() throws ClassNotFoundException {
         new ClassLoaderUtils.ByteArrayClassLoader(Collections.singletonMap(className, bytecodes));
     DefineClass.defineClass(className, DefineClassTest.class, loader, null, bytecodes);
     Class<?> clz = loader.loadClass(className);
-    Assert.assertEquals(clz.getClassLoader(), DefineClassTest.class.getClassLoader());
-    Assert.assertThrows(
-        Exception.class,
-        () -> {
-          DefineClass.defineClass(
-              className, null, DefineClassTest.class.getClassLoader(), null, bytecodes);
-        });
+    if (Platform.JAVA_VERSION >= 9) {
+      Assert.assertEquals(clz.getClassLoader(), DefineClassTest.class.getClassLoader());
+      Assert.assertThrows(
+          Exception.class,
+          () ->
+              DefineClass.defineClass(
+                  className, null, DefineClassTest.class.getClassLoader(), null, bytecodes));
+    } else {
+      Assert.assertEquals(clz.getClassLoader(), loader);
+      DefineClass.defineClass(
+          className, null, DefineClassTest.class.getClassLoader(), null, bytecodes);
+    }
   }
 }