From eecb1a6d36c13fae0f4306ce0efbcdae814a1a5f Mon Sep 17 00:00:00 2001
From: Filip Izydorczyk <filip.izydorczyk@protonmail.com>
Date: Tue, 29 Dec 2020 15:32:09 +0100
Subject: [PATCH] Added capacioty to course/all

---
 .../controllers/CoursesController.java           | 10 +++++-----
 .../models/CoursesWithGroupsResponse.java        | 16 ++++++++--------
 2 files changed, 13 insertions(+), 13 deletions(-)

diff --git a/restservice/src/main/java/com/plannaplan/controllers/CoursesController.java b/restservice/src/main/java/com/plannaplan/controllers/CoursesController.java
index 18e0f48..011c707 100755
--- a/restservice/src/main/java/com/plannaplan/controllers/CoursesController.java
+++ b/restservice/src/main/java/com/plannaplan/controllers/CoursesController.java
@@ -9,7 +9,7 @@ import com.plannaplan.entities.Course;
 import com.plannaplan.responses.mappers.CoursesResponseMappers;
 import com.plannaplan.responses.models.CoursesDefaultResponse;
 import com.plannaplan.responses.models.CoursesWithGroupsResponse;
-import com.plannaplan.responses.models.GroupDefaultResponse;
+import com.plannaplan.responses.models.GroupWithCapacityResponse;
 import com.plannaplan.services.CourseService;
 import com.plannaplan.services.GroupService;
 import com.plannaplan.types.GroupType;
@@ -53,16 +53,16 @@ public class CoursesController {
                 final List<CoursesWithGroupsResponse> response = new ArrayList<>();
 
                 courses.forEach(course -> {
-                    final List<GroupDefaultResponse> lectures = new ArrayList<>();
-                    final List<GroupDefaultResponse> classes = new ArrayList<>();
+                    final List<GroupWithCapacityResponse> lectures = new ArrayList<>();
+                    final List<GroupWithCapacityResponse> classes = new ArrayList<>();
 
                     final HashMap<Long, Integer> ammounts = this.groupService.getTakenPlaces(course.getGroups());
 
                     course.getGroups().stream().forEach(group -> {
                         if (group.getType() == GroupType.CLASS) {
-                            classes.add(new GroupDefaultResponse(group, ammounts.get(group.getId())));
+                            classes.add(new GroupWithCapacityResponse(group, ammounts.get(group.getId())));
                         } else {
-                            lectures.add(new GroupDefaultResponse(group, ammounts.get(group.getId())));
+                            lectures.add(new GroupWithCapacityResponse(group, ammounts.get(group.getId())));
                         }
                     });
 
diff --git a/restservice/src/main/java/com/plannaplan/responses/models/CoursesWithGroupsResponse.java b/restservice/src/main/java/com/plannaplan/responses/models/CoursesWithGroupsResponse.java
index 864618a..62820fa 100755
--- a/restservice/src/main/java/com/plannaplan/responses/models/CoursesWithGroupsResponse.java
+++ b/restservice/src/main/java/com/plannaplan/responses/models/CoursesWithGroupsResponse.java
@@ -12,32 +12,32 @@ import io.swagger.annotations.ApiModel;
 @ApiModel(description = "Response shows information about groups to given course.", value = "CoursesWithGroupsResponse")
 public class CoursesWithGroupsResponse extends CoursesResponse {
 
-    private List<GroupDefaultResponse> lectures = new ArrayList<>();
-    private List<GroupDefaultResponse> classes = new ArrayList<>();
+    private List<GroupWithCapacityResponse> lectures = new ArrayList<>();
+    private List<GroupWithCapacityResponse> classes = new ArrayList<>();
 
     public CoursesWithGroupsResponse(Course course) {
         super(course);
         course.getGroups().stream().forEach(group -> {
             if (group.getType() == GroupType.CLASS) {
-                this.classes.add(new GroupDefaultResponse(group));
+                this.classes.add(new GroupWithCapacityResponse(group));
             } else {
-                this.lectures.add(new GroupDefaultResponse(group));
+                this.lectures.add(new GroupWithCapacityResponse(group));
             }
         });
     }
 
-    public CoursesWithGroupsResponse(Course course, List<GroupDefaultResponse> lectures,
-            List<GroupDefaultResponse> classes) {
+    public CoursesWithGroupsResponse(Course course, List<GroupWithCapacityResponse> lectures,
+            List<GroupWithCapacityResponse> classes) {
         super(course);
         this.lectures = lectures;
         this.classes = classes;
     }
 
-    public List<GroupDefaultResponse> getClasses() {
+    public List<GroupWithCapacityResponse> getClasses() {
         return this.classes;
     }
 
-    public List<GroupDefaultResponse> getLectures() {
+    public List<GroupWithCapacityResponse> getLectures() {
         return this.lectures;
     }