Skip to content

Commit 61fd2e1

Browse files
refactor: reuse Lambda client
1 parent b73afbd commit 61fd2e1

1 file changed

Lines changed: 9 additions & 4 deletions

File tree

tests/lambda_service_test.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,6 @@
3737
WaitOptions,
3838
)
3939

40-
4140
# =============================================================================
4241
# Fixtures
4342
# =============================================================================
@@ -1922,7 +1921,9 @@ def test_lambda_client_constructor():
19221921

19231922
@patch.dict("os.environ", {}, clear=True)
19241923
@patch("boto3.client")
1925-
def test_lambda_client_initialize_client_default(mock_boto_client, reset_lambda_client_cache):
1924+
def test_lambda_client_initialize_client_default(
1925+
mock_boto_client, reset_lambda_client_cache
1926+
):
19261927
"""Test LambdaClient.initialize_client with default endpoint."""
19271928
mock_client = Mock()
19281929
mock_boto_client.return_value = mock_client
@@ -1942,7 +1943,9 @@ def test_lambda_client_initialize_client_default(mock_boto_client, reset_lambda_
19421943

19431944
@patch.dict("os.environ", {"AWS_ENDPOINT_URL_LAMBDA": "http://localhost:3000"})
19441945
@patch("boto3.client")
1945-
def test_lambda_client_initialize_client_with_endpoint(mock_boto_client, reset_lambda_client_cache):
1946+
def test_lambda_client_initialize_client_with_endpoint(
1947+
mock_boto_client, reset_lambda_client_cache
1948+
):
19461949
"""Test LambdaClient.initialize_client with custom endpoint (boto3 handles it automatically)."""
19471950
mock_client = Mock()
19481951
mock_boto_client.return_value = mock_client
@@ -2020,7 +2023,9 @@ def test_checkpoint_error_handling():
20202023

20212024
@patch.dict("os.environ", {}, clear=True)
20222025
@patch("boto3.client")
2023-
def test_lambda_client_initialize_client_no_endpoint(mock_boto_client, reset_lambda_client_cache):
2026+
def test_lambda_client_initialize_client_no_endpoint(
2027+
mock_boto_client, reset_lambda_client_cache
2028+
):
20242029
"""Test LambdaClient.initialize_client without AWS_ENDPOINT_URL_LAMBDA."""
20252030
mock_client = Mock()
20262031
mock_boto_client.return_value = mock_client

0 commit comments

Comments
 (0)