add tests for overshare

Author: aschey-forpeople

apps/bfd-pipeline/bfd-pipeline-idr/test_samples2/merged_benes/SYNTHETIC_BENE_HSTRY.csv

@@ -3,3 +3,5 @@ BENE_SK,BENE_XREF_EFCTV_SK,BENE_XREF_SK,BENE_MBI_ID,BENE_LAST_NAME,BENE_1ST_NAME
 848484848,517782585,517782585,2B19C89AA35,Heeler,Alex,Joey,1959-03-14,,~,1,4,14152 ND-27,,,,,,Lisbon,58054,ND,~,2024-06-12 01:06:49.588129,2024-11-05 18:51:29.174693,9999-12-31 00:00:00.000000,9999-12-31 00:00:00.000000,Y
 121212121,517782585,517782585,2B19C89AA36,Heeler,Alex,Joey,1959-03-14,,~,1,4,14152 ND-27,,,,,,Lisbon,58054,ND,~,2024-06-12 01:06:49.588129,2024-11-05 18:51:29.174693,9999-12-31 00:00:00.000000,9999-12-31 00:00:00.000000,Y
 232323232,517782585,517782585,2B19C89AA37,Heeler,Alex,Joey,1959-03-14,,~,1,4,14152 ND-27,,,,,,Lisbon,58054,ND,~,2024-06-12 01:06:49.588129,2024-11-05 18:51:29.174693,9999-12-31 00:00:00.000000,9999-12-31 00:00:00.000000,Y
+617782586,617782585,617782585,2B19C89AA38,Heeler,Alex,Joey,1959-03-14,,~,1,4,14152 ND-27,,,,,,Lisbon,58054,ND,~,2024-06-12 01:06:49.588129,2024-11-05 18:51:29.174693,9999-12-31 00:00:00.000000,9999-12-31 00:00:00.000000,Y
+617782587,617782585,617782587,2B19C89AA38,Heeler,Alex,Joey,1959-03-14,,~,1,4,14152 ND-27,,,,,,Lisbon,58054,ND,~,2024-06-12 01:06:49.588129,2024-11-05 18:51:29.174693,9999-12-31 00:00:00.000000,9999-12-31 00:00:00.000000,Y

apps/bfd-server-ng/src/test/java/gov/cms/bfd/server/ng/CoverageSearchIT.java

@@ -14,7 +14,6 @@
 import ca.uhn.fhir.rest.gclient.TokenClientParam;
 import ca.uhn.fhir.rest.server.exceptions.InvalidRequestException;
 import gov.cms.bfd.server.ng.util.DateUtil;
-import jakarta.persistence.EntityManager;
 import java.time.ZonedDateTime;
 import java.util.Comparator;
 import java.util.stream.Stream;
@@ -25,12 +24,8 @@
 import org.junit.jupiter.params.provider.Arguments;
 import org.junit.jupiter.params.provider.EnumSource;
 import org.junit.jupiter.params.provider.MethodSource;
