Write Unit Tests on Related Entity Location sync strategy enchancements

android/quest/src/test/java/org/smartregister/fhircore/quest/ui/questionnaire/QuestionnaireViewModelTest.kt

@@ -41,6 +41,7 @@ import io.mockk.verify
 import java.util.Date
 import java.util.UUID
 import javax.inject.Inject
+import kotlin.test.assertEquals
 import kotlin.time.Duration.Companion.seconds
 import kotlinx.coroutines.ExperimentalCoroutinesApi
 import kotlinx.coroutines.runBlocking
@@ -1143,4 +1144,123 @@ class QuestionnaireViewModelTest : RobolectricTest() {
       bundle.entry.any { it.resource is Basic && it.resource.id == "basic-resource-id" },
     )
   }
+
+  @Test
+  fun testApplyRelatedEntityLocationMetaTagWithResourceIdentifier() = runBlocking {
+    val questionnaireConfig =
+      questionnaireConfig.copy(
+        resourceIdentifier = "resourceId",
+        resourceType = ResourceType.Observation,
+        saveQuestionnaireResponse = false,
+        type = "EDIT",
+        extractedResourceUniquePropertyExpressions =
+          listOf(
+            ExtractedResourceUniquePropertyExpression(
+              ResourceType.Observation,
+              "Observation.code.where(coding.code='obs1').coding.code",
+            ),
+          ),
+      )
+    val questionnaireResponse =
+      QuestionnaireResponse().apply {
+        id = "resourceId"
+        meta.lastUpdated = Date()
+        subject = patient.asReference()
+        questionnaire = samplePatientRegisterQuestionnaire.asReference().reference
+      }
+    val subjectType = ResourceType.Observation
+
+    assertEquals("resourceId", questionnaireResponse.id)
+    assertEquals(subjectType, questionnaireConfig.resourceType)
+  }
+
+  @Test
+  fun `test applyRelatedEntityLocationMetaTag with groupIdentifier`() = runBlocking {
+    val bundleSlot = slot<Bundle>()
+    val bundle = bundleSlot.captured
+    val questionnaireConfig =
+      questionnaireConfig.copy(
+        resourceIdentifier = "groupId",
+        resourceType = ResourceType.Group,
+        saveQuestionnaireResponse = false,
+        type = "EDIT",
+        extractedResourceUniquePropertyExpressions =
+          listOf(
+            ExtractedResourceUniquePropertyExpression(
+              ResourceType.Observation,
+              "Observation.code.where(coding.code='obs1').coding.code",
+            ),
+          ),
+      )
+    bundle.entry.forEach {
+      val resource = it.resource
+      val subjectType = ResourceType.Group
+      resource.id = "groupId"
+
+      assertEquals(subjectType, resource.resourceType)
+      assertEquals(questionnaireConfig.resourceIdentifier, resource.id)
+    }
+  }
+
+  @Test
+  fun `test saveExtractedResources for resource modification`() = runBlocking {
+    val questionnaire = extractionQuestionnaire()
+    val questionnaireConfig =
+      questionnaireConfig.copy(
+        resourceIdentifier = "groupId",
+        resourceType = ResourceType.Group,
+        saveQuestionnaireResponse = false,
+        type = "EDIT",
+        extractedResourceUniquePropertyExpressions =
+          listOf(
+            ExtractedResourceUniquePropertyExpression(
+              ResourceType.Observation,
+              "Observation.code.where(coding.code='obs1').coding.code",
+            ),
+          ),
+      )
+    val questionnaireResponse =
+      QuestionnaireResponse().apply {
+        id = "resourceId"
+        meta.lastUpdated = Date()
+        subject = patient.asReference()
+      }
+
+    val bundle =
+      Bundle().apply {
+        addEntry(
+          Bundle.BundleEntryComponent().apply {
+            PractitionerDetails().apply {
+              fhirPractitionerDetails =
+                FhirPractitionerDetails().apply {
+                  id = "practitionerId1"
+                  practitionerId = StringType("practitionerId1")
+                }
+            }
+            id = "patientId1"
+          },
+        )
+        addEntry(
+          Bundle.BundleEntryComponent().apply {
+            PractitionerDetails().apply {
+              fhirPractitionerDetails =
+                FhirPractitionerDetails().apply {
+                  id = "practitionerId2"
+                  practitionerId = StringType("practitionerId2")
+                }
+            }
+            id = "patientId1"
+          },
+        )
+      }
+    questionnaireViewModel.saveExtractedResources(
+      bundle,
+      questionnaire,
+      questionnaireConfig,
+      questionnaireResponse,
+      context,
+    )
+
+    assertEquals("patientLogicalId", questionnaireResponse.subject.reference)
+  }
 }