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()
