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(); }