Prevent re-instantiation

dylibso · Oct 13, 2023 · 19132cb · 19132cb
1 parent 8ac8043
commit 19132cb
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/test-gen-plugin/src/main/java/com/dylibso/chicory/maven/JavaTestGen.java b/test-gen-plugin/src/main/java/com/dylibso/chicory/maven/JavaTestGen.java
@@ -14,11 +14,7 @@
 import com.github.javaparser.ast.body.ClassOrInterfaceDeclaration;
 import com.github.javaparser.ast.body.MethodDeclaration;
 import com.github.javaparser.ast.body.VariableDeclarator;
-import com.github.javaparser.ast.expr.ClassExpr;
-import com.github.javaparser.ast.expr.Expression;
-import com.github.javaparser.ast.expr.IntegerLiteralExpr;
-import com.github.javaparser.ast.expr.NameExpr;
-import com.github.javaparser.ast.expr.VariableDeclarationExpr;
+import com.github.javaparser.ast.expr.*;
 import com.github.javaparser.ast.type.ClassOrInterfaceType;
 import com.github.javaparser.utils.SourceRoot;
 import com.github.javaparser.utils.StringEscapeUtils;
@@ -70,6 +66,7 @@ public CompilationUnit generate(
             cu.addImport("org.junit.jupiter.api.MethodOrderer");
             cu.addImport("org.junit.jupiter.api.TestMethodOrder");
             cu.addImport("org.junit.jupiter.api.Order");
+            cu.addImport("org.junit.jupiter.api.TestInstance");
         }
         cu.addImport("org.junit.jupiter.api.Assertions.assertEquals", true, false);
         cu.addImport("org.junit.jupiter.api.Assertions.assertThrows", true, false);
@@ -92,6 +89,11 @@ public CompilationUnit generate(
             testClass.addSingleMemberAnnotation(
                     "TestMethodOrder",
                     new ClassExpr(new ClassOrInterfaceType("MethodOrderer.OrderAnnotation")));
+            testClass.addSingleMemberAnnotation(
+                    "TestInstance",
+                    new FieldAccessExpr(
+                            new FieldAccessExpr(new NameExpr("TestInstance"), "Lifecycle"),
+                            "PER_CLASS"));
         }
 
         MethodDeclaration method;