def setUp(self): self.region = 'us-east-1' self.config_file = 'config-file' self.installer = 'install' self.bucket = 'aws-codedeploy-{0}'.format(self.region) self.key = 'latest/{0}'.format(self.installer) self.agent_installer = 's3://{0}/{1}'.format(self.bucket, self.key) self.system_patcher = patch('platform.system') self.system = self.system_patcher.start() self.system.return_value = 'Linux' self.linux_distribution_patcher = patch('awscli.compat.linux_distribution') self.linux_distribution = self.linux_distribution_patcher.start() self.linux_distribution.return_value = ('Ubuntu', '', '') self.urlopen_patcher = patch( 'awscli.customizations.codedeploy.utils.urlopen' ) self.urlopen = self.urlopen_patcher.start() self.urlopen.side_effect = timeout('Not EC2 instance') self.geteuid_patcher = patch('os.geteuid', create=True) self.geteuid = self.geteuid_patcher.start() self.geteuid.return_value = 0 self.isfile_patcher = patch('os.path.isfile') self.isfile = self.isfile_patcher.start() self.isfile.return_value = False self.makedirs_patcher = patch('os.makedirs') self.makedirs = self.makedirs_patcher.start() self.copyfile_patcher = patch('shutil.copyfile') self.copyfile = self.copyfile_patcher.start() self.open_patcher = patch( 'awscli.customizations.codedeploy.systems.open', mock_open(), create=True ) self.open = self.open_patcher.start() self.args = Namespace() self.args.override_config = False self.args.config_file = self.config_file self.args.agent_installer = None self.globals = Namespace() self.globals.region = self.region self.body = 'install-script' self.reader = MagicMock() self.reader.read.return_value = self.body self.s3 = MagicMock() self.s3.get_object.return_value = {'Body': self.reader} self.session = MagicMock() self.session.create_client.return_value = self.s3 self.install = Install(self.session)
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)