def setUp(self):
self.popen_patcher = patch('subprocess.Popen')
self.popen = self.popen_patcher.start()
self.check_call_patcher = patch('subprocess.check_call')
self.check_call = self.check_call_patcher.start()
self.open_patcher = patch(
'awscli.customizations.codedeploy.systems.open',
mock_open(),
create=True)
self.open = self.open_patcher.start()
self.environ_patcher = patch('os.environ')
self.environ = self.environ_patcher.start()
self.environ.copy.return_value = dict()
self.config_dir = '/etc/codedeploy-agent/conf'
self.config_file = 'codedeploy.onpremises.yml'
self.config_path = '{0}/{1}'.format(self.config_dir, self.config_file)
self.installer = 'install'
self.bucket = 'bucket'
self.key = 'key'
self.region = 'us-east-1'
self.access_key_id = 'ACCESSKEYID'
self.secret_access_key = 'SECRETACCESSKEY'
self.session_token = 'SESSION_TOKEN'
self.credentials = MagicMock()
self.credentials.access_key = self.access_key_id
self.credentials.secret_key = self.secret_access_key
self.credentials.token = self.session_token
self.environment = dict({
'AWS_REGION': self.region,
'AWS_ACCESS_KEY_ID': self.access_key_id,
'AWS_SECRET_ACCESS_KEY': self.secret_access_key,
'AWS_SESSION_TOKEN': self.session_token
})
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.session.get_credentials.return_value = self.credentials
self.params = Namespace()
self.params.session = self.session
self.params.region = self.region
self.params.bucket = self.bucket
self.params.key = self.key
self.ubuntu = Ubuntu(self.params)
def validate_instance(params):
if platform.system() == 'Linux':
if 'Ubuntu' in platform.linux_distribution()[0]:
params.system = Ubuntu(params)
elif platform.system() == 'Windows':
params.system = Windows(params)
if 'system' not in params:
raise RuntimeError(
'Only Ubuntu Server and Windows Server operating systems are '
'supported.')
try:
urlopen('http://169.254.169.254/latest/meta-data/', timeout=1)
raise RuntimeError('Amazon EC2 instances are not supported.')
except (URLError, timeout):
pass
def validate_instance(params):
if platform.system() == 'Linux':
if 'Ubuntu' in platform.linux_distribution()[0]:
params.system = Ubuntu(params)
if 'Red Hat Enterprise Linux Server' in platform.linux_distribution(
)[0]:
params.system = RHEL(params)
elif platform.system() == 'Windows':
params.system = Windows(params)
if 'system' not in params:
raise RuntimeError(System.UNSUPPORTED_SYSTEM_MSG)
try:
urlopen('http://169.254.169.254/latest/meta-data/', timeout=1)
raise RuntimeError('Amazon EC2 instances are not supported.')
except (URLError, timeout):
pass
def setUp(self):
super(self.__class__, self).setUp()
self.ubuntu = Ubuntu(self.params)
