def test_register_task_definition(self, Session): """ the register_task_definition task should pass a dockerrun.aws.json template as kwargs to boto3.ecs.register_task_definition """ session = Session.return_value client = session.client.return_value with patch('mc.builders.ECSBuilder') as ECSBuilder: ecsbuild = ECSBuilder.return_value ecsbuild.render_template.return_value = '''{ "family": "unittest-family", "containerDefinitions": [], "volumes": [] }''' register_task_revision(ecsbuild) session.client.assert_called_with('ecs') client.register_task_definition.assert_called_with( family="unittest-family", containerDefinitions=[], volumes=[], ) register_task_revision('{"valid": "json"}') client.register_task_definition.assert_called_with(valid="json")
def run(self, task_definition, app=app): with app.app_context(): register_task_revision(task_definition)