def setUp(self): ec2.credentials.ACCESS_KEY_ID = 'abc' ec2.credentials.SECRET_ACCESS_KEY = 'xyz' # Build up two reservations, with two instances each, totalling 4 instances # Two running, two stopped reservations = [] instance_count = 0 for i in xrange(2): i1 = Instance() i1.id = 'i-abc%d' % instance_count i1.state = 'running' i1.tags = {'Name': 'instance-%d' % instance_count} instance_count += 1 i2 = Instance() i2.id = 'i-abc%d' % instance_count i2.state = 'stopped' i2.tags = {'Name': 'instance-%d' % instance_count} instance_count += 1 reservation = MagicMock() reservation.instances.__iter__ = MagicMock( return_value=iter([i1, i2])) reservations.append(reservation) self.connection = MagicMock() self.connection.get_all_instances = MagicMock( return_value=reservations)
def setUp(self): ec2.credentials.ACCESS_KEY_ID = 'abc' ec2.credentials.SECRET_ACCESS_KEY = 'xyz' # Build up two reservations, with two instances each, totalling 4 instances # Two running, two stopped reservations = [] instance_count = 0 for i in xrange(2): i1 = Instance() i1.id = 'i-abc%d' % instance_count i1._state = RUNNING_STATE i1.tags = {'Name': 'instance-%d' % instance_count} instance_count += 1 i2 = Instance() i2.id = 'i-abc%d' % instance_count i2._state = STOPPED_STATE i2.tags = {'Name': 'instance-%d' % instance_count} instance_count += 1 reservation = MagicMock() reservation.instances.__iter__ = MagicMock(return_value=iter([i1, i2])) reservations.append(reservation) security_groups = [] for i in xrange(2): sg = SecurityGroup() sg.id = 'sg-abc%d' % i sg.name = 'group-%d' % i sg.description = 'Group %d' % i security_groups.append(sg) self.connection = MagicMock() self.connection.get_all_instances = MagicMock(return_value=reservations) self.connection.get_all_security_groups = MagicMock(return_value=security_groups)
def setUp(self): ec2.credentials.ACCESS_KEY_ID = 'abc' ec2.credentials.SECRET_ACCESS_KEY = 'xyz' # Build up two reservations, with two instances each, totalling 4 instances # Two running, two stopped reservations = [] instance_count = 0 for i in xrange(2): i1 = Instance() i1.id = 'i-abc%d' % instance_count i1.state = 'running' i1.tags = {'Name': 'instance-%d' % instance_count} instance_count += 1 i2 = Instance() i2.id = 'i-abc%d' % instance_count i2.state = 'stopped' i2.tags = {'Name': 'instance-%d' % instance_count} instance_count += 1 reservation = MagicMock() reservation.instances.__iter__ = MagicMock(return_value=iter([i1, i2])) reservations.append(reservation) self.connection = MagicMock() self.connection.get_all_instances = MagicMock(return_value=reservations)
def setUp(self): ec2.credentials.ACCESS_KEY_ID = 'abc' ec2.credentials.SECRET_ACCESS_KEY = 'xyz' # Build up two reservations, with two instances each, totalling 4 instances # Two running, two stopped reservations = [] instance_count = 0 for i in xrange(2): i1 = Instance() i1.id = 'i-abc%d' % instance_count i1._state = RUNNING_STATE i1.tags = {'Name': 'instance-%d' % instance_count} instance_count += 1 i2 = Instance() i2.id = 'i-abc%d' % instance_count i2._state = STOPPED_STATE i2.tags = {'Name': 'instance-%d' % instance_count} instance_count += 1 reservation = MagicMock() reservation.instances.__iter__ = MagicMock(return_value=iter([i1, i2])) reservations.append(reservation) security_groups = [] for i in xrange(2): sg = SecurityGroup() sg.id = 'sg-abc%d' % i sg.name = 'group-%d' % i sg.description = 'Group %d' % i security_groups.append(sg) vpcs = [] for i in xrange(2): vpc = VPC() vpc.id = 'vpc-abc%d' % i if i % 2: vpc.state = 'pending' vpc.is_default = False vpc.instance_tenancy = 'default' else: vpc.state = 'available' vpc.is_default = True vpc.instance_tenancy = 'dedicated' vpc.cidr_block = '10.%d.0.0/16' % i vpc.dhcp_options_id = 'dopt-abc%d' % i vpcs.append(vpc) self.connection = MagicMock() self.connection.get_all_instances = MagicMock(return_value=reservations) self.connection.get_all_security_groups = MagicMock(return_value=security_groups) self.vpc_connection = MagicMock() self.vpc_connection.get_all_vpcs = MagicMock(return_value=vpcs)
def generate_instance(self, id, ip): i = Instance() i.id = id i.private_ip_address = ip i.private_dns_name = 'ip-{}.eu-west-1.compute.internal'.format(ip.replace('.', '-')) i.tags = {'aws:cloudformation:stack-name': 'etc-cluster', 'aws:autoscaling:groupName': 'etc-cluster-postgres'} return i