WIP

src/tests/unit/common/providers/resolve_address/test_resolve_address.py

@@ -62,7 +62,7 @@ def setUp(self):
             "name": "myEngine",
             "description": "",
             "fuelPump": {
-                "address": "$fuel_pump_id",
+                "address": "dmss://datasource2/$fuel_pump_id",
                 "type": SIMOS.REFERENCE.value,
                 "referenceType": REFERENCE_TYPES.LINK.value,
             },
@@ -78,8 +78,14 @@ def setUp(self):
         self.document_repository.name = "datasource"
         self.document_repository.get = self.mock_get
         self.document_repository.find = self.mock_find
+
+        self.document_repository2 = mock.Mock()
+        self.document_repository2.name = "datasource2"
+        self.document_repository2.get = self.mock_get2
+        # self.document_repository2.find = self.mock_find2
+
         self.document_service = get_mock_document_service(
-            repository_provider=lambda x, y: self.document_repository,
+            repository_provider=lambda x, y: next((dr for dr in (self.document_repository, self.document_repository2) if dr.name == x)),
             blueprint_provider=None,
         )
 
@@ -93,7 +99,10 @@ def mock_get(self, document_id: str):
         if document_id == "customer_id":
             return {**self.customer}
         return None
-
+    def mock_get2(self, document_id: str):
+        if document_id == "fuel_pump_id":
+            return {**self.fuel_pump}
+        return None
     def mock_find(self, query: dict) -> list[dict]:
         documents: list[dict] = [
             self.car_rental_company,
@@ -231,3 +240,7 @@ def test_invalid_reference_to_primitive(self):
                 Address.from_absolute("datasource/$car_rental_company_id.cars[0].plateNumber"),
                 self.document_service.get_data_source,
             )
+
+    def test_invalid_reference_to_primitive(self):
+        Address.from_absolute(address)
+