-import org.springframework.beans.factory.annotation.Autowired;
 
 public class CoverageSearchIT extends IntegrationTestBase {
-
-  @Autowired private EntityManager entityManager;
-
   private IQuery<Bundle> searchBundle() {
     return getFhirClient().search().forResource(Coverage.class).returnBundle(Bundle.class);
   }

apps/bfd-server-ng/src/test/java/gov/cms/bfd/server/ng/EobSearchIT.java

@@ -12,7 +12,6 @@
 import gov.cms.bfd.server.ng.util.DateUtil;
 import gov.cms.bfd.server.ng.util.IdrConstants;
 import gov.cms.bfd.server.ng.util.SystemUrls;
-import jakarta.persistence.EntityManager;
 import java.time.LocalDate;
 import java.time.ZonedDateTime;
 import org.hl7.fhir.r4.model.Bundle;
@@ -21,10 +20,8 @@
 import org.junit.jupiter.params.ParameterizedTest;
 import org.junit.jupiter.params.provider.EnumSource;
 import org.junit.jupiter.params.provider.ValueSource;
-import org.springframework.beans.factory.annotation.Autowired;
 
 public class EobSearchIT extends IntegrationTestBase {
-  @Autowired private EntityManager entityManager;
 
   private IQuery<Bundle> searchBundle() {
     return getFhirClient()

apps/bfd-server-ng/src/test/java/gov/cms/bfd/server/ng/IntegrationTestBase.java

@@ -4,13 +4,15 @@
 import au.com.origin.snapshots.junit5.SnapshotExtension;
 import ca.uhn.fhir.context.FhirContext;
 import ca.uhn.fhir.rest.client.api.IGenericClient;
+import jakarta.persistence.EntityManager;
 import java.util.List;
 import org.hl7.fhir.r4.model.Bundle;
 import org.hl7.fhir.r4.model.Coverage;
 import org.hl7.fhir.r4.model.DomainResource;
 import org.hl7.fhir.r4.model.Extension;
 import org.hl7.fhir.r4.model.Patient;
 import org.junit.jupiter.api.extension.ExtendWith;
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.boot.test.context.SpringBootTest;
 import org.springframework.boot.test.web.server.LocalServerPort;
 import org.springframework.context.annotation.Import;
@@ -23,6 +25,7 @@
 public class IntegrationTestBase {
   @LocalServerPort protected int port;
   protected Expect expect;
+  @Autowired protected EntityManager entityManager;
 
   protected static final String HISTORICAL_MERGED_BENE_SK = "848484848";
   protected static final String HISTORICAL_MERGED_BENE_SK2 = "121212121";
@@ -31,6 +34,9 @@ public class IntegrationTestBase {
   protected static final String HISTORICAL_MERGED_MBI_KILL_CREDIT = "2B19C89AA37";
   protected static final String HISTORICAL_MERGED_MBI = "2B19C89AA36";
   protected static final String HISTORICAL_AND_CURRENT_MBI = "2B19C89AA35";
+  protected static final String OVERSHARE_MBI = "2B19C89AA38";
+  protected static final String OVERSHARE_BENE_SK1 = "617782586";
+  protected static final String OVERSHARE_BENE_SK2 = "617782587";
 
   protected static final String BENE_ID_PART_A_ONLY = "178083966";
   protected static final String BENE_ID_PART_B_ONLY = "365359727";

apps/bfd-server-ng/src/test/java/gov/cms/bfd/server/ng/PatientReadIT.java

@@ -13,6 +13,7 @@
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.params.ParameterizedTest;
 import org.junit.jupiter.params.provider.EmptySource;
+import org.junit.jupiter.params.provider.ValueSource;
 
 public class PatientReadIT extends IntegrationTestBase {
   private IReadTyped<Patient> patientRead() {
@@ -96,6 +97,27 @@ void patientReadIdNotFound() {
     assertThrows(ResourceNotFoundException.class, readWithId::execute);
   }
 
+  @ValueSource(strings = {OVERSHARE_BENE_SK1, OVERSHARE_BENE_SK2})
+  @ParameterizedTest
+  void patientOvershareIdNotFound(String beneSk) {
+    var mbis =
+        entityManager
+            .createNativeQuery(
+                "SELECT bene_mbi_id FROM idr.beneficiary WHERE bene_sk = :beneSk", String.class)
+            .setParameter("beneSk", Long.parseLong(beneSk))
+            .getResultList();
+    assertEquals(1, mbis.size());
+    var overshare =
+        entityManager
+            .createNativeQuery(
+                "SELECT * FROM idr.beneficiary_overshare_mbi WHERE bene_mbi_id = :mbi")
+            .setParameter("mbi", mbis.getFirst())
+            .getResultList();
+    assertEquals(1, overshare.size());
+    var readWithBeneSk = patientRead().withId(beneSk);
+    assertThrows(ResourceNotFoundException.class, readWithBeneSk::execute);
+  }
+
   @Test
   void patientReadInvalidIdBadRequest() {
     var readWithId = patientRead().withId("abc");

apps/bfd-server-ng/src/test/java/gov/cms/bfd/server/ng/PatientSearchIT.java

@@ -13,7 +13,6 @@
 import ca.uhn.fhir.rest.server.exceptions.InvalidRequestException;
 import gov.cms.bfd.server.ng.util.DateUtil;
 import gov.cms.bfd.server.ng.util.SystemUrls;
-import jakarta.persistence.EntityManager;
 import java.time.ZonedDateTime;
 import java.util.Set;
 import java.util.stream.Collectors;
@@ -27,11 +26,8 @@
 import org.junit.jupiter.params.provider.EmptySource;
 import org.junit.jupiter.params.provider.EnumSource;
 import org.junit.jupiter.params.provider.MethodSource;
-import org.springframework.beans.factory.annotation.Autowired;
 
 public class PatientSearchIT extends IntegrationTestBase {
-  @Autowired private EntityManager entityManager;
-
   private IQuery<Bundle> searchBundle() {
     return getFhirClient().search().forResource(Patient.class).returnBundle(Bundle.class);
   }
@@ -263,6 +259,26 @@ void patientSearchByDate(ICriterion<TokenClientParam> searchCriteriaId, String b
     }
   }
 
+  @Test
+  void patientOvershareIdNotFound() {
+    var overshare =
+        entityManager
+            .createNativeQuery(
+                "SELECT * FROM idr.beneficiary_overshare_mbi WHERE bene_mbi_id = :mbi")
+            .setParameter("mbi", OVERSHARE_MBI)
+            .getResultList();
+    assertEquals(1, overshare.size());
+
+    var searchWithId =
+        searchBundle()
+            .where(
+                new TokenClientParam(Patient.SP_IDENTIFIER)
+                    .exactly()
+                    .systemAndIdentifier(SystemUrls.CMS_MBI, OVERSHARE_MBI))
+            .execute();
+    assertEquals(0, searchWithId.getEntry().size());
+  }
+
   @ParameterizedTest
   @EmptySource
   void patientSearchEmptyIdBadRequest(String id) {