fxing the test case generation

Author: wadoon

key.core.testgen/src/main/java/de/uka/ilkd/key/testgen/Constants.java

@@ -13,6 +13,7 @@ public class Constants {
      */
     public static final String NEW_LINE = StringUtil.NEW_LINE;
     public static final String NULLABLE = "/*@ nullable */";
+
     public static final String ALL_OBJECTS = "allObjects";
     public static final String ALL_INTS = "allInts";
     public static final String ALL_BOOLS = "allBools";

key.core.testgen/src/main/java/de/uka/ilkd/key/testgen/ModelGenerator.java

@@ -6,7 +6,6 @@
 import java.io.IOException;
 import java.nio.file.Files;
 import java.nio.file.Path;
-import java.nio.file.StandardCopyOption;
 import java.util.Objects;
 
 /**
@@ -15,23 +14,49 @@
  * @author Alexander Weigl
  * @version 1 (02.02.24)
  */
-public record OutputEnvironment(Path targetFolder) {
-    /** Returns the source code folder */
+public record OutputEnvironment(Path targetFolder, boolean onlyTestDir) {
+    /**
+     * Returns the source code folder
+     */
     public Path getSourceDir() {
-        return targetFolder.resolve("src");
+        return targetFolder.resolve("src/main/java");
     }
 
-    /** Returns the test code folder */
+    /**
+     * Returns the test code folder
+     */
     public Path getTestSourceDir() {
-        return targetFolder.resolve("test");
+        if (onlyTestDir) {
+            return targetFolder;
+        }
+        return targetFolder.resolve("src/test/java");
     }
 
-    /** Returns the path to the ANT build.xml file */
+    /**
+     * Returns the path to the ANT build.xml file
+     */
     public Path getAntFile() {
         return targetFolder.resolve("build.xml");
     }
 
-    /** Returns the path to the README.md file */
+    /**
+     * Returns the path to the ANT build.xml file
+     */
+    public Path getGradleFile() {
+        return targetFolder.resolve("build.gradle.kts");
+    }
+
+    /**
+     * Returns the path to the pom.xml for Maven file
+     */
+    public Path getMavenFile() {
+        return targetFolder.resolve("pom.xml");
+    }
+
+
+    /**
+     * Returns the path to the README.md file
+     */
     public Path getReadmeFile() {
         return targetFolder.resolve("README.md");
     }
@@ -42,15 +67,30 @@ public Path getReadmeFile() {
      * @throws IOException if the output folder is not write or the folders can not be created.
      */
     public void init() throws IOException {
-        Files.createDirectories(getSourceDir());
         Files.createDirectories(getTestSourceDir());
-        installAntFile();
+        if (!onlyTestDir) {
+            Files.createDirectories(getSourceDir());
+            installProjectFile();
+        }
     }
 
-    private void installAntFile() throws IOException {
-        try (var buildXml = getClass().getResourceAsStream("/de/uka/ilkd/key/tcg/build.xml")) {
-            Files.copy(Objects.requireNonNull(buildXml), getAntFile(),
-                StandardCopyOption.REPLACE_EXISTING);
+    private void installProjectFile() throws IOException {
+        if (!Files.exists(getAntFile())) {
+            try (var buildXml = getClass().getResourceAsStream("/de/uka/ilkd/key/tcg/build.xml")) {
+                Files.copy(Objects.requireNonNull(buildXml), getAntFile());
+            }
+        }
+
+        if (!Files.exists(getMavenFile())) {
+            try (var pomXml = getClass().getResourceAsStream("/de/uka/ilkd/key/tcg/pom.xml")) {
+                Files.copy(Objects.requireNonNull(pomXml), getMavenFile());
+            }
+        }
+
+        if (!Files.exists(getGradleFile())) {
+            try (var gradleKts = getClass().getResourceAsStream("/de/uka/ilkd/key/tcg/build.gradle.kts")) {
+                Files.copy(Objects.requireNonNull(gradleKts), getGradleFile());
+            }
         }
     }
 }