fix: Improve test to allow multiple word street names and added cover…

…age for suffixes
cieslarmichal · Oct 4, 2024 · 17a9c30 · 17a9c30
1 parent 66688b0
commit 17a9c30
Showing 1 changed file with 16 additions and 4 deletions.
diff --git a/tests/modules/location_test.cpp b/tests/modules/location_test.cpp
@@ -883,11 +883,23 @@ TEST_F(LocationTest, shouldGenerateSlovakiaStreetAddress)
 {
     const auto generatedStreetAddress = streetAddress(Locale::sk_SK);
 
-    const auto generatedStreetAddressElements = common::split(generatedStreetAddress, " ");
+    const auto generatedAddresses = common::split(generatedStreetAddress, ", ");
+    const auto generatedStreetAddressElements = common::split(generatedAddresses[0], " ");
+
+    const auto& generatedBuildingNumber = generatedStreetAddressElements[generatedStreetAddressElements.size()-1];
+    const auto& generatedStreetSuffix = generatedStreetAddressElements[generatedStreetAddressElements.size()-2];
+    const auto& generatedStreet =
+        common::join({generatedStreetAddressElements.begin(), generatedStreetAddressElements.end() - 2});
+
+    if (generatedAddresses.size() > 1)
+    {
+        const auto& generatedSecondaryAddressParts = common::split(generatedAddresses[1], " ");
+
+        const auto& generatedUnitNumber = generatedSecondaryAddressParts[generatedSecondaryAddressParts.size()-1];
 
-    const auto& generatedStreet = generatedStreetAddressElements[0];
-    const auto& generatedStreetSuffix = generatedStreetAddressElements[1];
-    const auto& generatedBuildingNumber = generatedStreetAddressElements[2];
+        ASSERT_TRUE(generatedUnitNumber.size() == 1 || generatedUnitNumber.size() == 3);
+        ASSERT_TRUE(checkIfAllCharactersAreNumeric(generatedUnitNumber));
+    }
 
     ASSERT_TRUE(generatedBuildingNumber.size() >= 1 && generatedBuildingNumber.size() <= 3);
     ASSERT_TRUE(checkIfAllCharactersAreNumeric(generatedBuildingNumber));