def setUp(self): super(AwsVpcTestCase, self).setUp() p = mock.patch('perfkitbenchmarker.providers.aws.' 'util.IssueRetryableCommand') p.start() self.addCleanup(p.stop) self.vpc = aws_network.AwsVpc('region')
def setUp(self): p = mock.patch('perfkitbenchmarker.providers.aws.' 'util.IssueRetryableCommand') p.start() self.addCleanup(p.stop) self.vpc = aws_network.AwsVpc('region') self.vpc.id = 'vpc-foo'
def testCreate(self): self.SetExpectedCommands(VPC_CREATE, MODIFY_VPC, VPC_QUERY_ONE, DESCRIBE_SECURITY_GROUPS_ONLY_DEFAULT) vpc = aws_network.AwsVpc(REGION) vpc.Create() self.assertEqual(VPC_ID, vpc.id) self.assertEqual(SG_DEFAULT, vpc.default_security_group_id) self.assertCommandsCalled()
def testStaticVpc(self, mock_cmd): vpc_id = 'vpc-1234' security_group_id = 'sg-1234' security_group_res = {'SecurityGroups': [{'GroupId': security_group_id}]} does_exist_res = {'Vpcs': [1]} does_not_exist_res = {'Vpcs': []} mock_cmd.return_value = json.dumps(security_group_res), '', 0 util.IssueRetryableCommand.side_effect = [(json.dumps(does_exist_res), ''), (json.dumps(does_not_exist_res), '')] vpc = aws_network.AwsVpc('region', vpc_id) self.assertTrue(vpc.user_managed) # show Create() works vpc.Create() self.assertEqual(vpc_id, vpc.id) self.assertEqual(security_group_id, vpc.default_security_group_id) # show Delete() doesn't delete, but it does call _Exists() again vpc.Delete()
def testExistsVpcsOne(self): self.SetExpectedCommands(VPC_QUERY_ONE) vpc = aws_network.AwsVpc(REGION) self.assertTrue(vpc._Exists()) self.assertCommandsCalled()
def testInitWithVpc(self): self.SetExpectedCommands(DESCRIBE_SECURITY_GROUPS_ONLY_DEFAULT) vpc = aws_network.AwsVpc(REGION, VPC_ID) self.assertEqual(VPC_ID, vpc.id) self.assertEqual(SG_DEFAULT, vpc.default_security_group_id) self.assertCommandsCalled()