3 files changed, 17 insertions, 3 deletions
diff --git a/src/test/java/ch/fhnw/thga/gradleplugins/Builder.java b/src/test/java/ch/fhnw/thga/gradleplugins/Builder.java
index cebaaf3..c8aeda7 100644
--- a/src/test/java/ch/fhnw/thga/gradleplugins/Builder.java
+++ b/src/test/java/ch/fhnw/thga/gradleplugins/Builder.java
@@ -11,5 +11,7 @@ public interface Builder {
 
     Builder outputDir(String outputDir);
 
+    Builder mainModule(String mainModule);
+
     FregeDTO build();
 }
diff --git a/src/test/java/ch/fhnw/thga/gradleplugins/FregeDTO.java b/src/test/java/ch/fhnw/thga/gradleplugins/FregeDTO.java
index 63d7dae..271d07b 100644
--- a/src/test/java/ch/fhnw/thga/gradleplugins/FregeDTO.java
+++ b/src/test/java/ch/fhnw/thga/gradleplugins/FregeDTO.java
@@ -12,14 +12,16 @@ public class FregeDTO {
     public final String compilerDownloadDir;
     public final String mainSourceDir;
     public final String outputDir;
+    public final String mainModule;
 
     public FregeDTO(String version, String release, String compilerDownloadDir, String mainSourceDir,
-            String outputDir) {
+            String outputDir, String mainModule) {
         this.version = version;
         this.release = release;
         this.compilerDownloadDir = compilerDownloadDir;
         this.mainSourceDir = mainSourceDir;
         this.outputDir = outputDir;
+        this.mainModule = mainModule;
     }
 
     public String getVersion() {
@@ -42,6 +44,10 @@ public class FregeDTO {
         return outputDir;
     }
 
+    public String getMainModule() {
+        return mainModule;
+    }
+
     private String getFieldValue(Field field) {
         try {
             return field.get(this).toString();
diff --git a/src/test/java/ch/fhnw/thga/gradleplugins/FregeDTOBuilder.java b/src/test/java/ch/fhnw/thga/gradleplugins/FregeDTOBuilder.java
index afdc862..5c24f10 100644
--- a/src/test/java/ch/fhnw/thga/gradleplugins/FregeDTOBuilder.java
+++ b/src/test/java/ch/fhnw/thga/gradleplugins/FregeDTOBuilder.java
@@ -6,6 +6,7 @@ public class FregeDTOBuilder implements Builder {
     private String compilerDownloadDir = "";
     private String mainSourceDir = "";
     private String outputDir = "";
+    private String mainModule = "";
 
     @Override
     public Builder version(String version) {
@@ -41,8 +42,13 @@ public class FregeDTOBuilder implements Builder {
         return this;
     }
 
-    public FregeDTO build() {
-        return new FregeDTO(version, release, compilerDownloadDir, mainSourceDir, outputDir);
+    @Override
+    public Builder mainModule(String mainModule) {
+        this.mainModule = mainModule;
+        return this;
     }
 
+    public FregeDTO build() {
+        return new FregeDTO(version, release, compilerDownloadDir, mainSourceDir, outputDir, mainModule);
+    }
 }