diff --git a/tests/test_start_execution.py b/tests/test_start_execution.py index 02fbb8274..85a96026d 100644 --- a/tests/test_start_execution.py +++ b/tests/test_start_execution.py @@ -185,7 +185,6 @@ def test_lambda_handler_500_jobs(): with ( patch('dynamo.jobs.get_jobs_waiting_for_execution') as mock_get_jobs_waiting_for_execution, patch('start_execution.submit_jobs') as mock_submit_jobs, - patch.dict(os.environ, {'START_EXECUTION_WORKER_ARN': 'test-worker-function-arn'}, clear=True), patch.dict(os.environ, {'STEP_FUNCTION_ARN': 'test-step-function-arn'}, clear=True), ): mock_jobs = list(range(500)) @@ -204,7 +203,6 @@ def test_lambda_handler_400_jobs(): with ( patch('dynamo.jobs.get_jobs_waiting_for_execution') as mock_get_jobs_waiting_for_execution, patch('start_execution.submit_jobs') as mock_submit_jobs, - patch.dict(os.environ, {'START_EXECUTION_WORKER_ARN': 'test-worker-function-arn'}, clear=True), patch.dict(os.environ, {'STEP_FUNCTION_ARN': 'test-step-function-arn'}, clear=True), ): mock_jobs = list(range(400)) @@ -223,7 +221,6 @@ def test_lambda_handler_50_jobs(): with ( patch('dynamo.jobs.get_jobs_waiting_for_execution') as mock_get_jobs_waiting_for_execution, patch('start_execution.submit_jobs') as mock_submit_jobs, - patch.dict(os.environ, {'START_EXECUTION_WORKER_ARN': 'test-worker-function-arn'}, clear=True), patch.dict(os.environ, {'STEP_FUNCTION_ARN': 'test-step-function-arn'}, clear=True), ): mock_jobs = list(range(50)) @@ -242,7 +239,6 @@ def test_lambda_handler_no_jobs(): with ( patch('dynamo.jobs.get_jobs_waiting_for_execution') as mock_get_jobs_waiting_for_execution, patch('start_execution.submit_jobs') as mock_submit_jobs, - patch.dict(os.environ, {'START_EXECUTION_WORKER_ARN': 'test-worker-function-arn'}, clear=True), patch.dict(os.environ, {'STEP_FUNCTION_ARN': 'test-step-function-arn'}, clear=True), ): mock_get_jobs_waiting_for_execution.return_value = []