import boto3
from moto import mock_ecs
from get_number_of_tasks import get_clusters, get_services, get_list_of_tasks


@mock_ecs
def test_get_clusters():
    client = boto3.client('ecs', region_name='us-east-1')
    cluster = client.create_cluster(
        clusterName="test",
        capacityProviders=["FARGATE"]
    )

    clusters = get_clusters(client)
    assert len(clusters) == 1
    assert cluster["cluster"]["clusterArn"] == clusters[0]


@mock_ecs
def test_get_services():
    client = boto3.client('ecs', region_name='us-east-1')
    cluster = client.create_cluster(
        clusterName="test",
        capacityProviders=["FARGATE"]
    )
    task_def = client.register_task_definition(
        family="test",
        taskRoleArn="test",
        executionRoleArn="test",
        networkMode="awsvpc",
        containerDefinitions=[
            {
                "image": "hello",
                "name": "hello"
            }
        ]
    )
    service = client.create_service(
        cluster=cluster["cluster"]["clusterArn"],
        serviceName="test",
        taskDefinition=task_def["taskDefinition"]["taskDefinitionArn"],
        desiredCount=1,
        launchType="FARGATE",
        role="test"
    )

    services = get_services(client, cluster["cluster"]["clusterArn"])

    assert len(services) == 1
    assert service["service"]["serviceArn"] == services[0]


# @mock_ecs
# def test_get_list_of_tasks():
#     client = boto3.client('ecs', region_name='us-east-1')
#     cluster = client.create_cluster(
#         clusterName="test",
#         capacityProviders=["FARGATE"]
#     )
#     task_def = client.register_task_definition(
#         family="test",
#         taskRoleArn="test",
#         executionRoleArn="test",
#         networkMode="awsvpc",
#         containerDefinitions=[
#             {
#                 "image": "hello",
#                 "name": "hello"
#             }
#         ]
#     )
#     service = client.create_service(
#         cluster=cluster["cluster"]["clusterArn"],
#         serviceName="test",
#         taskDefinition=task_def["taskDefinition"]["taskDefinitionArn"],
#         desiredCount=2,
#         launchType="FARGATE",
#         role="test",
#         schedulingStrategy="DAEMON"
#     )
#
#     tasks = get_list_of_tasks(client,
#                               cluster["cluster"]["clusterArn"],
#                               service["service"]["serviceArn"]
#                               )
#
#     assert len(tasks) == 1