From bbb70514a757b446f6cd79988bbd4061b1ff6ebd Mon Sep 17 00:00:00 2001
From: fikrimilano <fikrimilano1@gmail.com>
Date: Fri, 16 Aug 2024 13:03:47 +0700
Subject: [PATCH] Add missing response items to repeated group child items

---
 .../datacapture/QuestionnaireViewModel.kt     |  11 +-
 .../datacapture/QuestionnaireViewModelTest.kt | 135 ++++++++++++++++++
 2 files changed, 145 insertions(+), 1 deletion(-)

diff --git a/datacapture/src/main/java/com/google/android/fhir/datacapture/QuestionnaireViewModel.kt b/datacapture/src/main/java/com/google/android/fhir/datacapture/QuestionnaireViewModel.kt
index 95d75b8460..591abbd5f7 100644
--- a/datacapture/src/main/java/com/google/android/fhir/datacapture/QuestionnaireViewModel.kt
+++ b/datacapture/src/main/java/com/google/android/fhir/datacapture/QuestionnaireViewModel.kt
@@ -422,7 +422,8 @@ internal class QuestionnaireViewModel(application: Application, state: SavedStat
    * might not contain answers to unanswered or disabled questions. Note : this only applies to
    * [QuestionnaireItemComponent]s nested under a group.
    */
-  private fun addMissingResponseItems(
+  @VisibleForTesting
+  internal fun addMissingResponseItems(
     questionnaireItems: List<QuestionnaireItemComponent>,
     responseItems: MutableList<QuestionnaireResponseItemComponent>,
   ) {
@@ -446,6 +447,14 @@ internal class QuestionnaireViewModel(application: Application, state: SavedStat
             responseItems = responseItemMap[it.linkId]!!.single().item,
           )
         }
+        if (it.type == Questionnaire.QuestionnaireItemType.GROUP && it.repeats) {
+          responseItemMap[it.linkId]!!.forEach { rItem ->
+            addMissingResponseItems(
+              questionnaireItems = it.item,
+              responseItems = rItem.item,
+            )
+          }
+        }
         responseItems.addAll(responseItemMap[it.linkId]!!)
       }
     }
diff --git a/datacapture/src/test/java/com/google/android/fhir/datacapture/QuestionnaireViewModelTest.kt b/datacapture/src/test/java/com/google/android/fhir/datacapture/QuestionnaireViewModelTest.kt
index f4de30897b..f364a198f0 100644
--- a/datacapture/src/test/java/com/google/android/fhir/datacapture/QuestionnaireViewModelTest.kt
+++ b/datacapture/src/test/java/com/google/android/fhir/datacapture/QuestionnaireViewModelTest.kt
@@ -1138,6 +1138,141 @@ class QuestionnaireViewModelTest {
     }
   }
 
+  @Test
+  fun `should add missing response item inside a repeated group`() {
+    val questionnaireString =
+      """
+          {
+            "resourceType": "Questionnaire",
+            "item": [
+              {
+                "linkId": "1",
+                "type": "group",
+                "text": "Repeated Group",
+                "repeats": true,
+                "item": [
+                  {
+                    "linkId": "1-1",
+                    "type": "date",
+                    "extension": [
+                      {
+                        "url": "http://hl7.org/fhir/StructureDefinition/entryFormat",
+                        "valueString": "yyyy-mm-dd"
+                      }
+                    ]
+                  },
+                  {
+                    "linkId": "1-2",
+                    "type": "boolean"
+                  }
+                ]
+              }
+            ]
+          }
+            """
+        .trimIndent()
+
+    val questionnaireResponseString =
+      """
+              {
+                "resourceType": "QuestionnaireResponse",
+                "item": [
+                  {
+                    "linkId": "1",
+                    "text": "Repeated Group",
+                    "item": [
+                      {
+                        "linkId": "1-1",
+                        "answer": [
+                          {
+                            "valueDate": "2023-06-14"
+                          }
+                        ]
+                      }
+                    ]
+                  },
+                  {
+                    "linkId": "1",
+                    "text": "Repeated Group",
+                    "item": [
+                      {
+                        "linkId": "1-1",
+                        "answer": [
+                          {
+                            "valueDate": "2023-06-13"
+                          }
+                        ]
+                      }
+                    ]
+                  }
+                ]
+              }
+            """
+        .trimIndent()
+
+    val expectedQuestionnaireResponseString =
+      """
+              {
+                "resourceType": "QuestionnaireResponse",
+                "item": [
+                  {
+                    "linkId": "1",
+                    "text": "Repeated Group",
+                    "item": [
+                      {
+                        "linkId": "1-1",
+                        "answer": [
+                          {
+                            "valueDate": "2023-06-14"
+                          }
+                        ]
+                      },
+                      {
+                        "linkId": "1-2"
+                      }
+                    ]
+                  },
+                  {
+                    "linkId": "1",
+                    "text": "Repeated Group",
+                    "item": [
+                      {
+                        "linkId": "1-1",
+                        "answer": [
+                          {
+                            "valueDate": "2023-06-13"
+                          }
+                        ]
+                      },
+                      {
+                        "linkId": "1-2"
+                      }
+                    ]
+                  }
+                ]
+              }
+            """
+        .trimIndent()
+
+    val questionnaire =
+      printer.parseResource(Questionnaire::class.java, questionnaireString) as Questionnaire
+
+    val response =
+      printer.parseResource(QuestionnaireResponse::class.java, questionnaireResponseString)
+        as QuestionnaireResponse
+
+    val expectedResponse =
+      printer.parseResource(QuestionnaireResponse::class.java, expectedQuestionnaireResponseString)
+        as QuestionnaireResponse
+
+    val viewModel = createQuestionnaireViewModel(questionnaire, response)
+
+    runTest {
+      viewModel.addMissingResponseItems(questionnaire.item, response.item)
+      assertResourceEquals(response, expectedResponse)
+    }
+  }
+
   // ==================================================================== //
   //                                                                      //
   //                       Questionnaire State Flow                       //