chore: cleanup and optimise test imports

Author: igorlg

unicorn_properties/src/properties_service/contract_status_changed_event_handler.py

@@ -9,8 +9,7 @@
 from aws_lambda_powertools.metrics import Metrics
 from aws_lambda_powertools.tracing import Tracer
 from aws_lambda_powertools.event_handler.exceptions import InternalServerError
-from schema.unicorn_contracts.contractstatuschanged import (
-    AWSEvent, ContractStatusChanged, Marshaller)
+from schema.unicorn_contracts.contractstatuschanged import AWSEvent, ContractStatusChanged, Marshaller
 
 # Initialise Environment variables
 if (SERVICE_NAMESPACE := os.environ.get("SERVICE_NAMESPACE")) is None:

unicorn_properties/tests/unit/test_contract_exists_checker_function.py

@@ -2,21 +2,15 @@
 # SPDX-License-Identifier: MIT-0
 
 import os
-# import json
 from importlib import reload
 
-# import pytest
+import pytest
 from unittest import mock
-from moto import mock_stepfunctions
-
-# from aws_lambda_powertools.event_handler.exceptions import InternalServerError
-# from botocore.exceptions import ClientError
 
 from .lambda_context import LambdaContext
 from .helper import load_event, return_env_vars_dict, create_ddb_table_contracts_with_entry
 
 
-@mock_stepfunctions
 @mock.patch.dict(os.environ, return_env_vars_dict(), clear=True)
 def test_handle_contract_exists_checker_function(dynamodb, mocker):
     stepfunctions_event = load_event('tests/events/lambda/contract_exists_checker.json')

unicorn_properties/tests/unit/test_contract_status_changed_event_handler.py

@@ -2,15 +2,11 @@
 # SPDX-License-Identifier: MIT-0
 
 import os
-# import json
 from importlib import reload
 
-# import pytest
+import pytest
 from unittest import mock
-# from moto import mock_dynamodb, mock_events
-
-# from aws_lambda_powertools.event_handler.exceptions import InternalServerError
-# from botocore.exceptions import ClientError
+from botocore.exceptions import ClientError
 
 from .lambda_context import LambdaContext
 from .helper import load_event, return_env_vars_dict, create_ddb_table_properties