1 files changed, 18 insertions, 2 deletions
diff --git a/src/main/java/at/ac/tuwien/sepm/assignment/groupphase/einsatzverwaltung/controller/CreateNewEmployeeController.java b/src/main/java/at/ac/tuwien/sepm/assignment/groupphase/einsatzverwaltung/controller/CreateNewEmployeeController.java
index 86a1014..c5572ed 100644
--- a/src/main/java/at/ac/tuwien/sepm/assignment/groupphase/einsatzverwaltung/controller/CreateNewEmployeeController.java
+++ b/src/main/java/at/ac/tuwien/sepm/assignment/groupphase/einsatzverwaltung/controller/CreateNewEmployeeController.java
@@ -23,7 +23,6 @@ import javafx.scene.control.CheckBox;
 import javafx.scene.control.ChoiceBox;
 import javafx.scene.control.Label;
 import javafx.scene.control.TextField;
-import javafx.stage.Stage;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.stereotype.Controller;
@@ -45,6 +44,9 @@ public class CreateNewEmployeeController {
     private Employee employee;
     private boolean isEdit;
 
+    private Runnable consumerCancelClicked;
+    private Runnable consumerCreateClicked;
+
     public CreateNewEmployeeController(EmployeeService employeeService) {
         this.employeeService = employeeService;
     }
@@ -68,7 +70,9 @@ public class CreateNewEmployeeController {
 
     @FXML
     public void onCancelClicked() {
-        ((Stage) inputQualification.getScene().getWindow()).close();
+        if (consumerCancelClicked != null) {
+            consumerCancelClicked.run();
+        }
     }
 
     @FXML
@@ -111,6 +115,10 @@ public class CreateNewEmployeeController {
                 AlertType.INFORMATION,
                 "Erfolgreich angelegt",
                 "Mitarbeiter wurde erfolgreich angelegt und gespeichert!");
+
+        if (consumerCreateClicked != null) {
+            consumerCreateClicked.run();
+        }
     }
 
     private void showModalDialogWithOkButton(
@@ -160,4 +168,12 @@ public class CreateNewEmployeeController {
     public Node getRootElement() {
         return rootElement;
     }
+
+    public void setConsumerCancelClicked(Runnable consumerCancelClicked) {
+        this.consumerCancelClicked = consumerCancelClicked;
+    }
+
+    public void setConsumerCreateClicked(Runnable consumerCreateClicked) {
+        this.consumerCreateClicked = consumerCreateClicked;
+    }
 }