def test_attach_classic_link_instance_object_groups(self): sec_group_1 = SecurityGroup() sec_group_1.id = 'sg-foo' sec_group_2 = SecurityGroup() sec_group_2.id = 'sg-bar' groups = [sec_group_1, sec_group_2] self.set_http_response(status_code=200) response = self.vpc.attach_classic_instance( instance_id='my_instance_id', groups=groups, dry_run=True) self.assertTrue(response) self.assert_request_parameters( { 'Action': 'AttachClassicLinkVpc', 'VpcId': self.vpc_id, 'InstanceId': 'my_instance_id', 'SecurityGroupId.1': 'sg-foo', 'SecurityGroupId.2': 'sg-bar', 'DryRun': 'true' }, ignore_params_values=[ 'AWSAccessKeyId', 'SignatureMethod', 'SignatureVersion', 'Timestamp', 'Version' ])
def test_attach_classic_link_instance_object_groups(self): sec_group_1 = SecurityGroup() sec_group_1.id = 'sg-foo' sec_group_2 = SecurityGroup() sec_group_2.id = 'sg-bar' groups = [sec_group_1, sec_group_2] self.set_http_response(status_code=200) response = self.vpc.attach_classic_instance( instance_id='my_instance_id', groups=groups, dry_run=True ) self.assertTrue(response) self.assert_request_parameters({ 'Action': 'AttachClassicLinkVpc', 'VpcId': self.vpc_id, 'InstanceId': 'my_instance_id', 'SecurityGroupId.1': 'sg-foo', 'SecurityGroupId.2': 'sg-bar', 'DryRun': 'true'}, ignore_params_values=['AWSAccessKeyId', 'SignatureMethod', 'SignatureVersion', 'Timestamp', 'Version'])
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 create_security_group(self, name, description, vpc_id=None): if self.create_security_group_callback: self.create_security_group_callback(vpc_id) sg = SecurityGroup() sg.id = 'sg-%s' % new_id() sg.vpc_id = vpc_id or self.default_vpc.id self.security_groups[sg.id] = sg return sg
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)