def test_get_running_tasks_count_new_revision(client, service, task_definition_revision_2): client.describe_tasks.return_value = RESPONSE_DESCRIBE_TASKS action = EcsAction(client, CLUSTER_NAME, SERVICE_NAME) service.set_task_definition(task_definition_revision_2) running_count = action.get_running_tasks_count(service, [TASK_ARN_1, TASK_ARN_2]) assert running_count == 0
def test_get_running_tasks_count(client, service): client.describe_tasks.return_value = RESPONSE_DESCRIBE_TASKS action = EcsAction(client, CLUSTER_NAME, SERVICE_NAME) running_count = action.get_running_tasks_count(service, [TASK_ARN_1, TASK_ARN_2]) assert running_count == 2