Simplify entity model builders

1 files changed, 0 insertions, 11 deletions

diff --git a/src/test/java/com/stileeducation/markr/util/TestBuilder.java b/src/test/java/com/stileeducation/markr/util/TestBuilder.java
index 5c513f3..e247f67 100644
--- a/src/test/java/com/stileeducation/markr/util/TestBuilder.java
+++ b/src/test/java/com/stileeducation/markr/util/TestBuilder.java
@@ -1,16 +1,11 @@
 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;
@@ -27,17 +22,11 @@ public class TestBuilder {
     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;
   }
 }