håndter null-check

navikt · Jan 14, 2025 · 4a5b204 · 4a5b204
1 parent 66a4606
commit 4a5b204
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 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
@@ -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
@@ -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
@@ -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),
         )
     }