def inject_commands(command_table, session, **kwargs):
"""
Inject custom 'aws deploy' commands.
"""
command_table['push'] = Push(session)
command_table['register'] = Register(session)
command_table['deregister'] = Deregister(session)
command_table['install'] = Install(session)
command_table['uninstall'] = Uninstall(session)
def inject_commands(command_table, session, **kwargs):
"""
Inject custom 'aws deploy' commands.
"""
command_table['push'] = Push(session)
def setUp(self):
self.application_name = 'MyApp'
self.description = 'MyApp revision'
self.source = '/tmp'
self.appspec = 'appspec.yml'
self.appspec_path = '{0}/{1}'.format(self.source, self.appspec)
self.bucket = 'foo'
self.key = 'bar/baz.zip'
self.s3_location = 's3://' + self.bucket + '/' + self.key
self.eTag = '"1a2b3cd45e"'
self.version_id = '12341234-1234-1234-1234-123412341234'
self.upload_id = 'upload_id'
self.region = 'us-east-1'
self.endpoint_url = 'https://codedeploy.aws.amazon.com'
self.args = Namespace()
self.args.application_name = self.application_name
self.args.s3_location = self.s3_location
self.args.ignore_hidden_files = False
self.args.no_ignore_hidden_files = False
self.args.description = self.description
self.args.source = self.source
self.globals = Namespace()
self.globals.region = self.region
self.globals.endpoint_url = self.endpoint_url
self.globals.verify_ssl = False
self.upload_response = {
'ETag': self.eTag,
'VersionId': self.version_id
}
self.revision = {
'revisionType': 'S3',
's3Location': {
'bucket': self.bucket,
'key': self.key,
'bundleType': 'zip',
'eTag': self.eTag,
'version': self.version_id
}
}
self.bundle_mock = MagicMock()
self.bundle_mock.tell.return_value = (5 << 20)
self.bundle_mock.read.return_value = b'a' * (5 << 20)
self.bundle_mock.__enter__.return_value = self.bundle_mock
self.bundle_mock.__exit__.return_value = None
self.zipfile_mock = MagicMock()
self.zipfile_mock.write.return_value = None
self.zipfile_mock.close.return_value = None
self.zipfile_mock.__enter__.return_value = self.zipfile_mock
self.zipfile_mock.__exit__.return_value = None
self.session = MagicMock()
self.push = Push(self.session)
self.push.s3 = MagicMock()
self.push.s3.put_object.return_value = self.upload_response
self.push.s3.create_multipart_upload.return_value = {
'UploadId': self.upload_id
}
self.push.s3.upload_part.return_value = {'ETag': self.eTag}
self.push.s3.complete_multipart_upload\
.return_value = self.upload_response
self.push.codedeploy = MagicMock()
