def test_parse_task_run_kwargs(monkeypatch, runner_token):
boto3_client = MagicMock()
monkeypatch.setattr("boto3.client", boto3_client)
agent = FargateAgent()
kwarg_dict = {
"cluster": "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 = agent._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(monkeypatch, runner_token):
boto3_client = MagicMock()
monkeypatch.setattr("boto3.client", boto3_client)
agent = FargateAgent()
kwarg_dict = {"test": "not_real"}
task_definition_kwargs, task_run_kwargs = agent._parse_kwargs(kwarg_dict)
assert task_definition_kwargs == {}
assert task_run_kwargs == {}
def test_parse_task_definition_kwargs_errors(monkeypatch, runner_token):
boto3_client = MagicMock()
monkeypatch.setattr("boto3.client", boto3_client)
agent = FargateAgent()
kwarg_dict = {
"placementConstraints": "taskRoleArn='arn:aws:iam::543216789012:role/Dev",
}
task_definition_kwargs, task_run_kwargs = agent._parse_kwargs(kwarg_dict)
assert task_definition_kwargs == kwarg_dict
assert task_run_kwargs == {
"placementConstraints": "taskRoleArn='arn:aws:iam::543216789012:role/Dev"
}
def test_parse_task_definition_kwargs(monkeypatch, runner_token):
boto3_client = MagicMock()
monkeypatch.setattr("boto3.client", boto3_client)
agent = FargateAgent()
kwarg_dict = {
"taskRoleArn": "test",
"executionRoleArn": "test",
"volumes": "test",
"placementConstraints": "test",
"cpu": "test",
"memory": "test",
"tags": "test",
"pidMode": "test",
"ipcMode": "test",
"proxyConfiguration": "test",
"inferenceAccelerators": "test",
}
task_definition_kwargs, task_run_kwargs = agent._parse_kwargs(kwarg_dict)
assert task_definition_kwargs == kwarg_dict
assert task_run_kwargs == {"placementConstraints": "test", "tags": "test"}
def test_parse_task_definition_and_run_kwargs(monkeypatch, runner_token):
boto3_client = MagicMock()
monkeypatch.setattr("boto3.client", boto3_client)
agent = FargateAgent()
def_kwarg_dict = {
"taskRoleArn": "test",
"executionRoleArn": "test",
"volumes": "test",
"placementConstraints": "test",
"cpu": "test",
"memory": "test",
"tags": "test",
"pidMode": "test",
"ipcMode": "test",
"proxyConfiguration": "test",
"inferenceAccelerators": "test",
}
run_kwarg_dict = {
"cluster": "test",
"count": "test",
"startedBy": "test",
"group": "test",
"placementConstraints": "test",
"placementStrategy": "test",
"platformVersion": "test",
"networkConfiguration": "test",
"tags": "test",
"enableECSManagedTags": "test",
"propagateTags": "test",
}
kwarg_dict = {
"taskRoleArn": "test",
"executionRoleArn": "test",
"volumes": "test",
"placementConstraints": "test",
"cpu": "test",
"memory": "test",
"tags": "test",
"pidMode": "test",
"ipcMode": "test",
"proxyConfiguration": "test",
"inferenceAccelerators": "test",
"cluster": "test",
"count": "test",
"startedBy": "test",
"group": "test",
"placementStrategy": "test",
"platformVersion": "test",
"networkConfiguration": "test",
"enableECSManagedTags": "test",
"propagateTags": "test",
}
task_definition_kwargs, task_run_kwargs = agent._parse_kwargs(kwarg_dict)
assert task_definition_kwargs == def_kwarg_dict
assert task_run_kwargs == run_kwarg_dict
