BFD-4182: Add more test cases to V3 load tests (#2756)

Co-authored-by: Austin Schey <austin.schey@forpeople.us>

Author: MahiFentaye

apps/utils/locust_tests/common/db_idr.py

@@ -43,6 +43,30 @@ def get_regression_bene_sks(uri: str, table_sample_pct=None) -> List[str]:
     return [str(r[0]) for r in _execute(uri, bene_query)]
 
 
+def get_regression_current_part_a_bene_sks(uri: str, table_sample_pct=None) -> List[str]:
+    """Retrieves a random list of beneficiary IDs
+
+    Args:
+        uri (str): Database URI
+
+    Returns:
+        List[str]: A list of synthetic beneficiary IDs used for the regression suites
+    """
+    bene_query = "SELECT b. bene_sk FROM idr.beneficiary  b join idr.beneficiary_third_party tp on b.bene_sk = tp.bene_sk where b. bene_sk = b. bene_xref_efctv_sk and tp.bene_tp_type_cd  = 'A' limit 1000"
+    return [str(r[0]) for r in _execute(uri, bene_query)]
+
+def get_regression_current_part_b_bene_sks(uri: str, table_sample_pct=None) -> List[str]:
+    """Retrieves a random list of beneficiary IDs
+
+    Args:
+        uri (str): Database URI
+
+    Returns:
+        List[str]: A list of synthetic beneficiary IDs used for the regression suites
+    """
+    bene_query = "SELECT b. bene_sk FROM idr.beneficiary  b join idr.beneficiary_third_party tp on b.bene_sk = tp.bene_sk where b. bene_sk = b. bene_xref_efctv_sk and tp.bene_tp_type_cd  = 'B' limit 1000"
+    return [str(r[0]) for r in _execute(uri, bene_query)]
+
 def get_regression_bene_mbis(uri: str, table_sample_pct=None) -> List[str]:
     """Retrieves a random list list of MBIs
 
@@ -54,3 +78,16 @@ def get_regression_bene_mbis(uri: str, table_sample_pct=None) -> List[str]:
     """
     bene_query = "SELECT bene_mbi_id FROM idr.beneficiary limit 1000"
     return [str(r[0]) for r in _execute(uri, bene_query)]
+
+
+def get_regression_claim_ids(uri: str, table_sample_pct=None) -> List[str]:
+    """Retrieves a random list list of clam Ids
+
+    Args:
+        uri (str): Database URI
+
+    Returns:
+        List[str]: A list of synthetic beneficiary IDs used for the regression suites
+    """
+    claim_query = "SELECT clm_uniq_id from idr.claim limit 1000"
+    return [str(r[0]) for r in _execute(uri, claim_query)]

apps/utils/locust_tests/v3/regression_suite.py

@@ -3,12 +3,17 @@
 from locust import events, tag, task
 from locust.env import Environment
 
+import logging
+
 from common import data, db_idr
 from common.bfd_user_base import BFDUserBase
 from common.locust_utils import is_distributed, is_locust_master
 
 master_bene_sks: Collection[str] = []
+master_bene_sks_part_a: Collection[str] = []
+master_bene_sks_part_b: Collection[str] = []
 master_bene_mbis: Collection[str] = []
+master_claim_ids: Collection[str] = []
 
 
 @events.test_start.add_listener
@@ -29,13 +34,35 @@ def _(environment: Environment, **kwargs):
         data_type_name="bene_ids",
     )
 
+    global master_bene_sks_part_a
+    master_bene_sks_part_a = data.load_from_parsed_opts(
+        environment.parsed_options,
+        db_idr.get_regression_current_part_a_bene_sks,
+        data_type_name="bene_ids_part_a",
+    )
+
+    global master_bene_sks_part_b
+    master_bene_sks_part_b = data.load_from_parsed_opts(
+        environment.parsed_options,
+        db_idr.get_regression_current_part_b_bene_sks,
+        data_type_name="bene_ids_part_b",
+    )
+
     global master_bene_mbis
     master_bene_mbis = data.load_from_parsed_opts(
         environment.parsed_options,
         db_idr.get_regression_bene_mbis,
         data_type_name="bene_mbis",
     )
 
