def test_parse_task_definition_kwargs():
environment = FargateTaskEnvironment()
kwarg_dict = {
"family": "test",
"taskRoleArn": "test",
"executionRoleArn": "test",
"networkMode": "test",
"containerDefinitions": "test",
"volumes": "test",
"placementConstraints": "test",
"requiresCompatibilities": "test",
"cpu": "test",
"memory": "test",
"tags": "test",
"pidMode": "test",
"ipcMode": "test",
"proxyConfiguration": "test",
"inferenceAccelerators": "test",
}
task_definition_kwargs, task_run_kwargs = environment._parse_kwargs(kwarg_dict)
assert task_definition_kwargs == kwarg_dict
assert task_run_kwargs == {"placementConstraints": "test", "tags": "test"}
def test_parse_task_run_kwargs():
environment = FargateTaskEnvironment()
kwarg_dict = {
"cluster": "test",
"taskDefinition": "test",
"count": "test",
"startedBy": "test",
"group": "test",
"placementConstraints": "test",
"placementStrategy": "test",
"platformVersion": "test",
"networkConfiguration": "test",
"tags": "test",
"enableECSManagedTags": "test",
"propagateTags": "test",
}
task_definition_kwargs, task_run_kwargs = environment._parse_kwargs(
kwarg_dict)
assert task_run_kwargs == kwarg_dict
assert task_definition_kwargs == {
"placementConstraints": "test",
"tags": "test"
}
def test_parse_task_kwargs_invalid_value_removed():
environment = FargateTaskEnvironment()
kwarg_dict = {"test": "not_real"}
task_definition_kwargs, task_run_kwargs = environment._parse_kwargs(kwarg_dict)
assert task_definition_kwargs == {}
assert task_run_kwargs == {}
def test_parse_task_definition_and_run_kwargs():
environment = FargateTaskEnvironment()
def_kwarg_dict = {
"family": "test",
"taskRoleArn": "test",
"executionRoleArn": "test",
"networkMode": "test",
"containerDefinitions": "test",
"volumes": "test",
"placementConstraints": "test",
"requiresCompatibilities": "test",
"cpu": "test",
"memory": "test",
"tags": "test",
"pidMode": "test",
"ipcMode": "test",
"proxyConfiguration": "test",
"inferenceAccelerators": "test",
}
run_kwarg_dict = {
"cluster": "test",
"taskDefinition": "test",
"count": "test",
"startedBy": "test",
"group": "test",
"placementConstraints": "test",
"placementStrategy": "test",
"platformVersion": "test",
"networkConfiguration": "test",
"tags": "test",
"enableECSManagedTags": "test",
"propagateTags": "test",
}
kwarg_dict = {
"family": "test",
"taskRoleArn": "test",
"executionRoleArn": "test",
"networkMode": "test",
"containerDefinitions": "test",
"volumes": "test",
"placementConstraints": "test",
"requiresCompatibilities": "test",
"cpu": "test",
"memory": "test",
"tags": "test",
"pidMode": "test",
"ipcMode": "test",
"proxyConfiguration": "test",
"inferenceAccelerators": "test",
"cluster": "test",
"taskDefinition": "test",
"count": "test",
"startedBy": "test",
"group": "test",
"placementStrategy": "test",
"platformVersion": "test",
"networkConfiguration": "test",
"enableECSManagedTags": "test",
"propagateTags": "test",
}
task_definition_kwargs, task_run_kwargs = environment._parse_kwargs(kwarg_dict)
assert task_definition_kwargs == def_kwarg_dict
assert task_run_kwargs == run_kwarg_dict
