From b4dfa72852ab0f23201e51aab264a053ce1ce095 Mon Sep 17 00:00:00 2001 From: Eren Asker Date: Thu, 23 Nov 2023 16:37:30 +0100 Subject: [PATCH] Unify adding staff Co-authored-by: Denis Natusch Co-authored-by: Simon Bruder --- src/main/java/catering/staff/StaffManagement.java | 8 ++------ .../catering/staff/StaffControllerIntegrationTests.java | 2 +- 2 files changed, 3 insertions(+), 7 deletions(-) diff --git a/src/main/java/catering/staff/StaffManagement.java b/src/main/java/catering/staff/StaffManagement.java index 52a81f1..426e20e 100644 --- a/src/main/java/catering/staff/StaffManagement.java +++ b/src/main/java/catering/staff/StaffManagement.java @@ -17,16 +17,12 @@ public class StaffManagement { this.staffRepository = staffRepository; } - public Staff addStaff(Staff staff) { - return staffRepository.save(staff); - } - public Optional findById(Long staffID) { return staffRepository.findById(staffID); } - public void save(Staff staff) { - staffRepository.save(staff); + public Staff save(Staff staff) { + return staffRepository.save(staff); } public Streamable findAll() { diff --git a/src/test/java/catering/staff/StaffControllerIntegrationTests.java b/src/test/java/catering/staff/StaffControllerIntegrationTests.java index 704b3d9..206c391 100644 --- a/src/test/java/catering/staff/StaffControllerIntegrationTests.java +++ b/src/test/java/catering/staff/StaffControllerIntegrationTests.java @@ -39,7 +39,7 @@ class StaffControllerIntegrationTests { @BeforeEach void setup() throws Exception { - defaultEmployee = staffManagement.addStaff(new Staff("Dieter Baum", JobType.COOK)); + defaultEmployee = staffManagement.save(new Staff("Dieter Baum", JobType.COOK)); defaultEmployeeId = defaultEmployee.getId(); }