@unittest.skipif

Author: nhsdevws

e2e/test_authorization.py

@@ -1,5 +1,5 @@
 import uuid
-import sys
+import unittest
 from typing import Set
 
 from lib.apigee import ApigeeApp
@@ -19,11 +19,9 @@
 from utils.constants import env_internal_dev
 
 
+@unittest.skipIf(env_internal_dev, "Skip tests for internal-dev environment")
 class TestApplicationRestrictedAuthorization(ImmunizationBaseTest):
 
-    if env_internal_dev:
-        sys.exit()
-
     my_app: ApigeeApp
     my_imms_api: ImmunisationApi
 

e2e/test_create_immunization.py

@@ -1,14 +1,12 @@
-import sys
+import unittest
 from utils.base_test import ImmunizationBaseTest
 from utils.resource import generate_imms_resource, get_full_row_from_identifier
 from utils.constants import env_internal_dev
 
 
+@unittest.skipIf(env_internal_dev, "Skip tests for internal-dev environment")
 class TestCreateImmunization(ImmunizationBaseTest):
 
-    if env_internal_dev:
-        sys.exit()
-
     def test_create_imms(self):
         """it should create a FHIR Immunization resource"""
         for imms_api in self.imms_apis:

e2e/test_delete_immunization.py

@@ -1,15 +1,13 @@
-import sys
+import unittest
 from utils.base_test import ImmunizationBaseTest
 from utils.immunisation_api import parse_location
 from utils.resource import generate_imms_resource
 from utils.constants import env_internal_dev
 
 
+@unittest.skipIf(env_internal_dev, "Skip tests for internal-dev environment")
 class TestDeleteImmunization(ImmunizationBaseTest):
 
-    if env_internal_dev:
-        sys.exit()
-
     def test_delete_imms(self):
         """it should delete a FHIR Immunization resource"""
         for imms_api in self.imms_apis:

e2e/test_delta_immunization.py

@@ -9,14 +9,12 @@
 import boto3
 from botocore.config import Config
 import os
-import sys
+import unittest
 
 
+@unittest.skipIf(env_internal_dev, "Skip tests for internal-dev environment")
 class TestDeltaImmunization(ImmunizationBaseTest):
 
-    if env_internal_dev:
-        sys.exit()
-
     CREATE_OPERATION = "CREATE"
     UPDATE_OPERATION = "UPDATE"
     DELETE_OPERATION = "DELETE"

e2e/test_get_immunization.py

@@ -1,6 +1,6 @@
 from decimal import Decimal
 import uuid
-import sys
+import unittest
 
 from utils.base_test import ImmunizationBaseTest
 from utils.immunisation_api import parse_location
@@ -9,11 +9,9 @@
 from utils.constants import env_internal_dev
 
 
+@unittest.skipIf(env_internal_dev, "Skip tests for internal-dev environment")
 class TestGetImmunization(ImmunizationBaseTest):
 
-    if env_internal_dev:
-        sys.exit()
-
     def test_get_imms(self):
         """it should get a FHIR Immunization resource"""
         for imms_api in self.imms_apis:

e2e/test_proxy.py

@@ -2,18 +2,16 @@
 import subprocess
 import unittest
 import uuid
-import sys
+import unittest
 import requests
 
 from lib.env import get_service_base_path, get_status_endpoint_api_key
 from utils.constants import env_internal_dev
 
 
+@unittest.skipIf(env_internal_dev, "Skip tests for internal-dev environment")
 class TestProxyHealthcheck(unittest.TestCase):
 
-    if env_internal_dev:
-        sys.exit()
-
     proxy_url: str
     status_api_key: str
 
@@ -36,12 +34,10 @@ def test_status(self):
                          f"service is not healthy: status: {body['status']}")
 
 
+@unittest.skipIf(env_internal_dev, "Skip tests for internal-dev environment")
 class TestMtls(unittest.TestCase):
     """Our backend is secured using mTLS. This test makes sure you can't hit the backend directly"""
 
-    if env_internal_dev:
-        sys.exit()
-
     def test_mtls(self):
         """backend should reject unauthorized connections"""
         backend_url = TestMtls.get_backend_url()
@@ -76,13 +72,11 @@ def get_backend_url() -> str:
             raise RuntimeError(f"Failed to run command\n{e}")
 
 
+@unittest.skipIf(env_internal_dev, "Skip tests for internal-dev environment")
 class TestProxyAuthorization(unittest.TestCase):
     """Our apigee proxy has its own authorization.
     This class test different authorization access levels/roles authentication types that are supported"""
 
-    if env_internal_dev:
-        sys.exit()
-
     proxy_url: str
 
     @classmethod

e2e/test_search_immunization.py

@@ -1,6 +1,6 @@
 import pprint
 import uuid
-import sys
+import unittest
 from typing import NamedTuple, Literal, Optional, List
 from decimal import Decimal
 
@@ -11,14 +11,12 @@
 from utils.constants import env_internal_dev
 
 
+@unittest.skipIf(env_internal_dev, "Skip tests for internal-dev environment")
 class TestSearchImmunization(ImmunizationBaseTest):
     # NOTE: In each test, the result may contain more hits. We only assert if the resource that we created is
     #  in the result set and assert the one that we don't expect is not present.
     # This is to make these tests stateless otherwise; we need to clean up the db after each test
 
-    if env_internal_dev:
-        sys.exit()
-
     def store_records(self, *resources):
         ids = []
         for res in resources:

e2e/test_sqs_dlq.py

@@ -2,18 +2,16 @@
 import json
 import boto3
 import os
-import sys
+import unittest
 from utils.delete_sqs_messages import read_and_delete_messages
 from utils.get_sqs_url import get_queue_url
 from utils.constants import env_internal_dev
 from botocore.exceptions import ClientError  # Handle potential errors
 
 
+@unittest.skipIf(env_internal_dev, "Skip tests for internal-dev environment")
 class TestSQS(unittest.TestCase):
 
-    if env_internal_dev:
-        sys.exit()
-
     def setUp(self):
         # Get SQS queue url
         self.queue_name = os.environ["AWS_SQS_QUEUE_NAME"]

e2e/test_update_immunization.py

@@ -1,17 +1,15 @@
 import copy
 import uuid
-import sys
+import unittest
 from utils.base_test import ImmunizationBaseTest
 from utils.immunisation_api import parse_location
 from utils.resource import generate_imms_resource
 from utils.constants import env_internal_dev
 
 
+@unittest.skipIf(env_internal_dev, "Skip tests for internal-dev environment")
 class TestUpdateImmunization(ImmunizationBaseTest):
 
-    if env_internal_dev:
-        sys.exit()
-
     def test_update_imms(self):
         """it should update a FHIR Immunization resource"""
         for imms_api in self.imms_apis:

e2e_batch/test_e2e_batch.py

@@ -1,5 +1,4 @@
 import time
-import sys
 import unittest
 from utils import (
     generate_csv,
@@ -25,11 +24,9 @@
 )
 
 
+@unittest.skipIf(env_internal_dev, "Skip tests for internal-dev environment")
 class TestE2EBatch(unittest.TestCase):
 
-    if env_internal_dev:
-        sys.exit()
-
     if env_value != "ref":
 
         def test_create_success(self):