90%

src/main/java/rocks/zipcode/io/quiz3/collections/Student.java

@@ -14,33 +14,23 @@ import static rocks.zipcode.io.quiz3.objectorientation.enums.LabStatus.PENDING;
  */
 public class Student extends Lab {
 
-    private Lab lab;
-    private LabStatus labStatus;
+    private Lab lab = new Lab();
+    private LabStatus labStatus = INCOMPLETE;
 
-    private Map<Lab, LabStatus> mMap;
-
-    public Student(Lab lab, LabStatus labStatus) {
-        this.lab = lab;
-        this.labStatus = labStatus;
-    }
-
-    public Student(String labName, Lab lab, LabStatus labStatus) {
-        super(labName);
-        this.lab = lab;
-        this.labStatus = labStatus;
-    }
+    private HashMap<Lab, LabStatus> mMap;
 
     public Student() {
         this(new HashMap<>());
     }
 
-    public Student(Map<Lab, LabStatus> map) {
+    public Student(HashMap<Lab, LabStatus> map) {
         this.mMap = map;
     }
 
     public void setLabStatus(Lab lab, LabStatus labStatus) {
-        mMap.replace(lab, labStatus);
-    }
+        if (mMap.containsKey(lab)) {
+             mMap.replace(lab, labStatus);
+        }    }
 
     public void forkLab(Lab lab) {
         mMap.replace(lab, INCOMPLETE, PENDING);
@@ -53,7 +43,7 @@ public class Student extends Lab {
         if (mMap.containsKey(labName)) {
             return mMap.get(labName);
         }
-        return mMap.get(labName);
+        return PENDING;
     }