Update CqlContentTest.kt

opensrp · sevenreup · Aug 1, 2023 · Apr 26, 2023 · May 26, 2023 · Jun 5, 2023
commit 067b2ca43408dd6a021a54f1768885b494afa08b
diff --git a/android/quest/src/test/java/org/smartregister/fhircore/quest/CqlContentTest.kt b/android/quest/src/test/java/org/smartregister/fhircore/quest/CqlContentTest.kt
@@ -50,8 +50,7 @@ import org.smartregister.fhircore.quest.robolectric.RobolectricTest
 
 @HiltAndroidTest
 class CqlContentTest : RobolectricTest() {
-  @get:Rule
-  var hiltRule = HiltAndroidRule(this)
+  @get:Rule var hiltRule = HiltAndroidRule(this)
   val fhirContext: FhirContext = FhirContext.forCached(FhirVersionEnum.R4)
   val parser = fhirContext.newJsonParser()!!
   val evaluator = LibraryEvaluator().apply { initialize() }
@@ -68,58 +67,58 @@ class CqlContentTest : RobolectricTest() {
 
     val cqlElm = toJsonElm(cql).readStringToBase64Encoded()
     val cqlLibrary =
-            parser.parseResource(
-                    "$resourceDir/library.json".readFile().replace("#library-elm.json", cqlElm)
-            ) as
-                    Library
+      parser.parseResource(
+        "$resourceDir/library.json".readFile().replace("#library-elm.json", cqlElm)
+      ) as
+        Library
 
     println(cqlLibrary.convertToString(false) as String)
 
     val fhirHelpersLibrary = "cql-common/helper.json".parseSampleResourceFromFile() as Library
 
     val patient =
-            "patient-registration-questionnaire/sample/patient.json".parseSampleResourceFromFile() as
-                    Patient
+      "patient-registration-questionnaire/sample/patient.json".parseSampleResourceFromFile() as
+        Patient
     val dataBundle =
-            Bundle().apply {
-              // output of test results extraction is input of this cql
-              "test-results-questionnaire/sample"
-                      .readDir()
-                      .map { it.parseSampleResource() as Resource }
-                      .forEach { addEntry().apply { resource = it } }
-
-              // output of test results cql is also added to input of this cql
-              "cql/test-results/sample".readDir().map { it.parseSampleResource() as Resource }.forEach {
-                addEntry().apply { resource = it }
-              }
-            }
+      Bundle().apply {
+        // output of test results extraction is input of this cql
+        "test-results-questionnaire/sample"
+          .readDir()
+          .map { it.parseSampleResource() as Resource }
+          .forEach { addEntry().apply { resource = it } }
+
+        // output of test results cql is also added to input of this cql
+        "cql/test-results/sample".readDir().map { it.parseSampleResource() as Resource }.forEach {
+          addEntry().apply { resource = it }
+        }
+      }
 
     val fhirEngine = mockk<FhirEngine>()
     val defaultRepository = mockk<DefaultRepository>()
 
     coEvery { fhirEngine.get(ResourceType.Library, cqlLibrary.logicalId) } returns cqlLibrary
     coEvery { fhirEngine.get(ResourceType.Library, fhirHelpersLibrary.logicalId) } returns
-            fhirHelpersLibrary
+      fhirHelpersLibrary
     every { defaultRepository.fhirEngine } returns fhirEngine
     coEvery { defaultRepository.save(any()) } just runs
     coEvery { defaultRepository.search(any()) } returns listOf()
 
     val result = runBlocking {
       evaluator.runCqlLibrary(
-              cqlLibrary.logicalId,
-              patient,
-              dataBundle.apply {
-                this.entry.removeIf { it.resource.resourceType == ResourceType.Patient }
-              },
-              defaultRepository,
-              true
+        cqlLibrary.logicalId,
+        patient,
+        dataBundle.apply {
+          this.entry.removeIf { it.resource.resourceType == ResourceType.Patient }
+        },
+        defaultRepository,
+        true
       )
     }
 
     assertOutput(
-            "$resourceDir/output_medication_request.json",
-            result,
-            ResourceType.MedicationRequest
+      "$resourceDir/output_medication_request.json",
+      result,
+      ResourceType.MedicationRequest
     )
 
     coVerify { defaultRepository.save(any()) }
@@ -132,59 +131,59 @@ class CqlContentTest : RobolectricTest() {
 
     val cqlElm = toJsonElm(cql).readStringToBase64Encoded()
     val cqlLibrary =
-            parser.parseResource(
-                    "$resourceDir/library.json".readFile().replace("#library-elm.json", cqlElm)
-            ) as
-                    Library
+      parser.parseResource(
+        "$resourceDir/library.json".readFile().replace("#library-elm.json", cqlElm)
+      ) as
+        Library
 
     println(cqlLibrary.convertToString(false) as String)
 
     val fhirHelpersLibrary = "cql-common/helper.json".parseSampleResourceFromFile() as Library
 
     val patient =
-            "patient-registration-questionnaire/sample/patient.json".parseSampleResourceFromFile() as
-                    Patient
+      "patient-registration-questionnaire/sample/patient.json".parseSampleResourceFromFile() as
+        Patient
     val dataBundle =
-            Bundle().apply {
-              // output of test results extraction is input of this cql
-              "test-results-questionnaire/sample"
-                      .readDir()
-                      .map { it.parseSampleResource() as Resource }
-                      .forEach { addEntry().apply { resource = it } }
-            }
+      Bundle().apply {
+        // output of test results extraction is input of this cql
+        "test-results-questionnaire/sample"
+          .readDir()
+          .map { it.parseSampleResource() as Resource }
+          .forEach { addEntry().apply { resource = it } }
+      }
 
     val fhirEngine = mockk<FhirEngine>()
     val defaultRepository = mockk<DefaultRepository>()
 
     coEvery { fhirEngine.get(ResourceType.Library, cqlLibrary.logicalId) } returns cqlLibrary
     coEvery { fhirEngine.get(ResourceType.Library, fhirHelpersLibrary.logicalId) } returns
-            fhirHelpersLibrary
+      fhirHelpersLibrary
     coEvery { defaultRepository.save(any()) } just runs
     every { defaultRepository.fhirEngine } returns fhirEngine
     coEvery { defaultRepository.search(any()) } returns listOf()
 
     val result = runBlocking {
       evaluator.runCqlLibrary(
-              cqlLibrary.logicalId,
-              patient,
-              dataBundle.apply {
-                this.entry.removeIf { it.resource.resourceType == ResourceType.Patient }
-              },
-              defaultRepository,
-              true
+        cqlLibrary.logicalId,
+        patient,
+        dataBundle.apply {
+          this.entry.removeIf { it.resource.resourceType == ResourceType.Patient }
+        },
+        defaultRepository,
+        true
       )
     }
 
     assertOutput("$resourceDir/sample/output_condition.json", result, ResourceType.Condition)
     assertOutput(
-            "$resourceDir/sample/output_service_request.json",
-            result,
-            ResourceType.ServiceRequest
+      "$resourceDir/sample/output_service_request.json",
+      result,
+      ResourceType.ServiceRequest
     )
     assertOutput(
-            "$resourceDir/sample/output_diagnostic_report.json",
-            result,
-            ResourceType.DiagnosticReport
+      "$resourceDir/sample/output_diagnostic_report.json",
+      result,
+      ResourceType.DiagnosticReport
     )
 
     coVerify(exactly = 3) { defaultRepository.save(any()) }
@@ -197,31 +196,31 @@ class CqlContentTest : RobolectricTest() {
 
     val cqlElm = toJsonElm(cql).readStringToBase64Encoded()
     val cqlLibrary =
-            parser.parseResource(
-                    "$resourceDir/library.json".readFile().replace("#library-elm.json", cqlElm)
-            ) as
-                    Library
+      parser.parseResource(
+        "$resourceDir/library.json".readFile().replace("#library-elm.json", cqlElm)
+      ) as
+        Library
 
     println(cqlLibrary.convertToString(false) as String)
 
     val fhirHelpersLibrary = "cql-common/helper.json".parseSampleResourceFromFile() as Library
 
     val dataBundle =
-            Bundle().apply {
-              addEntry().apply {
-                // questionnaire-response of test results is input of this cql
-                resource =
-                        "test-results-questionnaire/questionnaire-response.json".parseSampleResourceFromFile() as
-                                Resource
-              }
-            }
+      Bundle().apply {
+        addEntry().apply {
+          // questionnaire-response of test results is input of this cql
+          resource =
+            "test-results-questionnaire/questionnaire-response.json".parseSampleResourceFromFile() as
+              Resource
+        }
+      }
 
     val fhirEngine = mockk<FhirEngine>()
     val defaultRepository = mockk<DefaultRepository>()
 
     coEvery { fhirEngine.get(ResourceType.Library, cqlLibrary.logicalId) } returns cqlLibrary
     coEvery { fhirEngine.get(ResourceType.Library, fhirHelpersLibrary.logicalId) } returns
-            fhirHelpersLibrary
+      fhirHelpersLibrary
     every { defaultRepository.fhirEngine } returns fhirEngine
     coEvery { defaultRepository.save(any()) } just runs
     coEvery { defaultRepository.search(any<DataRequirement>()) } returns listOf()
@@ -234,23 +233,23 @@ class CqlContentTest : RobolectricTest() {
 
     Assert.assertTrue(result.contains("OUTPUT -> Correct Result"))
     Assert.assertTrue(
-            result.contains(
-                    "OUTPUT -> \nDetails:\n" +
-                            "Value (3.0) is in Normal G6PD Range 0-3\n" +
-                            "Value (11.0) is in Normal Haemoglobin Range 8-12"
-            )
+      result.contains(
+        "OUTPUT -> \nDetails:\n" +
+          "Value (3.0) is in Normal G6PD Range 0-3\n" +
+          "Value (11.0) is in Normal Haemoglobin Range 8-12"
+      )
     )
 
     val observationSlot = slot<Observation>()
     coVerify { defaultRepository.save(capture(observationSlot)) }
 
     Assert.assertEquals(
-            "QuestionnaireResponse/TEST_QUESTIONNAIRE_RESPONSE",
-            observationSlot.captured.focusFirstRep.reference
+      "QuestionnaireResponse/TEST_QUESTIONNAIRE_RESPONSE",
+      observationSlot.captured.focusFirstRep.reference
     )
     Assert.assertEquals(
-            "Correct Result",
-            observationSlot.captured.valueCodeableConcept.codingFirstRep.display
+      "Correct Result",
+      observationSlot.captured.valueCodeableConcept.codingFirstRep.display
     )
     Assert.assertEquals("Device Operation", observationSlot.captured.code.codingFirstRep.display)
   }
@@ -260,7 +259,7 @@ class CqlContentTest : RobolectricTest() {
     libraryManager.librarySourceLoader.registerProvider(FhirLibrarySourceProvider())
 
     val translator: CqlTranslator =
-            CqlTranslator.fromText(cql, evaluator.modelManager, libraryManager)
+      CqlTranslator.fromText(cql, evaluator.modelManager, libraryManager)
 
     return translator.toJxson().also { println(it.replace("\n", "").replace("   ", "")) }
   }
@@ -270,12 +269,12 @@ class CqlContentTest : RobolectricTest() {
 
     val expectedResource = resource.parseSampleResourceFromFile().convertToString(true)
     val cqlResultStr =
-            cqlResult.find { it.startsWith("OUTPUT") && it.contains("\"resourceType\":\"$type\"") }!!
-                    .replaceTimePart()
+      cqlResult.find { it.startsWith("OUTPUT") && it.contains("\"resourceType\":\"$type\"") }!!
+        .replaceTimePart()
 
     println(cqlResultStr)
     println(expectedResource as String)
 
     Assert.assertTrue(cqlResultStr.contains("OUTPUT -> $expectedResource"))
   }
-}
+}