add test to Test Result Row Test

CDCgov · Nov 25, 2024 · 179bd31 · 179bd31
1 parent c1b9df1
commit 179bd31
Showing 1 changed file with 31 additions and 0 deletions.
diff --git a/backend/src/test/java/gov/cdc/usds/simplereport/validators/TestResultRowTest.java b/backend/src/test/java/gov/cdc/usds/simplereport/validators/TestResultRowTest.java
@@ -527,6 +527,37 @@ void validatePositiveHepatitisCRequiredAoeFields() {
                     "This is required because the row contains a positive Hepatitis C test result."));
   }
 
+  @Test
+  void validatePositiveGonorrheaRequiredAoeFields() {
+    Map<String, String> missingGonorrheaRequiredAoeFields =
+        getPositiveResultRowMap("Gonorrhea Model", "5028-6");
+    missingGonorrheaRequiredAoeFields.put("pregnant", "");
+    missingGonorrheaRequiredAoeFields.put("genders_of_sexual_partners", "");
+    missingGonorrheaRequiredAoeFields.put("symptomatic_for_disease", "");
+
+    ResultsUploaderCachingService resultsUploaderCachingService =
+        mock(ResultsUploaderCachingService.class);
+    when(resultsUploaderCachingService.getModelAndTestPerformedCodeToDeviceMap())
+        .thenReturn(Map.of("gonorrhea model|5028-6", TestDataBuilder.createDeviceType()));
+    when(resultsUploaderCachingService.getHepatitisCEquipmentModelAndTestPerformedCodeSet())
+        .thenReturn(Set.of("gonorrhea model|5028-6"));
+
+    TestResultRow testResultRow =
+        new TestResultRow(
+            missingGonorrheaRequiredAoeFields,
+            resultsUploaderCachingService,
+            mock(FeatureFlagsConfig.class));
+
+    List<FeedbackMessage> actual = testResultRow.validateIndividualValues();
+
+    assertThat(actual).hasSize(3);
+    actual.forEach(
+        message ->
+            assertThat(message.getMessage())
+                .contains(
+                    "This is required because the row contains a positive Gonorrhea test result."));
+  }
+
   @NotNull
   private Map<String, String> getPositiveResultRowMap(
       String deviceModelName, String testPerformedCode) {