def test_find_required_key_error_missing_key(self): invalid_properties_dict = {'name': 'some-lambda-function'} test_key = 'taskDefinition' with self.assertRaises(MissingPropertyError): find_required_key('task definition', invalid_properties_dict, test_key)
def test_find_required_key_error_missing_key(self): invalid_properties_dict = { 'name': 'some-lambda-function' } test_key = 'taskDefinition' with self.assertRaises(MissingPropertyError): find_required_key('task definition', invalid_properties_dict, test_key)
def update_task_def_arn(self, new_arn): """ Inserts the ARN of the previously created ECS task definition into the provided appspec. Expected format of ECS appspec (YAML) is: version: 0.0 resources: - <service-name>: type: AWS::ECS::Service properties: taskDefinition: <value> # replace this loadBalancerInfo: containerName: <value> containerPort: <value> """ appspec_obj = self._appspec_dict resources_key = filehelpers.find_required_key('codedeploy-appspec', appspec_obj, 'resources') updated_resources = [] # 'resources' is a list of string:obj dictionaries for resource in appspec_obj[resources_key]: for name in resource: # get content of resource resource_content = resource[name] # get resource properties properties_key = filehelpers.find_required_key( name, resource_content, 'properties') properties_content = resource_content[properties_key] # find task definition property task_def_key = filehelpers.find_required_key( properties_key, properties_content, 'taskDefinition') # insert new task def ARN into resource properties_content[task_def_key] = new_arn updated_resources.append(resource) appspec_obj[resources_key] = updated_resources self._appspec_dict = appspec_obj
def update_task_def_arn(self, new_arn): """ Inserts the ARN of the previously created ECS task definition into the provided appspec. Expected format of ECS appspec (YAML) is: version: 0.0 resources: - <service-name>: type: AWS::ECS::Service properties: taskDefinition: <value> # replace this loadBalancerInfo: containerName: <value> containerPort: <value> """ appspec_obj = self._appspec_dict resources_key = filehelpers.find_required_key( 'codedeploy-appspec', appspec_obj, 'resources') updated_resources = [] # 'resources' is a list of string:obj dictionaries for resource in appspec_obj[resources_key]: for name in resource: # get content of resource resource_content = resource[name] # get resource properties properties_key = filehelpers.find_required_key( name, resource_content, 'properties') properties_content = resource_content[properties_key] # find task definition property task_def_key = filehelpers.find_required_key( properties_key, properties_content, 'taskDefinition') # insert new task def ARN into resource properties_content[task_def_key] = new_arn updated_resources.append(resource) appspec_obj[resources_key] = updated_resources self._appspec_dict = appspec_obj
def test_find_required_key(self): test_properties_dict = { "TaskDefinition": "arn:aws:ecs:::task-definition/test:1", "loadbalancerInfo": { "containerName": "web", "containerPort": 80 } } test_key = 'taskDefinition' expected_result = 'TaskDefinition' result = find_required_key( 'task definition', test_properties_dict, test_key) self.assertEqual(result, expected_result)
def test_find_required_key(self): test_properties_dict = { "TaskDefinition": "arn:aws:ecs:::task-definition/test:1", "loadbalancerInfo": { "containerName": "web", "containerPort": 80 } } test_key = 'taskDefinition' expected_result = 'TaskDefinition' result = find_required_key('task definition', test_properties_dict, test_key) self.assertEqual(result, expected_result)
def test_find_required_key_error_empty_object(self): test_key = 'taskDefinition' with self.assertRaises(MissingPropertyError): find_required_key('task definition', dict(), test_key)
def test_find_required_key_error_empty_object(self): test_key = 'taskDefinition' with self.assertRaises(MissingPropertyError): find_required_key('task definition', dict(), test_key)