def test_client(mock_ecs_client, mock_ecs_cluster): # pylint: disable=unused-argument
return client.ECSClient(
key_id="Testing",
access_key="Testing",
starter_client=mock_ecs_client,
starter_log_client=boto3.client("logs", region_name="us-east-2"),
)
def test_set_cluster():
log_client = boto3.client("logs", region_name="us-east-2")
ecs_client = boto3.client("ecs", region_name="us-east-2")
ecs_client.create_cluster(clusterName="test_clust")
testclient = client.ECSClient(
key_id="Testing",
access_key="Testing",
starter_client=ecs_client,
starter_log_client=log_client,
)
assert "test_clust" in testclient.cluster
def test_set_cluster():
log_client = boto3.client('logs', region_name='us-east-2')
ecs_client = boto3.client('ecs', region_name='us-east-2')
ecs_client.create_cluster(clusterName="test_clust")
testclient = client.ECSClient(
key_id='Testing',
access_key='Testing',
starter_client=ecs_client,
starter_log_client=log_client,
)
assert 'test_clust' in testclient.cluster
def test_one_run():
testclient = client.ECSClient()
testclient.set_and_register_task(
["echo start"], ["/bin/bash", "-c"], family="multimessage",
)
networkConfiguration = {
"awsvpcConfiguration": {
"subnets": ["subnet-0f3b1467",],
"securityGroups": ["sg-08627da7435350fa6",],
"assignPublicIp": "ENABLED",
}
}
testclient.run_task(networkConfiguration=networkConfiguration)
testclient.spin_all()
assert testclient.logs_messages == {0: ["start"]}
def test_entry():
log_client = boto3.client("logs", region_name="us-east-2")
ecs_client = boto3.client("ecs", region_name="us-east-2")
ecs_client.create_cluster(clusterName="test_ecs_cluster")
testclient = client.ECSClient(
key_id="Testing",
access_key="Testing",
starter_client=ecs_client,
starter_log_client=log_client,
)
testclient.set_and_register_task(
[" "], ["entries"], family=" ",
)
taskdef = ecs_client.describe_task_definition(taskDefinition=testclient.task_definition_arn)[
"taskDefinition"
]
assert taskdef["containerDefinitions"][0]["entryPoint"][0] == "entries"
def test_entry():
log_client = boto3.client('logs', region_name='us-east-2')
ecs_client = boto3.client('ecs', region_name='us-east-2')
ecs_client.create_cluster(clusterName="test_ecs_cluster")
testclient = client.ECSClient(
key_id='Testing',
access_key='Testing',
starter_client=ecs_client,
starter_log_client=log_client,
)
testclient.set_and_register_task(
[" "],
["entries"],
family=' ',
)
taskdef = ecs_client.describe_task_definition(
taskDefinition=testclient.task_definition_arn)['taskDefinition']
assert taskdef['containerDefinitions'][0]['entryPoint'][0] == "entries"
def test_region():
log_client = boto3.client("logs", region_name="us-east-2")
ecs_client = boto3.client("ecs", region_name="us-east-2")
ecs_client.create_cluster(clusterName="test_ecs_cluster")
testclient = client.ECSClient(
key_id="Testing",
access_key="Testing",
starter_client=ecs_client,
starter_log_client=log_client,
)
testclient.set_and_register_task(
["echoes"], [""], family="basefam",
)
taskdef = ecs_client.describe_task_definition(taskDefinition=testclient.task_definition_arn)[
"taskDefinition"
]
assert taskdef["family"] == "basefam"
logger = taskdef["containerDefinitions"][0]["logConfiguration"]["options"]
assert "us-east-2" in logger["awslogs-region"]
def test_region():
log_client = boto3.client('logs', region_name='us-east-2')
ecs_client = boto3.client('ecs', region_name='us-east-2')
ecs_client.create_cluster(clusterName="test_ecs_cluster")
testclient = client.ECSClient(
key_id='Testing',
access_key='Testing',
starter_client=ecs_client,
starter_log_client=log_client,
)
testclient.set_and_register_task(
["echoes"],
[""],
family='basefam',
)
taskdef = ecs_client.describe_task_definition(
taskDefinition=testclient.task_definition_arn)['taskDefinition']
assert taskdef['family'] == "basefam"
logger = taskdef['containerDefinitions'][0]['logConfiguration']['options']
assert 'us-east-2' in logger['awslogs-region']
def test_one_run():
testclient = client.ECSClient()
testclient.set_and_register_task(
["echo start"],
["/bin/bash", "-c"],
family='multimessage',
)
networkConfiguration = {
'awsvpcConfiguration': {
'subnets': [
'subnet-0f3b1467',
],
'securityGroups': [
'sg-08627da7435350fa6',
],
'assignPublicIp': 'ENABLED',
}
}
testclient.run_task(networkConfiguration=networkConfiguration)
testclient.spin_all()
assert testclient.logs_messages == {0: ['start']}
def test_full_run():
testclient = client.ECSClient()
testclient.set_and_register_task(
["echo $TEST; echo start; echo middle"],
["/bin/bash", "-c"],
family='multimessage',
)
networkConfiguration = {
'awsvpcConfiguration': {
'subnets': [
'subnet-0f3b1467',
],
'securityGroups': [
'sg-08627da7435350fa6',
],
'assignPublicIp': 'ENABLED',
}
}
testclient.run_task(networkConfiguration=networkConfiguration)
testclient.spin_til_done()
assert " ".join(testclient.logs_messages) == '\"end\" start middle'
