Update new medical history screen diagnosis question options based on feature flag

Author: sagarwal

app/src/main/java/org/simple/clinic/medicalhistory/newentry/NewMedicalHistoryModel.kt

@@ -22,6 +22,7 @@ data class NewMedicalHistoryModel(
     val hasShownChangeDiagnosisError: Boolean,
     val showIsSmokingQuestion: Boolean,
     val showSmokelessTobaccoQuestion: Boolean,
+    val isScreeningFeatureEnabled: Boolean,
 ) : Parcelable {
 
   val hasLoadedPatientEntry: Boolean
@@ -76,7 +77,8 @@ data class NewMedicalHistoryModel(
     fun default(
         country: Country,
         showIsSmokingQuestion: Boolean,
-        showSmokelessTobaccoQuestion: Boolean
+        showSmokelessTobaccoQuestion: Boolean,
+        isScreeningFeatureEnabled: Boolean,
     ): NewMedicalHistoryModel = NewMedicalHistoryModel(
         country = country,
         ongoingPatientEntry = null,
@@ -85,7 +87,8 @@ data class NewMedicalHistoryModel(
         nextButtonState = null,
         hasShownChangeDiagnosisError = false,
         showIsSmokingQuestion = showIsSmokingQuestion,
-        showSmokelessTobaccoQuestion = showSmokelessTobaccoQuestion
+        showSmokelessTobaccoQuestion = showSmokelessTobaccoQuestion,
+        isScreeningFeatureEnabled = isScreeningFeatureEnabled,
     )
   }
 

app/src/main/java/org/simple/clinic/medicalhistory/newentry/NewMedicalHistoryViewModel.kt

@@ -22,7 +22,8 @@ class NewMedicalHistoryViewModel(
         country = country,
         showIsSmokingQuestion = features.isEnabled(Feature.NonLabBasedStatinNudge) ||
             features.isEnabled(Feature.LabBasedStatinNudge),
-        showSmokelessTobaccoQuestion = country.isoCountryCode != Country.ETHIOPIA
+        showSmokelessTobaccoQuestion = country.isoCountryCode != Country.ETHIOPIA,
+        isScreeningFeatureEnabled = features.isEnabled(Feature.Screening),
     ),
     init = NewMedicalHistoryInit(),
     loopFactoryProvider = { viewEffectsConsumer ->

app/src/main/java/org/simple/clinic/medicalhistory/ui/MedicalHistoryDiagnosisWithTreatment.kt

@@ -29,6 +29,7 @@ fun MedicalHistoryDiagnosisWithTreatment(
     treatmentQuestion: MedicalHistoryQuestion,
     treatmentAnswer: Answer?,
     showTreatmentQuestion: Boolean,
+    isScreeningFeatureEnabled: Boolean,
     modifier: Modifier = Modifier,
     onSelectionChange: (MedicalHistoryQuestion, Answer) -> Unit,
 ) {
@@ -43,10 +44,16 @@ fun MedicalHistoryDiagnosisWithTreatment(
         modifier = Modifier.padding(dimensionResource(R.dimen.spacing_16))
     ) {
 
+      val options = if (isScreeningFeatureEnabled) {
+        listOf(Answer.Yes, Answer.No, Answer.Suspected)
+      } else {
+        listOf(Answer.Yes, Answer.No)
+      }
+
       MedicalHistoryDiagnosisQuestionItem(
           header = diagnosisLabel,
           question = diagnosisQuestion,
-          options = listOf(Answer.Yes, Answer.No, Answer.Suspected),
+          options = options,
           selectedAnswer = diagnosisAnswer,
           onAnswerChange = onSelectionChange
       )
@@ -85,6 +92,7 @@ private fun MedicalHistoryDiagnosisWithTreatmentPreview() {
         treatmentQuestion = IsOnHypertensionTreatment(Country.INDIA),
         treatmentAnswer = Answer.Unanswered,
         showTreatmentQuestion = true,
+        isScreeningFeatureEnabled = true,
         onSelectionChange = { _, _ -> }
     )
   }

app/src/main/java/org/simple/clinic/medicalhistory/ui/NewMedicalHistoryUi.kt

@@ -94,6 +94,7 @@ fun NewMedicalHistoryUi(
             treatmentQuestion = IsOnHypertensionTreatment(model.country.isoCountryCode),
             treatmentAnswer = model.ongoingMedicalHistoryEntry.isOnHypertensionTreatment,
             showTreatmentQuestion = model.showOngoingHypertensionTreatment,
+            isScreeningFeatureEnabled = model.isScreeningFeatureEnabled,
             onSelectionChange = onSelectionChange
         )
         if (showDiabetesDiagnosis) {
@@ -104,6 +105,7 @@ fun NewMedicalHistoryUi(
               treatmentQuestion = IsOnDiabetesTreatment,
               treatmentAnswer = model.ongoingMedicalHistoryEntry.isOnDiabetesTreatment,
               showTreatmentQuestion = model.showOngoingDiabetesTreatment,
+              isScreeningFeatureEnabled = model.isScreeningFeatureEnabled,
               onSelectionChange = onSelectionChange
           )
         }
@@ -141,7 +143,8 @@ private val previewMedicalHistoryModel = NewMedicalHistoryModel(
     nextButtonState = null,
     hasShownChangeDiagnosisError = true,
     showIsSmokingQuestion = true,
-    showSmokelessTobaccoQuestion = true
+    showSmokelessTobaccoQuestion = true,
+    isScreeningFeatureEnabled = true
 )
 
 @Preview

app/src/test/java/org/simple/clinic/medicalhistory/newentry/NewMedicalHistoryInitTest.kt

@@ -14,7 +14,8 @@ class NewMedicalHistoryInitTest {
   private val defaultModel = NewMedicalHistoryModel.default(
       country = country,
       showIsSmokingQuestion = false,
-      showSmokelessTobaccoQuestion = false
+      showSmokelessTobaccoQuestion = false,
+      isScreeningFeatureEnabled = true,
   )
 
   private val initSpec = InitSpec(NewMedicalHistoryInit())

app/src/test/java/org/simple/clinic/medicalhistory/newentry/NewMedicalHistoryScreenLogicTest.kt

@@ -306,7 +306,8 @@ class NewMedicalHistoryScreenLogicTest {
         defaultModel = NewMedicalHistoryModel.default(
             country = country,
             showIsSmokingQuestion = true,
-            showSmokelessTobaccoQuestion = true
+            showSmokelessTobaccoQuestion = true,
+            isScreeningFeatureEnabled = true,
         ),
         init = NewMedicalHistoryInit(),
         update = NewMedicalHistoryUpdate(),