single group fix

Reviewed-on: http://git.plannaplan.pl/filipizydorczyk/backend/pulls/53

restservice/src/main/java/com/plannaplan/responses/models/AssignmentResponse.java

@@ -16,6 +16,7 @@ import io.swagger.annotations.ApiModel;
 public class AssignmentResponse {
     private Long id;
     private String name;
+    private String symbol;
     private GroupWithCapacityResponse classes;
     private GroupWithCapacityResponse lecture;
 
@@ -27,10 +28,18 @@ public class AssignmentResponse {
     public AssignmentResponse(Course course, Groups lecture, Groups classes) {
         this.id = course.getId();
         this.name = course.getName();
+        this.symbol = course.getSymbol();
         this.lecture = lecture == null ? null : new GroupWithCapacityResponse(lecture);
         this.classes = classes == null ? null : new GroupWithCapacityResponse(classes);
     }
 
+    /**
+     * @return returns symbol of assigned course
+     */
+    public String getSymbol() {
+        return symbol;
+    }
+
     /**
      * @param course   course entity
      * @param lecture  lecture Groups entity
@@ -41,6 +50,7 @@ public class AssignmentResponse {
     public AssignmentResponse(Course course, Groups lecture, Groups classes, HashMap<Long, Integer> ammounts) {
         this.id = course.getId();
         this.name = course.getName();
+        this.symbol = course.getSymbol();
         this.lecture = lecture == null ? null : new GroupWithCapacityResponse(lecture, ammounts.get(lecture.getId()));
         this.classes = classes == null ? null : new GroupWithCapacityResponse(classes, ammounts.get(classes.getId()));
     }
@@ -97,8 +107,13 @@ public class AssignmentResponse {
     public AssignmentResponse(Course course, Assignment lecture, Assignment classes, HashMap<Long, Integer> ammounts) {
         this.id = course.getId();
         this.name = course.getName();
-        this.classes = new GroupWithCapacityResponse(classes, ammounts.get(classes.getGroup().getId()));
-        this.lecture = new GroupWithCapacityResponse(lecture, ammounts.get(lecture.getGroup().getId()));
+        this.symbol = course.getSymbol();
+        this.classes = classes != null
+                ? new GroupWithCapacityResponse(classes, ammounts.get(classes.getGroup().getId()))
+                : null;
+        this.lecture = lecture != null
+                ? new GroupWithCapacityResponse(lecture, ammounts.get(lecture.getGroup().getId()))
+                : null;
     }
 
     /**
@@ -109,8 +124,9 @@ public class AssignmentResponse {
     public AssignmentResponse(Course course, Assignment lecture, Assignment classes) {
         this.id = course.getId();
         this.name = course.getName();
-        this.classes = new GroupWithCapacityResponse(classes);
-        this.lecture = new GroupWithCapacityResponse(lecture);
+        this.symbol = course.getSymbol();
+        this.classes = classes != null ? new GroupWithCapacityResponse(classes) : null;
+        this.lecture = lecture != null ? new GroupWithCapacityResponse(lecture) : null;
     }
 
     /**

restservice/src/main/java/com/plannaplan/responses/models/GroupWithCapacityResponse.java

@@ -50,6 +50,7 @@ public class GroupWithCapacityResponse extends GroupDefaultResponse {
      */
     public GroupWithCapacityResponse(Assignment assignment, int takenPlaces) {
         super(assignment, takenPlaces);
+        this.capacity = assignment.getGroup().getCapacity();
     }
 
     /**