fix(core): allows to check existence of an attribute whatever its dat…

…asetId (#1284)
stellio-hub · Dec 13, 2024 · b7d18a2 · b7d18a2
1 parent c1c9f8f
commit b7d18a2
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 3 deletions.
diff --git a/...h-service/src/main/kotlin/com/egm/stellio/search/entity/service/EntityAttributeService.kt b/...h-service/src/main/kotlin/com/egm/stellio/search/entity/service/EntityAttributeService.kt
@@ -545,8 +545,12 @@ class EntityAttributeService(
     suspend fun checkEntityAndAttributeExistence(
         entityId: URI,
         attributeName: ExpandedTerm,
-        datasetId: URI? = null
+        datasetId: URI? = null,
+        anyAttributeInstance: Boolean = false
     ): Either<APIException, Unit> {
+        val datasetIdFilter =
+            if (anyAttributeInstance) ""
+            else datasetId.toDatasetIdFilter()
         val selectQuery =
             """
             select 
@@ -560,7 +564,7 @@ class EntityAttributeService(
                     from temporal_entity_attribute 
                     where entity_id = :entity_id 
                     and attribute_name = :attribute_name
-                    ${datasetId.toDatasetIdFilter()}
+                    $datasetIdFilter
                 ) as attributeNameExists;
             """.trimIndent()
 

diff --git a/search-service/src/main/kotlin/com/egm/stellio/search/entity/service/EntityService.kt b/search-service/src/main/kotlin/com/egm/stellio/search/entity/service/EntityService.kt
@@ -605,7 +605,8 @@ class EntityService(
             entityAttributeService.checkEntityAndAttributeExistence(
                 entityId,
                 attributeName,
-                datasetId
+                datasetId,
+                deleteAll
             ).bind()
             entityAttributeService.deleteAttribute(
                 entityId,

diff --git a/...vice/src/test/kotlin/com/egm/stellio/search/entity/service/EntityAttributeServiceTests.kt b/...vice/src/test/kotlin/com/egm/stellio/search/entity/service/EntityAttributeServiceTests.kt
@@ -639,6 +639,18 @@ class EntityAttributeServiceTests : WithTimescaleContainer, WithKafkaContainer {
             .shouldSucceed()
     }
 
+    @Test
+    fun `it should return a right unit if entiy and attribute exist whatever its datasetId`() = runTest {
+        val rawEntity = loadSampleData("beehive_multi_instance_property.jsonld")
+
+        coEvery { attributeInstanceService.create(any()) } returns Unit.right()
+
+        entityAttributeService.createAttributes(rawEntity, APIC_COMPOUND_CONTEXTS)
+
+        entityAttributeService.checkEntityAndAttributeExistence(beehiveTestCId, INCOMING_PROPERTY, null, true)
+            .shouldSucceed()
+    }
+
     @Test
     fun `it should return a left attribute not found if entity exists but not the attribute`() = runTest {
         val rawEntity = loadSampleData()