From 19132cb481f98344ef18ff826ab61b3d14fcaa47 Mon Sep 17 00:00:00 2001
From: Benjamin Eckel <bhelx@simst.im>
Date: Fri, 13 Oct 2023 11:15:19 -0500
Subject: [PATCH] Prevent re-instantiation

---
 .../java/com/dylibso/chicory/maven/JavaTestGen.java  | 12 +++++++-----
 1 file changed, 7 insertions(+), 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
index 5970c102b..2449fb8fa 100644
--- 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;