+    global master_claim_ids
+    master_claim_ids = data.load_from_parsed_opts(
+        environment.parsed_options,
+        db_idr.get_regression_claim_ids,
+        data_type_name="claim_ids",
+    )
+
+logger = logging.getLogger(__name__)
 
 class RegressionV3User(BFDUserBase):
     """Regression test suite for V3 BFD Server endpoints.
@@ -47,12 +74,22 @@ class RegressionV3User(BFDUserBase):
 
     # Do we terminate the tests when a test runs out of data and paginated URLs?
     END_ON_NO_DATA = False
-    BENE_LAST_UPDATED = {"_lastUpdated": "gt2020-05-05"}
+    LAST_UPDATED_FILTER = {"_lastUpdated": "gt2020-05-05"}
+    SERVICE_DATE_RANGE = {"service-date": "ge2020-01-01"}
 
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
+        logger.info(f"Loaded {len(master_bene_sks)} bene_sks.")
+        logger.info(f"Loaded {len(master_bene_sks_part_a)} bene_sks_part_a.")
+        logger.info(f"Loaded {len(master_bene_sks_part_b)} bene_sks_part_b.")
+        logger.info(f"Loaded {len(master_bene_mbis)} bene_mbis.")
+        logger.info(f"Loaded {len(master_claim_ids)} claim_ids.")
+
         self.bene_sks = itertools.cycle(list(master_bene_sks))
+        self.bene_sks_part_a = itertools.cycle(list(master_bene_sks_part_a))
+        self.bene_sks_part_b = itertools.cycle(list(master_bene_sks_part_b))
         self.bene_mbis = itertools.cycle(list(master_bene_mbis))
+        self.claim_ids = itertools.cycle(list(master_claim_ids))
 
     @tag("patient", "patient_read_id")
     @task
@@ -114,7 +151,7 @@ def patient_search_mbi_last_updated(self):
                 "identifier": f"http://hl7.org/fhir/sid/us-mbi|{next(self.bene_mbis)}",
                 "_format": "application/fhir+json",
             }
-            | self.BENE_LAST_UPDATED,
+            | self.LAST_UPDATED_FILTER,
             name="/v3/fhir/Patient search by mbi with last updated",
         )
 
@@ -129,3 +166,156 @@ def patient_search_mbi_post(self):
             },
             name="/v3/fhir/Patient search by mbi (POST)",
         )
+
+
+    # ==========================================================
+    # Coverage Endpoint Tests
+    # ==========================================================
+
+    @tag("coverage", "coverage_read_id_part_a")
+    @task
+    def coverage_read_id_part_a(self):
+        """Coverage read by composite ID for Part A"""
+        coverage_id = f"part-a-{next(self.bene_sks_part_a)}"
+
+        self.run_task_by_parameters(
+            base_path=f"/v3/fhir/Coverage/{coverage_id}",
+            params={
+                "_format": "application/fhir+json",
+            },
+            name="/v3/fhir/Coverage read part A",
+        )
+
+    @tag("coverage", "coverage_read_id_part_b")
+    @task
+    def coverage_read_id_part_b(self):
+        """Coverage read by composite ID for Part B"""
+        coverage_id = f"part-b-{next(self.bene_sks_part_b)}"
+
+        self.run_task_by_parameters(
+            base_path=f"/v3/fhir/Coverage/{coverage_id}",
+            params={
+                "_format": "application/fhir+json",
+            },
+            name="/v3/fhir/Coverage read Part B",
+        )
+
+    @tag("coverage", "coverage_search_id")
+    @task
+    def coverage_search_id_part_b(self):
+        """Coverage search by composite ID for Part B"""
+        bene_sk = next(self.bene_sks)
+        coverage_id = f"part-b-{bene_sk}"
+        self.run_task_by_parameters(
+            base_path="/v3/fhir/Coverage",
+            params={
+                "_id": coverage_id,
+                "_format": "application/fhir+json",
+            },
+            name="/v3/fhir/Coverage search by id",
+        )
+
+    @tag("coverage", "coverage_search_beneficiary")
+    @task
+    def coverage_search_by_beneficiary(self):
+        """Coverage search by beneficiary ID"""
+        self.run_task_by_parameters(
+            base_path="/v3/fhir/Coverage",
+            params={
+                "beneficiary": f"Patient/{next(self.bene_sks)}",
+                "_format": "application/fhir+json",
+            },
+            name="/v3/fhir/Coverage search by beneficiary",
+        )
+
+    @tag("coverage", "coverage_search_beneficiary_post")
+    @task
+    def coverage_search_by_beneficiary_post(self):
+        self.run_task_by_parameters(
+            base_path="/v3/fhir/Coverage/_search",
+            body={
+                "beneficiary": f"Patient/{next(self.bene_sks)}",
+            },
+            name="/v3/fhir/Coverage search by beneficiary (POST)",
+        )
+
+
+    @tag("coverage", "coverage_search_last_updated")
+    @task
+    def coverage_search_last_updated(self):
+        """Coverage search by beneficiary with _lastUpdated filter."""
+        self.run_task_by_parameters(
+            base_path="/v3/fhir/Coverage",
+            params={
+                "beneficiary": f"Patient/{next(self.bene_sks)}",
+                "_format": "application/fhir+json",
+            } | self.LAST_UPDATED_FILTER,
+            name="/v3/fhir/Coverage search by beneficiary with last updated",
+        )
+
+#     ==========================================================
+#     ExplanationOfBenefit (EOB) Endpoint Tests
+#     ==========================================================
+
+    @tag("eob", "eob_read")
+    @task
+    def eob_read(self):
+        """ExplanationOfBenefit read by its unique claim ID."""
+
+        self.run_task_by_parameters(
+            base_path=f"/v3/fhir/ExplanationOfBenefit/{next(self.claim_ids)}",
+            params={
+            "_format": "application/fhir+json"
+            },
+            name="/v3/fhir/ExplanationOfBenefit read",
+        )
+
+
+    @tag("eob", "eob_search_id")
+    @task
+    def eob_search_id(self):
+        """ExplanationOfBenefit search by its unique claim ID."""
+        claim_id = next(self.claim_ids)
+
+        self.run_task_by_parameters(
+            base_path="/v3/fhir/ExplanationOfBenefit",
+            params={"_id": claim_id, "_format": "application/fhir+json"},
+            name="/v3/fhir/ExplanationOfBenefit search by id",
+        )
+
+    @tag("eob", "eob_search_patient")
+    @task
+    def eob_search_patient(self):
+        """ExplanationOfBenefit search by patient ID (bene_sk)."""
+
+        self.run_task_by_parameters(
+            base_path="/v3/fhir/ExplanationOfBenefit",
+            params={"patient": next(self.bene_sks), "_format": "application/fhir+json"},
+            name="/v3/fhir/ExplanationOfBenefit search by patient",
+        )
+
+    @tag("eob", "eob_search_service_date")
+    @task
+    def eob_search_service_date(self):
+        """ExplanationOfBenefit search by patient and service-date."""
+        self.run_task_by_parameters(
+            base_path="/v3/fhir/ExplanationOfBenefit",
+            params={
+                "patient": next(self.bene_sks),
+                "_format": "application/fhir+json",
+            } | self.SERVICE_DATE_RANGE,
+            name="/v3/fhir/ExplanationOfBenefit search by service-date",
+        )
+
+    @tag("eob", "eob_search_last_updated")
+    @task
+    def eob_search_last_updated(self):
+        """ExplanationOfBenefit search by patient with _lastUpdated."""
+        self.run_task_by_parameters(
+            base_path="/v3/fhir/ExplanationOfBenefit",
+            params={
+                "patient": next(self.bene_sks),
+                "_format": "application/fhir+json",
+            } | self.LAST_UPDATED_FILTER,
+            name="/v3/fhir/ExplanationOfBenefit search by patient with last updated",
+        )