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 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 __init__(self): self.id = 'v-test1' self.attach_data = _Keys(instance_id='i-test1', device='/dev/sda1', status='attached')