1 files changed, 55 insertions, 0 deletions
diff --git a/src/test/java/dev/dnpm/services/StudieTest.java b/src/test/java/dev/dnpm/services/StudieTest.java
new file mode 100644
index 0000000..debf6b9
--- /dev/null
+++ b/src/test/java/dev/dnpm/services/StudieTest.java
@@ -0,0 +1,55 @@
+package dev.dnpm.services;
+
+import dev.dnpm.dto.Studie;
+import org.junit.jupiter.api.Test;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+class StudieTest {
+
+    @Test
+    void shouldDetectStudieWithNctNumber() {
+        var studie = new Studie(
+                "Kat 1",
+                1,
+                null,
+                "Nct-12345678",
+                "Teststudie 1",
+                "Teststudie 1",
+                true
+        );
+
+        assertThat(studie.getType()).isEqualTo(Studie.Type.NCT);
+    }
+
+    @Test
+    void shouldDetectStudieWithEudraCtNumber() {
+        var studie = new Studie(
+                "Kat 1",
+                1,
+                null,
+                "2023-012345-12",
+                "Teststudie 1",
+                "Teststudie 1",
+                true
+        );
+
+        assertThat(studie.getType()).isEqualTo(Studie.Type.EUDRA_CT);
+    }
+
+    @Test
+    void shouldReturnStudieWithUnknownNumberScheme() {
+        var studie = new Studie(
+                "Kat 1",
+                1,
+                "teststudie1",
+                null,
+                "Teststudie 1",
+                "Teststudie 1",
+                true
+        );
+
+        assertThat(studie.getType()).isEqualTo(Studie.Type.UNKNOWN);
+    }
+
+}