def test_merge_run_task_kwargs_top_level(self):
opt1 = {"cluster": "testing", "launchType": "FARGATE"}
opt2 = {"cluster": "new", "enableECSManagedTags": False}
assert merge_run_task_kwargs(opt1, {}) == opt1
assert merge_run_task_kwargs({}, opt1) == opt1
assert merge_run_task_kwargs(opt1, opt2) == {
"cluster": "new",
"launchType": "FARGATE",
"enableECSManagedTags": False,
}
def test_merge_run_task_kwargs_overrides(self):
opt1 = {"overrides": {"cpu": "1024", "memory": "2048"}}
opt2 = {"overrides": {"cpu": "2048", "taskRoleArn": "testing"}}
assert merge_run_task_kwargs(opt1, {}) == opt1
assert merge_run_task_kwargs(opt1, {"overrides": {}}) == opt1
assert merge_run_task_kwargs({}, opt1) == opt1
assert merge_run_task_kwargs({"overrides": {}}, opt1) == opt1
assert merge_run_task_kwargs(opt1, opt2) == {
"overrides": {"cpu": "2048", "memory": "2048", "taskRoleArn": "testing"}
}
def test_merge_run_task_kwargs_top_level_capacity_provider(self):
opt1 = {"cluster": "testing", "launchType": "FARGATE"}
opt2 = {
"cluster": "new",
"capacityProviderStrategy": [{"capacityProvider": "FARGATE_SPOT"}],
}
assert merge_run_task_kwargs(opt1, opt2) == {
"capacityProviderStrategy": [{"capacityProvider": "FARGATE_SPOT"}],
"cluster": "new",
}
assert merge_run_task_kwargs(opt2, opt1) == {
"launchType": "FARGATE",
"cluster": "testing",
}
def test_merge_run_task_kwargs_container_overrides(self):
opt1 = {}
opt2 = {"overrides": {}}
opt3 = {"overrides": {"containerOverrides": []}}
opt4 = {
"overrides": {
"taskRoleArn": "my-task-role",
"containerOverrides": [{"name": "a", "cpu": 1, "memory": 2}],
}
}
opt5 = {
"overrides": {
"executionRoleArn": "my-ex-role",
"containerOverrides": [{"name": "a", "cpu": 3, "memoryReservation": 4}],
}
}
opt6 = {
"overrides": {
"executionRoleArn": "my-ex-role",
"containerOverrides": [{"name": "b", "cpu": 5}],
}
}
assert merge_run_task_kwargs(opt4, opt1) == opt4
assert merge_run_task_kwargs(opt4, opt2) == opt4
assert merge_run_task_kwargs(opt4, opt3) == opt4
assert merge_run_task_kwargs(opt1, opt4) == opt4
assert merge_run_task_kwargs(opt2, opt4) == opt4
assert merge_run_task_kwargs(opt3, opt4) == opt4
assert merge_run_task_kwargs(opt4, opt5) == {
"overrides": {
"taskRoleArn": "my-task-role",
"executionRoleArn": "my-ex-role",
"containerOverrides": [
{"name": "a", "cpu": 3, "memory": 2, "memoryReservation": 4}
],
}
}
assert merge_run_task_kwargs(opt4, opt6) == {
"overrides": {
"taskRoleArn": "my-task-role",
"executionRoleArn": "my-ex-role",
"containerOverrides": [
{"name": "a", "cpu": 1, "memory": 2},
{"name": "b", "cpu": 5},
],
}
}
def test_merge_run_task_kwargs_no_op(self):
assert merge_run_task_kwargs({}, {}) == {}
