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