3 files changed, 89 insertions, 0 deletions

diff --git a/src/test/java/DNPM/DNPMHelperTest.java b/src/test/java/DNPM/DNPMHelperTest.java
index 51f91ff..188d97a 100644
--- a/src/test/java/DNPM/DNPMHelperTest.java
+++ b/src/test/java/DNPM/DNPMHelperTest.java
@@ -2,6 +2,8 @@ package DNPM;
 
 import DNPM.services.systemtherapie.SystemtherapieService;
 import de.itc.onkostar.api.IOnkostarApi;
+import de.itc.onkostar.api.Item;
+import de.itc.onkostar.api.Procedure;
 import org.hibernate.SQLQuery;
 import org.hibernate.Session;
 import org.hibernate.SessionFactory;
@@ -13,7 +15,9 @@ import org.mockito.ArgumentCaptor;
 import org.mockito.Mock;
 import org.mockito.junit.jupiter.MockitoExtension;
 
+import java.util.Date;
 import java.util.HashMap;
+import java.util.List;
 import java.util.Map;
 
 import static org.assertj.core.api.Assertions.assertThat;
@@ -150,6 +154,53 @@ class DNPMHelperTest {
             dnpmHelper.getProzedurenFromDiagnose(Map.of("dataForm", "OS.Example", "DiagnoseId", 1, "PatientId", 2));
             verify(onkostarApi, times(1)).getProceduresByPatientId(anyInt());
         }
+
+        @Test
+        void testShouldNotReturnProceduresNotRelatedToDisease() {
+            doAnswer(invocationOnMock -> {
+                var procedure = new Procedure(onkostarApi);
+                procedure.setFormName("OS.Example1");
+                procedure.setId(11);
+                procedure.addDiseaseId(4711);
+                procedure.setValue("formfield", new Item("formfield", "Wert11"));
+
+                return List.of(procedure);
+            }).when(onkostarApi).getProceduresByPatientId(anyInt());
+
+            var actual = dnpmHelper.getProzedurenFromDiagnose(Map.of("dataForm", "OS.Example", "DiagnoseId", 1, "PatientId", 2));
+            assertThat(actual).isEqualTo("[]");
+        }
+
+        @Test
+        void testShouldReturnProcedures() {
+            doAnswer(invocationOnMock -> {
+                var procedure1 = new Procedure(onkostarApi);
+                procedure1.setFormName("OS.Example1");
+                procedure1.setId(11);
+                procedure1.addDiseaseId(1);
+                procedure1.setStartDate(new Date());
+                procedure1.setValue("formfield", new Item("formfield", "Wert11"));
+
+                var procedure2 = new Procedure(onkostarApi);
+                procedure2.setFormName("OS.Example2");
+                procedure2.setId(21);
+                procedure2.addDiseaseId(1);
+                procedure2.setStartDate(new Date());
+                procedure2.setValue("formfield", new Item("formfield", "Wert21"));
+
+                var procedure3 = new Procedure(onkostarApi);
+                procedure3.setFormName("OS.Example1");
+                procedure3.setId(12);
+                procedure3.addDiseaseId(1);
+                procedure3.setStartDate(new Date());
+                procedure3.setValue("formfield", new Item("formfield", "Wert12"));
+
+                return List.of(procedure1, procedure2, procedure3);
+            }).when(onkostarApi).getProceduresByPatientId(anyInt());
+
+            var actual = dnpmHelper.getProzedurenFromDiagnose(Map.of("dataForm", "OS.Example", "DiagnoseId", 1, "PatientId", 2));
+            assertThat(actual).contains("OS.Example1", "OS.Example2", "Wert11", "Wert21", "Wert12");
+        }
     }
 
     @Nested
diff --git a/src/test/java/de/itc/utils/DateFormats.java b/src/test/java/de/itc/utils/DateFormats.java
new file mode 100644
index 0000000..6a5f6cd
--- /dev/null
+++ b/src/test/java/de/itc/utils/DateFormats.java
@@ -0,0 +1,16 @@
+package de.itc.utils;
+
+import java.util.Date;
+
+/**
+ * Diese Klasse dient nur Testzwecken, da die Klasse und enthaltene(n) Methode(n) von der Onkostar-API
+ * benötigt und verwendet werden, nicht jedoch in der Onkostar-API-JAR enthalten sind.
+ * Das tatsächliche Verhalten innerhalb von Onkostar wird abweichen.
+ */
+public class DateFormats {
+
+    public static Date parseDate(String value) {
+        return null;
+    }
+
+}
diff --git a/src/test/java/de/itc/utils/MapUtilities.java b/src/test/java/de/itc/utils/MapUtilities.java
new file mode 100644
index 0000000..2dfe153
--- /dev/null
+++ b/src/test/java/de/itc/utils/MapUtilities.java
@@ -0,0 +1,22 @@
+package de.itc.utils;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+
+import java.util.LinkedHashMap;
+
+/**
+ * Diese Klasse dient nur Testzwecken, da die Klasse und enthaltene(n) Methode(n) von der Onkostar-API
+ * benötigt und verwendet werden, nicht jedoch in der Onkostar-API-JAR enthalten sind.
+ * Das tatsächliche Verhalten innerhalb von Onkostar wird abweichen.
+ */
+public class MapUtilities {
+
+    public static ObjectMapper getObjectMapper() {
+        return null;
+    }
+
+    public static LinkedHashMap stringToMap(String value) {
+        return null;
+    }
+
+}