def run(self,
_count=1,
max_count=1,
key_name=None,
security_groups=[],
user_data=None,
addressing_type=None,
instance_type='m1.small',
placement=None,
kernel_id=None,
ramdisk_id=None,
monitoring_enabled=False,
subnet_id=None,
block_device_map=None,
disable_api_termination=False,
instance_initiated_shutdown_behavior=None,
private_ip_address=None,
placement_group=None,
security_group_ids=None):
return _Keys(id='r-reservation',
connection=self._cnx,
instances=[
MockInstance('pending',
instance_type=instance_type,
image_id=self.id,
groups=[
_Keys(id=g, name=g)
for g in security_groups
],
cnx=self._cnx)
])
def test_key_filename_makes_new_key(self):
config = _Keys(client='test-client--test_key_filename')
cnx = MockConnection('key', 'secret')
cnx._key_pairs.append(_Keys(name='another-host'))
with mock.patch('socket.gethostname', lambda: 'test-host'):
dir_made = []
with mock.patch('os.mkdir', lambda p: dir_made.append(p)):
filename = get_private_key_filename(config, cnx)
self.assertTrue(dir_made)
self.assertIn('test-host', cnx._key_pairs_created)
cnx._key_pairs_created = []
filename = get_private_key_filename(config, cnx)
self.assertNotIn('test-host', cnx._key_pairs_created)
def run(self, _count=1, max_count=1, key_name=None, security_groups=[],
user_data=None, addressing_type=None, instance_type='m1.small',
placement=None, kernel_id=None, ramdisk_id=None,
monitoring_enabled=False, subnet_id=None, block_device_map=None,
disable_api_termination=False,
instance_initiated_shutdown_behavior=None,
private_ip_address=None, placement_group=None,
security_group_ids=None):
return _Keys(id='r-reservation',
connection=self._cnx,
instances=[MockInstance('pending',
instance_type=instance_type, image_id=self.id,
groups=[_Keys(id=g, name=g) for g in security_groups],
cnx=self._cnx)])
def test_key_filename(self):
config = _Keys(client='test-client--test_key_filename')
cnx = MockConnection('key', 'secret')
with mock.patch('socket.gethostname', lambda: 'test-host'):
dir_made = []
with mock.patch('os.mkdir', lambda p: dir_made.append(p)):
filename = get_private_key_filename(config, cnx)
self.assertTrue(dir_made)
def __init__(self, state, next_state='running',
instance_type='t1.micro', image_id='ami-12345', cnx=None,
groups=None):
self.dns_name = 'ec2-host'
self.groups = groups or [_Keys(name='default')]
self.id = 'i-test1'
self.ip_address = '10.56.32.4'
self.image_id = image_id
self.instance_type = instance_type
self.key_name = 'host'
self.placement = 'ec2-zone'
self.state = state
self.tags = {}
if cnx:
self.region = cnx.region
self.__next_state = next_state
def __init__(self,
state,
next_state='running',
instance_type='t1.micro',
image_id='ami-12345',
cnx=None,
groups=None):
self.dns_name = 'ec2-host'
self.groups = groups or [_Keys(name='default')]
self.id = 'i-test1'
self.ip_address = '10.56.32.4'
self.image_id = image_id
self.instance_type = instance_type
self.key_name = 'host'
self.placement = 'ec2-zone'
self.state = state
self.tags = {}
if cnx:
self.region = cnx.region
self.__next_state = next_state
def get_all_instances(self):
return [_Keys(id='r-reservation',
connection=self,
instances=[MockInstance(self, 'running')])]
def create_key_pair(self, host):
self._key_pairs_created.append(host)
key_pair = _Keys(name=host)
key_pair.save = lambda f: None
self._key_pairs.append(key_pair)
return key_pair
def get_all_instances(self):
return [
_Keys(id='r-reservation',
connection=self,
instances=[MockInstance(self, 'running')])
]
def create_key_pair(self, host):
self._key_pairs_created.append(host)
key_pair = _Keys(name=host)
key_pair.save = lambda f: None
self._key_pairs.append(key_pair)
return key_pair
def __init__(self):
self.id = 'v-test1'
self.attach_data = _Keys(instance_id='i-test1',
device='/dev/sda1',
status='attached')
def __init__(self):
self.id = 'v-test1'
self.attach_data = _Keys(instance_id='i-test1',
device='/dev/sda1', status='attached')
