test updated to old field

PrefectHQ · Apr 8, 2024 · 5c3c54b · 5c3c54b
1 parent e0335aa
commit 5c3c54b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/prefect_aws/workers/ecs_worker.py b/prefect_aws/workers/ecs_worker.py
@@ -107,7 +107,7 @@
   name: "{{ container_name }}"
 cpu: "{{ cpu }}"
 family: "{{ family }}"
-memory: "{{ task_memory }}"
+memory: "{{ memory }}"
 executionRoleArn: "{{ execution_role_arn }}"
 """
 

diff --git a/tests/workers/test_ecs_worker.py b/tests/workers/test_ecs_worker.py
@@ -535,7 +535,9 @@ async def test_launch_types(
 
 @pytest.mark.usefixtures("ecs_mocks")
 @pytest.mark.parametrize("launch_type", ["EC2", "FARGATE", "FARGATE_SPOT"])
-@pytest.mark.parametrize("cpu,memory", [(2048, 4096)])
+@pytest.mark.parametrize(
+    "cpu,memory", [(None, None), (1024, None), (None, 2048), (2048, 4096)]
+)
 @pytest.mark.parametrize("container_cpu,container_memory", [(1024, 2048)])
 async def test_cpu_and_memory(
     aws_credentials: AwsCredentials,
@@ -549,8 +551,8 @@ async def test_cpu_and_memory(
     configuration = await construct_configuration(
         aws_credentials=aws_credentials,
         launch_type=launch_type,
-        task_cpu=cpu,
-        task_memory=memory,
+        cpu=cpu,
+        memory=memory,
         container_cpu=container_cpu,
         container_memory=container_memory,
     )