add tests

CDCgov · Nov 3, 2023 · 8a999d3 · 8a999d3
1 parent 0fc9252
commit 8a999d3
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/backend/src/test/java/gov/cdc/usds/simplereport/service/ResultServiceTest.java b/backend/src/test/java/gov/cdc/usds/simplereport/service/ResultServiceTest.java
@@ -13,6 +13,7 @@
 import gov.cdc.usds.simplereport.db.model.TestOrder;
 import gov.cdc.usds.simplereport.db.model.auxiliary.PersonName;
 import gov.cdc.usds.simplereport.db.model.auxiliary.PersonRole;
+import gov.cdc.usds.simplereport.db.model.auxiliary.TestCorrectionStatus;
 import gov.cdc.usds.simplereport.db.model.auxiliary.TestResult;
 import gov.cdc.usds.simplereport.db.repository.ResultRepository;
 import gov.cdc.usds.simplereport.db.repository.TestEventRepository;
@@ -112,6 +113,26 @@ void getOrganizationResults_noFilter() {
       assertEquals(7, res.size());
     }
 
+    @Test
+    @SliceTestConfiguration.WithSimpleReportOrgAdminUser
+    void getOrganizationResults_doesNotShowCorrectEvents() {
+      var testEvent = testDataFactory.createTestEvent(personA, facilityA);
+      testDataFactory.createTestEventCorrection(testEvent, TestCorrectionStatus.CORRECTED);
+      var res = _service.getOrganizationResults(null, null, null, null, null, null, 0, 10).toList();
+
+      assertEquals(8, res.size());
+    }
+
+    @Test
+    @SliceTestConfiguration.WithSimpleReportOrgAdminUser
+    void getOrganizationResults_showsRemovedEvents() {
+      var testEvent = testDataFactory.createTestEvent(personA, facilityA);
+      testDataFactory.createTestEventCorrection(testEvent, TestCorrectionStatus.REMOVED);
+      var res = _service.getOrganizationResults(null, null, null, null, null, null, 0, 10).toList();
+
+      assertEquals(8, res.size());
+    }
+
     @Test
     @SliceTestConfiguration.WithSimpleReportOrgAdminUser
     void getFacilityResults_noFilter() {