Add Country::getCode functionality

Author: bjoern-hempel

src/Country.php

@@ -53,6 +53,24 @@ private function getLanguage(array $values, string $language): string
         return $values[$language];
     }
 
+    /**
+     * Returns the code of country.
+     *
+     * @return string
+     */
+    public function getCode(): string
+    {
+        if ($this->country === '') {
+            return CountryUnknown::COUNTRY_CODE_IV;
+        }
+
+        if (!array_key_exists($this->country, CountryAll::COUNTRY_NAMES)) {
+            return CountryUnknown::COUNTRY_CODE_UK;
+        }
+
+        return $this->country;
+    }
+
     /**
      * Returns the name of country.
      *

tests/Unit/CountryTest.php

@@ -82,19 +82,27 @@ public function dataProvider(): array
             /**
              * getName: Africa
              */
+            [++$number, 'getCode', null, CountryAfrica::COUNTRY_CODE_CI, CountryAfrica::COUNTRY_CODE_CI],
             [++$number, 'getName', null, CountryAfrica::COUNTRY_CODE_CI, CountryAfrica::COUNTRY_NAME_CI[Language::EN_GB]],
+            [++$number, 'getCode', null, CountryAfrica::COUNTRY_CODE_GH, CountryAfrica::COUNTRY_CODE_GH],
             [++$number, 'getName', null, CountryAfrica::COUNTRY_CODE_GH, CountryAfrica::COUNTRY_NAME_GH[Language::EN_GB]],
 
             /**
              * getName: Africa
              */
+            [++$number, 'getCode', null, CountryAmerica::COUNTRY_CODE_AR, CountryAmerica::COUNTRY_CODE_AR],
             [++$number, 'getName', null, CountryAmerica::COUNTRY_CODE_AR, CountryAmerica::COUNTRY_NAME_AR[Language::EN_GB]],
 
             /**
              * getName: Asia
              */
+            [++$number, 'getCode', null, CountryAsia::COUNTRY_CODE_AE, CountryAsia::COUNTRY_CODE_AE],
             [++$number, 'getName', null, CountryAsia::COUNTRY_CODE_AE, CountryAsia::COUNTRY_NAME_AE[Language::EN_GB]],
+            [++$number, 'getCode', null, CountryAsia::COUNTRY_CODE_IN, CountryAsia::COUNTRY_CODE_IN],
+            [++$number, 'getName', null, CountryAsia::COUNTRY_CODE_IN, CountryAsia::COUNTRY_NAME_IN[Language::EN_GB]],
+            [++$number, 'getCode', null, CountryAsia::COUNTRY_CODE_KZ, CountryAsia::COUNTRY_CODE_KZ],
             [++$number, 'getName', null, CountryAsia::COUNTRY_CODE_KZ, CountryAsia::COUNTRY_NAME_KZ[Language::EN_GB]],
+            [++$number, 'getCode', null, CountryAsia::COUNTRY_CODE_RU, CountryAsia::COUNTRY_CODE_RU],
             [++$number, 'getName', null, CountryAsia::COUNTRY_CODE_RU, CountryAsia::COUNTRY_NAME_RU[Language::EN_GB]],
 
             /**
@@ -130,7 +138,9 @@ public function dataProvider(): array
             /**
              * getName: Unknown/Invalid
              */
+            [++$number, 'getCode', null, '', CountryUnknown::COUNTRY_CODE_IV],
             [++$number, 'getName', null, '', CountryUnknown::COUNTRY_NAME_IV[Language::EN_GB]],
+            [++$number, 'getCode', null, 'XX', CountryUnknown::COUNTRY_CODE_UK],
             [++$number, 'getName', null, 'XX', CountryUnknown::COUNTRY_NAME_UK[Language::EN_GB]],
         ];
     }