Fix hospitals codes, closes #4

renekorss · Oct 15, 2020 · 4bd38d7 · 4bd38d7
1 parent e720a3e
commit 4bd38d7
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 18 deletions.
diff --git a/src/PersonalIdCode.php b/src/PersonalIdCode.php
@@ -201,11 +201,11 @@ public function getHospital()
     {
         // Only for persons born before 2013
         if ($this->getBirthYear() < 2013) {
-            $personHospitalCode = substr($this->code, -4, 3);
+            $personHospitalCode = (int)substr($this->code, -4, 3);
 
             $hospitals = [
-                110 => 'Kuressaare Haigla',
-                190 => 'Tartu Ülikooli Naistekliinik, Tartumaa, Tartu',
+                 11 => 'Kuressaare Haigla',
+                 19 => 'Tartu Ülikooli Naistekliinik, Tartumaa, Tartu',
                 220 => 'Ida-Tallinna Keskhaigla, Pelgulinna sünnitusmaja, Hiiumaa, Keila, Rapla haigla, Loksa haigla',
                 270 => 'Ida-Viru Keskhaigla (Kohtla-Järve, endine Jõhvi)',
                 370 => 'Maarjamõisa Kliinikum (Tartu), Jõgeva Haigla',

diff --git a/tests/PersonalIdCodeTest.php b/tests/PersonalIdCodeTest.php
@@ -147,21 +147,21 @@ public function testCanGetBirthDay() : void
     public function testCanGetHospital() : void
     {
         $hospitalTest = [
-           110 => 'Kuressaare Haigla',
-           190 => 'Tartu Ülikooli Naistekliinik, Tartumaa, Tartu',
-           220 => 'Ida-Tallinna Keskhaigla, Pelgulinna sünnitusmaja, Hiiumaa, Keila, Rapla haigla, Loksa haigla',
-           270 => 'Ida-Viru Keskhaigla (Kohtla-Järve, endine Jõhvi)',
-           370 => 'Maarjamõisa Kliinikum (Tartu), Jõgeva Haigla',
-           420 => 'Narva Haigla',
-           470 => 'Pärnu Haigla',
-           490 => 'Pelgulinna Sünnitusmaja (Tallinn), Haapsalu haigla',
-           520 => 'Järvamaa Haigla (Paide)',
-           570 => 'Rakvere, Tapa haigla',
-           600 => 'Valga Haigla',
-           650 => 'Viljandi Haigla',
-           710 => 'Lõuna-Eesti Haigla (Võru), Põlva Haigla',
-           950 => 'Väljaspool Eestit',
-           800 => 'Teadmata',
+            '011' => 'Kuressaare Haigla',
+            '019' => 'Tartu Ülikooli Naistekliinik, Tartumaa, Tartu',
+            220 => 'Ida-Tallinna Keskhaigla, Pelgulinna sünnitusmaja, Hiiumaa, Keila, Rapla haigla, Loksa haigla',
+            270 => 'Ida-Viru Keskhaigla (Kohtla-Järve, endine Jõhvi)',
+            370 => 'Maarjamõisa Kliinikum (Tartu), Jõgeva Haigla',
+            420 => 'Narva Haigla',
+            470 => 'Pärnu Haigla',
+            490 => 'Pelgulinna Sünnitusmaja (Tallinn), Haapsalu haigla',
+            520 => 'Järvamaa Haigla (Paide)',
+            570 => 'Rakvere, Tapa haigla',
+            600 => 'Valga Haigla',
+            650 => 'Viljandi Haigla',
+            710 => 'Lõuna-Eesti Haigla (Võru), Põlva Haigla',
+            950 => 'Väljaspool Eestit',
+            800 => 'Teadmata',
         ];
 
         foreach ($hospitalTest as $hospitalCode => $expectedResult) {