def setUp(self):
mocked_flags = mock_flags.PatchTestCaseFlags(self)
mocked_flags.cloud = providers.AWS
mocked_flags.os_type = os_types.DEBIAN
mocked_flags.run_uri = 'aaaaaa'
p = mock.patch('perfkitbenchmarker.providers.aws.'
'util.IssueRetryableCommand')
p.start()
self.addCleanup(p.stop)
# VM Creation depends on there being a BenchmarkSpec.
config_spec = benchmark_config_spec.BenchmarkConfigSpec(
_BENCHMARK_NAME, flag_values=mocked_flags, vm_groups={})
self.spec = benchmark_spec.BenchmarkSpec(config_spec, _BENCHMARK_NAME,
_BENCHMARK_UID)
self.addCleanup(context.SetThreadBenchmarkSpec, None)
self.vm = aws_virtual_machine.AwsVirtualMachine(
virtual_machine.BaseVmSpec('test_vm_spec.AWS',
zone='us-east-1a',
machine_type='c3.large'))
self.vm.id = 'i-foo'
path = os.path.join(os.path.dirname(__file__), 'data',
'aws-describe-instance.json')
with open(path) as f:
self.response = f.read()
def setUp(self):
for module in ('perfkitbenchmarker.virtual_machine',
'perfkitbenchmarker.vm_util',
'perfkitbenchmarker.providers.aws.aws_network'):
p = mock.patch('{0}.FLAGS'.format(module))
mock_flags = p.start()
mock_flags.run_uri = 'aaaaaa'
self.addCleanup(p.stop)
p = mock.patch('perfkitbenchmarker.providers.aws.'
'util.IssueRetryableCommand')
p.start()
self.addCleanup(p.stop)
# VM Creation depends on there being a BenchmarkSpec.
self.spec = benchmark_spec.BenchmarkSpec({}, 'name', 'benchmark_uid')
self.addCleanup(context.SetThreadBenchmarkSpec, None)
self.vm = aws_virtual_machine.AwsVirtualMachine(
virtual_machine.BaseVmSpec('test_vm_spec.AWS',
zone='us-east-1a',
machine_type='c3.large'))
self.vm.id = 'i-foo'
path = os.path.join(os.path.dirname(__file__), 'data',
'aws-describe-instance.json')
with open(path) as f:
self.response = f.read()
def CreateVm():
"""Returns the AWS run-instances command line."""
vm_util.IssueCommand.side_effect = [('', '', 0)]
util.IssueRetryableCommand.side_effect = [('', '', 0)]
vm_spec = aws_virtual_machine.AwsVmSpec(
'test_vm_spec.AWS', zone='us-west-1a', machine_type='c5n.18xlarge')
vm = aws_virtual_machine.AwsVirtualMachine(vm_spec)
vm.network.regional_network = mock.Mock()
vm.network.subnet = mock.Mock(id='subnet-1234')
vm.network.placement_group = mock.Mock()
vm.network.Create()
vm._Create()
return ' '.join(vm_util.IssueCommand.call_args[0][0])
def setUp(self):
super(AwsVirtualMachineTestCase, self).setUp()
FLAGS.cloud = providers.AWS
FLAGS.os_type = os_types.DEBIAN
FLAGS.run_uri = 'aaaaaa'
FLAGS.temp_dir = 'tmp'
p = mock.patch('perfkitbenchmarker.providers.aws.'
'util.IssueRetryableCommand')
p.start()
self.addCleanup(p.stop)
p2 = mock.patch('perfkitbenchmarker.' 'vm_util.IssueCommand')
p2.start()
self.addCleanup(p2.stop)
# VM Creation depends on there being a BenchmarkSpec.
config_spec = benchmark_config_spec.BenchmarkConfigSpec(
_BENCHMARK_NAME, flag_values=FLAGS, vm_groups={})
self.spec = benchmark_spec.BenchmarkSpec(mock.MagicMock(), config_spec,
_BENCHMARK_UID)
self.addCleanup(context.SetThreadBenchmarkSpec, None)
self.vm = aws_virtual_machine.AwsVirtualMachine(
aws_virtual_machine.AwsVmSpec('test_vm_spec.AWS',
zone='us-east-1a',
machine_type='c3.large',
spot_price=123.45))
self.vm.id = 'i-foo'
self.vm.image = 'ami-12345'
self.vm.client_token = '00000000-1111-2222-3333-444444444444'
network_mock = mock.MagicMock()
network_mock.subnet = mock.MagicMock(id='subnet-id')
placement_group = mock.MagicMock()
placement_group.name = 'placement_group_name'
placement_group.strategy = 'cluster'
network_mock.placement_group = placement_group
self.vm.network = network_mock
self.vm.placement_group = placement_group
self.response = self.open_json_data('aws-describe-instance.json')
self.sir_response = self.open_json_data(
'aws-describe-spot-instance-requests.json')
self.vm.network.is_static = False
self.vm.network.regional_network.vpc.default_security_group_id = 'sg-1234'
def setUp(self):
mocked_flags = mock_flags.PatchTestCaseFlags(self)
mocked_flags.cloud = providers.AWS
mocked_flags.os_type = os_types.DEBIAN
mocked_flags.run_uri = 'aaaaaa'
mocked_flags.temp_dir = 'tmp'
p = mock.patch('perfkitbenchmarker.providers.aws.'
'util.IssueRetryableCommand')
p.start()
self.addCleanup(p.stop)
p2 = mock.patch('perfkitbenchmarker.' 'vm_util.IssueCommand')
p2.start()
self.addCleanup(p2.stop)
# VM Creation depends on there being a BenchmarkSpec.
config_spec = benchmark_config_spec.BenchmarkConfigSpec(
_BENCHMARK_NAME, flag_values=mocked_flags, vm_groups={})
self.spec = benchmark_spec.BenchmarkSpec(mock.MagicMock(), config_spec,
_BENCHMARK_UID)
self.addCleanup(context.SetThreadBenchmarkSpec, None)
self.vm = aws_virtual_machine.AwsVirtualMachine(
aws_virtual_machine.AwsVmSpec('test_vm_spec.AWS',
zone='us-east-1a',
machine_type='c3.large',
spot_price=123.45))
self.vm.id = 'i-foo'
self.vm.image = 'ami-12345'
self.vm.client_token = '00000000-1111-2222-3333-444444444444'
network_mock = mock.MagicMock()
network_mock.subnet = mock.MagicMock(id='subnet-id')
placement_group = mock.MagicMock()
placement_group.name = 'placement_group_name'
network_mock.placement_group = placement_group
self.vm.network = network_mock
self.response = self.openJsonData('aws-describe-instance.json')
self.sir_response =\
self.openJsonData('aws-describe-spot-instance-requests.json')
