1 files changed, 43 insertions, 0 deletions
diff --git a/src/test/java/com/stileeducation/markr/util/TestBuilder.java b/src/test/java/com/stileeducation/markr/util/TestBuilder.java
new file mode 100644
index 0000000..5c513f3
--- /dev/null
+++ b/src/test/java/com/stileeducation/markr/util/TestBuilder.java
@@ -0,0 +1,43 @@
+package com.stileeducation.markr.util;
+
+import com.stileeducation.markr.entity.Test;
+import com.stileeducation.markr.entity.TestResult;
+
+import java.util.HashSet;
+import java.util.Set;
+
+public class TestBuilder {
+  private Long id;
+  private String testId;
+  private Integer marksAvailable;
+  private Set<TestResult> testResults = new HashSet<>();
+
+  public TestBuilder withId(Long id) {
+    this.id = id;
+    return this;
+  }
+
+  public TestBuilder withTestId(String testId) {
+    this.testId = testId;
+    return this;
+  }
+
+  public TestBuilder withMarksAvailable(Integer marksAvailable) {
+    this.marksAvailable = marksAvailable;
+    return this;
+  }
+
+  public TestBuilder withTestResults(Set<TestResult> testResults) {
+    this.testResults = testResults;
+    return this;
+  }
+
+  public Test build() {
+    Test test = new Test();
+    test.setId(id);
+    test.setTestId(testId);
+    test.setMarksAvailable(marksAvailable);
+    test.setTestResults(testResults);
+    return test;
+  }
+}