Simplify entity model builders

1 files changed, 1 insertions, 11 deletions

diff --git a/src/test/java/com/stileeducation/markr/util/StudentBuilder.java b/src/test/java/com/stileeducation/markr/util/StudentBuilder.java
index 5189f19..897d84a 100644
--- a/src/test/java/com/stileeducation/markr/util/StudentBuilder.java
+++ b/src/test/java/com/stileeducation/markr/util/StudentBuilder.java
@@ -1,17 +1,13 @@
 package com.stileeducation.markr.util;
 
 import com.stileeducation.markr.entity.Student;
-import com.stileeducation.markr.entity.TestResult;
-
-import java.util.HashSet;
-import java.util.Set;
 
 public class StudentBuilder {
   private Long id;
   private String firstName;
   private String lastName;
   private String studentNumber;
-  private Set<TestResult> testResults = new HashSet<>();
+
 
   public StudentBuilder withId(Long id) {
     this.id = id;
@@ -33,18 +29,12 @@ public class StudentBuilder {
     return this;
   }
 
-  public StudentBuilder withTestResults(Set<TestResult> testResults) {
-    this.testResults = testResults;
-    return this;
-  }
-
   public Student build() {
     Student student = new Student();
     student.setId(id);
     student.setFirstName(firstName);
     student.setLastName(lastName);
     student.setStudentNumber(studentNumber);
-    student.setTestResults(testResults);
     return student;
   }
 }
\ No newline at end of file