labs

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

@@ -1,24 +1,38 @@
 package rocks.zipcode.io.quiz3.collections;
 
+import static rocks.zipcode.io.quiz3.objectorientation.enums.LabStatus.PENDING;
+
 /**
  * @author leon on 10/12/2018.
  */
-public class Lab {
+public class Lab implements Comparable {
+
+    private String labName;
+    private String labStatus;
+
     public Lab() {
         this(null);
     }
 
     public Lab(String labName) {
+        this.labName = labName;
+        this.labStatus = PENDING.getLabStatusName();
     }
 
     public String getLabStatus() {
-        return null;
+        return labStatus;
     }
 
     public void setLabStatus(String labStatus) {
+        this.labStatus = labStatus;
     }
 
     public String getName() {
-        return null;
+        return labName;
+    }
+
+    @Override
+    public int compareTo(Object o) {
+        return this.getName().compareTo(((Lab) o).getName());
     }
 }

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

@@ -2,27 +2,55 @@ package rocks.zipcode.io.quiz3.collections;
 
 import rocks.zipcode.io.quiz3.objectorientation.enums.LabStatus;
 
+import java.util.HashMap;
 import java.util.Map;
+import java.util.TreeMap;
 
 /**
  * @author leon on 10/12/2018.
  */
 public class Student {
+
+    Map<Lab, LabStatus> map;
+
     public Student() {
-        this(null);
+        this(new TreeMap<>());
     }
 
     public Student(Map<Lab, LabStatus> map) {
+        this.map = map;
     }
 
     public void setLabStatus(Lab lab, LabStatus labStatus) {
+        if (map.containsKey(lab) && map.get(lab).equals(LabStatus.PENDING)) {
+            lab.setLabStatus(labStatus.getLabStatusName());
+            map.put(lab, labStatus);
+        }
+        else throw new UnsupportedOperationException();
     }
 
 
     public void forkLab(Lab lab) {
+        map.put(lab,LabStatus.PENDING);
     }
 
     public LabStatus getLabStatus(String labName) {
-        throw new UnsupportedOperationException("Method not yet implemented");
+        try {
+            for (Lab l : map.keySet())
+                if (l.getName().equals(labName))
+                    return map.get(l);
+        } catch (Exception e) {
+            throw new UnsupportedOperationException();
+        }
+
+        return null;
+    }
+
+    @Override
+    public String toString() {
+        StringBuilder sb = new StringBuilder();
+        for (Lab l : map.keySet())
+            sb.append(l.getName()).append(" > ").append(l.getLabStatus()).append("\n");
+        return sb.toString().trim();
     }
 }

src/main/java/rocks/zipcode/io/quiz3/objectorientation/enums/LabStatus.java

@@ -4,5 +4,17 @@ package rocks.zipcode.io.quiz3.objectorientation.enums;
  * @author leon on 10/12/2018.
  */
 public enum LabStatus {
-    ADD_ENUMERATIONS_HERE;
+    COMPLETED("COMPLETED"),
+    INCOMPLETE("INCOMPLETE"),
+    PENDING("PENDING");
+
+    private String labStatusName;
+
+    LabStatus(String labStatusName) {
+        this.labStatusName = labStatusName;
+    }
+
+    public String getLabStatusName() {
+        return labStatusName;
+    }
 }