From 4a5b204b37abc0f32eea2c1fde52fa2203014b59 Mon Sep 17 00:00:00 2001
From: "Anders.Vik.Lysne" <anders.vik.lysne@nav.no>
Date: Tue, 14 Jan 2025 08:31:26 +0100
Subject: [PATCH] =?UTF-8?q?h=C3=A5ndter=20null-check?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../min_side/models/Organisasjon.kt           |  5 ++++-
 .../services/digisyfo/DigisyfoService.kt      |  6 +++---
 .../services/digisyfo/DigisyfoServiceTest.kt  | 20 +++++++++----------
 3 files changed, 17 insertions(+), 14 deletions(-)

diff --git a/src/main/kotlin/no/nav/arbeidsgiver/min_side/models/Organisasjon.kt b/src/main/kotlin/no/nav/arbeidsgiver/min_side/models/Organisasjon.kt
index 62c100c3..d067d13d 100644
--- a/src/main/kotlin/no/nav/arbeidsgiver/min_side/models/Organisasjon.kt
+++ b/src/main/kotlin/no/nav/arbeidsgiver/min_side/models/Organisasjon.kt
@@ -9,7 +9,10 @@ data class Organisasjon(
     var organizationForm: String,
     ) {
     companion object {
-        fun fromEregOrganisasjon(eregOrganisasjon: EregOrganisasjon): Organisasjon {
+        fun fromEregOrganisasjon(eregOrganisasjon: EregOrganisasjon?): Organisasjon? {
+            if (eregOrganisasjon == null) {
+                return null
+            }
             return Organisasjon(
                 name = eregOrganisasjon.navn,
                 parentOrganizationNumber = eregOrganisasjon.overordnetEnhet,
diff --git a/src/main/kotlin/no/nav/arbeidsgiver/min_side/services/digisyfo/DigisyfoService.kt b/src/main/kotlin/no/nav/arbeidsgiver/min_side/services/digisyfo/DigisyfoService.kt
index 62a40f55..7cb0d3d3 100644
--- a/src/main/kotlin/no/nav/arbeidsgiver/min_side/services/digisyfo/DigisyfoService.kt
+++ b/src/main/kotlin/no/nav/arbeidsgiver/min_side/services/digisyfo/DigisyfoService.kt
@@ -86,7 +86,7 @@ class DigisyfoService(
         if (org.parentOrganizationNumber == null) return orgs
 
         val overenhet = eregService.hentOverenhet(org.parentOrganizationNumber!!).let {
-            Organisasjon.fromEregOrganisasjon(it!!)
+            Organisasjon.fromEregOrganisasjon(it)
         } ?: return orgs
 
         orgs.add(overenhet)
@@ -97,7 +97,7 @@ class DigisyfoService(
 
     private fun hentOverenhet(orgnr: String): List<VirksomhetOgAntallSykmeldte> {
         val hovedenhet = eregService.hentOverenhet(orgnr).let {
-            Organisasjon.fromEregOrganisasjon(it!!)
+            Organisasjon.fromEregOrganisasjon(it)
         } ?: return listOf()
         val forfedre = hentForfedre(hovedenhet)
         val result = mutableListOf<VirksomhetOgAntallSykmeldte>()
@@ -108,7 +108,7 @@ class DigisyfoService(
 
     private fun hentUnderenhet(virksomhetsinfo: Virksomhetsinfo): List<VirksomhetOgAntallSykmeldte> {
         val underenhet = eregService.hentUnderenhet(virksomhetsinfo.virksomhetsnummer).let {
-            Organisasjon.fromEregOrganisasjon(it!!)
+            Organisasjon.fromEregOrganisasjon(it)
         }
             ?: return listOf()
         return listOf(
diff --git a/src/test/kotlin/no/nav/arbeidsgiver/min_side/services/digisyfo/DigisyfoServiceTest.kt b/src/test/kotlin/no/nav/arbeidsgiver/min_side/services/digisyfo/DigisyfoServiceTest.kt
index f11ec212..6c29949e 100644
--- a/src/test/kotlin/no/nav/arbeidsgiver/min_side/services/digisyfo/DigisyfoServiceTest.kt
+++ b/src/test/kotlin/no/nav/arbeidsgiver/min_side/services/digisyfo/DigisyfoServiceTest.kt
@@ -82,11 +82,11 @@ class DigisyfoServiceTest {
 
         val result = digisyfoService.hentVirksomheterOgSykmeldte("42")
         assertThat(result).containsExactly(
-            VirksomhetOgAntallSykmeldte(Organisasjon.fromEregOrganisasjon(mkUnderenhet("10", "1")), 0),
-            VirksomhetOgAntallSykmeldte(Organisasjon.fromEregOrganisasjon(mkUnderenhet("11", "1")), 1),
-            VirksomhetOgAntallSykmeldte(Organisasjon.fromEregOrganisasjon(mkUnderenhet("20", "2")), 2),
-            VirksomhetOgAntallSykmeldte(Organisasjon.fromEregOrganisasjon(mkOverenhet("1")), 0),
-            VirksomhetOgAntallSykmeldte(Organisasjon.fromEregOrganisasjon(mkOverenhet("2")), 0),
+            VirksomhetOgAntallSykmeldte(Organisasjon.fromEregOrganisasjon(mkUnderenhet("10", "1"))!!, 0),
+            VirksomhetOgAntallSykmeldte(Organisasjon.fromEregOrganisasjon(mkUnderenhet("11", "1"))!!, 1),
+            VirksomhetOgAntallSykmeldte(Organisasjon.fromEregOrganisasjon(mkUnderenhet("20", "2"))!!, 2),
+            VirksomhetOgAntallSykmeldte(Organisasjon.fromEregOrganisasjon(mkOverenhet("1"))!!, 0),
+            VirksomhetOgAntallSykmeldte(Organisasjon.fromEregOrganisasjon(mkOverenhet("2"))!!, 0),
         )
     }
 
@@ -101,11 +101,11 @@ class DigisyfoServiceTest {
 
         val result = digisyfoService.hentVirksomheterOgSykmeldte("42")
         assertThat(result).containsExactly(
-            VirksomhetOgAntallSykmeldte(Organisasjon.fromEregOrganisasjon(mkUnderenhet("3000", "300")), 2),
-            VirksomhetOgAntallSykmeldte(Organisasjon.fromEregOrganisasjon(mkUnderenhet("301", "30")), 1),
-            VirksomhetOgAntallSykmeldte(Organisasjon.fromEregOrganisasjon(mkUnderenhet("300", "30")), 0),
-            VirksomhetOgAntallSykmeldte(Organisasjon.fromEregOrganisasjon(mkUnderenhet("30", "3")), 0),
-            VirksomhetOgAntallSykmeldte(Organisasjon.fromEregOrganisasjon(mkOverenhet("3")), 0),
+            VirksomhetOgAntallSykmeldte(Organisasjon.fromEregOrganisasjon(mkUnderenhet("3000", "300"))!!, 2),
+            VirksomhetOgAntallSykmeldte(Organisasjon.fromEregOrganisasjon(mkUnderenhet("301", "30"))!!, 1),
+            VirksomhetOgAntallSykmeldte(Organisasjon.fromEregOrganisasjon(mkUnderenhet("300", "30"))!!, 0),
+            VirksomhetOgAntallSykmeldte(Organisasjon.fromEregOrganisasjon(mkUnderenhet("30", "3"))!!, 0),
+            VirksomhetOgAntallSykmeldte(Organisasjon.fromEregOrganisasjon(mkOverenhet("3"))!!, 0),
         )
     }