Legg til databasetester

Author: magnusae

src/main/kotlin/DatabaseConfig.kt

@@ -0,0 +1,39 @@
+package no.nav.helsearbeidsgiver
+
+import no.nav.helsearbeidsgiver.utils.log.sikkerLogger
+import org.jetbrains.exposed.exceptions.ExposedSQLException
+import org.jetbrains.exposed.sql.Database
+import org.jetbrains.exposed.sql.insert
+import org.jetbrains.exposed.sql.selectAll
+import org.jetbrains.exposed.sql.transactions.transaction
+import java.util.UUID
+
+class DialogRepository(
+    private val db: Database,
+) {
+    fun lagreDialog(
+        dialogId: UUID,
+        sykmeldingId: UUID,
+    ) {
+        try {
+            transaction(db) {
+                DialogEntitet.insert {
+                    it[this.dialogId] = dialogId
+                    it[this.sykmeldingId] = sykmeldingId
+                }
+            }
+        } catch (e: ExposedSQLException) {
+            sikkerLogger().error("Klarte ikke å lagre dialog med id $dialogId i databasen", e)
+            throw e
+        }
+    }
+
+    fun finnDialogId(sykmeldingId: UUID): UUID? =
+        transaction(db) {
+            DialogEntitet
+                .selectAll()
+                .where { DialogEntitet.sykmeldingId eq sykmeldingId }
+                .map { it[DialogEntitet.dialogId] }
+                .firstOrNull()
+        }
+}

src/main/kotlin/DialogRepository.kt

@@ -25,8 +25,6 @@ object Env {
         val username = "NAIS_DATABASE_DIALOG_DIALOG_USERNAME".fromEnv()
         val password = "NAIS_DATABASE_DIALOG_DIALOG_PASSWORD".fromEnv()
         val name = "NAIS_DATABASE_DIALOG_DIALOG_DATABASE".fromEnv()
-        val host = "NAIS_DATABASE_DIALOG_DIALOG_HOST".fromEnv()
-        val port = "NAIS_DATABASE_DIALOG_DIALOG_PORT".fromEnv()
     }
 
     val navArbeidsgiverApiBaseUrl = "NAV_ARBEIDSGIVER_API_BASEURL".fromEnv()

src/main/kotlin/Env.kt

@@ -1,5 +1,10 @@
+import io.kotest.assertions.throwables.shouldThrow
 import io.kotest.matchers.collections.shouldBeEmpty
+import io.kotest.matchers.nulls.shouldNotBeNull
+import io.kotest.matchers.shouldBe
 import no.nav.helsearbeidsgiver.DialogEntitet
+import no.nav.helsearbeidsgiver.DialogRepository
+import org.jetbrains.exposed.exceptions.ExposedSQLException
 import org.jetbrains.exposed.sql.Database
 import org.jetbrains.exposed.sql.ResultRow
 import org.jetbrains.exposed.sql.selectAll
@@ -21,13 +26,46 @@ class TestRepo(
 
 class DialogRepositoryTest :
     FunSpecWithDb(listOf(DialogEntitet), { db ->
-
-        // val inntektsmeldingRepo = DialogRepository(db)
         val testRepo = TestRepo(db)
+        val dialogRepo = DialogRepository(db)
+
+        val sykmeldingId = UUID.randomUUID()
+        val dialogId = UUID.randomUUID()
 
-        test("skal lagre inntektsmeldingskjema og inntektsmelding") {
+        test("lagreDialog skal lagre dialog") {
             transaction {
                 DialogEntitet.selectAll().toList()
             }.shouldBeEmpty()
+            dialogRepo.lagreDialog(dialogId = dialogId, sykmeldingId = sykmeldingId)
+
+            val record = testRepo.hentRecordFraDialog(sykmeldingId).shouldNotBeNull()
+            record.getOrNull(DialogEntitet.dialogId) shouldBe dialogId
+        }
+
+        test("finnDialogId skal finne dialogId for sykmeldingId") {
+            transaction {
+                DialogEntitet.selectAll().toList()
+            }.shouldBeEmpty()
+
+            dialogRepo.lagreDialog(dialogId = dialogId, sykmeldingId = sykmeldingId)
+            dialogRepo.finnDialogId(sykmeldingId) shouldBe dialogId
+        }
+
+        test("lagreDialog skal kaste feil dersom den forsøker å lagre dialog med dialogId som allerede finnes") {
+            transaction {
+                DialogEntitet.selectAll().toList()
+            }.shouldBeEmpty()
+            dialogRepo.lagreDialog(dialogId = dialogId, sykmeldingId = sykmeldingId)
+
+            shouldThrow<ExposedSQLException> { dialogRepo.lagreDialog(dialogId = dialogId, sykmeldingId = UUID.randomUUID()) }
+        }
+
+        test("lagreDialog skal kaste feil dersom den forsøker å lagre dialog med sykmeldingId som allerede finnes") {
+            transaction {
+                DialogEntitet.selectAll().toList()
+            }.shouldBeEmpty()
+            dialogRepo.lagreDialog(dialogId = dialogId, sykmeldingId = sykmeldingId)
+
+            shouldThrow<ExposedSQLException> { dialogRepo.lagreDialog(dialogId = UUID.randomUUID(), sykmeldingId = sykmeldingId) }
         }
     })