filter tasks

Author: marcoacierno

backend/pycon/tasks.py

@@ -30,7 +30,11 @@ def launch_heavy_processing_worker():
     cluster_name = f"pythonit-{settings.ENVIRONMENT}"
     ecs_client = boto3.client("ecs", region_name=settings.AWS_REGION_NAME)
 
-    response = ecs_client.list_tasks(cluster=cluster_name, desiredStatus="RUNNING")
+    response = ecs_client.list_tasks(
+        cluster=cluster_name,
+        desiredStatus="RUNNING",
+        family=f"pythonit-{settings.ENVIRONMENT}-heavy-processing-worker",
+    )
 
     if len(response["taskArns"]) > 0:
         return

backend/pycon/tests/test_tasks.py

@@ -39,15 +39,17 @@ def test_launch_heavy_processing_worker_starts_task(settings, mocker):
     launch_heavy_processing_worker()
 
     mock_client.return_value.describe_tasks.assert_called_with(
-        cluster="pythonit-production-heavy-processing-worker", tasks=["arn-abc"]
+        cluster="pythonit-production", tasks=["arn-abc"]
     )
 
     mock_client.return_value.list_tasks.assert_called_with(
-        cluster="pythonit-production-heavy-processing-worker", desiredStatus="RUNNING"
+        cluster="pythonit-production",
+        desiredStatus="RUNNING",
+        family="pythonit-production-heavy-processing-worker",
     )
 
     mock_client.return_value.run_task.assert_called_with(
-        cluster="pythonit-production-heavy-processing-worker",
+        cluster="pythonit-production",
         taskDefinition="pythonit-production-heavy-processing-worker",
         count=1,
         networkConfiguration={

infrastructure/applications/pycon_backend/worker.tf

@@ -169,9 +169,7 @@ locals {
       value = jsonencode({
         subnets = [data.aws_subnet.public_1a.id],
         security_groups = [
-          data.aws_security_group.rds.id,
-          data.aws_security_group.lambda.id,
-          aws_security_group.instance.id
+          var.security_group_id
         ],
       })
     },