def testReadFromFile_NoErr(self):
s = ('[{'
' "ip_address": "174.12.14.1", '
' "user_name": "perfkitbenchmarker", '
' "keyfile_path": "perfkitbenchmarker.pem" '
'}, '
'{ '
' "ip_address": "174.12.14.121", '
' "user_name": "ubuntu", '
' "keyfile_path": "rackspace.pem", '
' "internal_ip": "10.10.10.2", '
' "zone": "rackspace_dallas" '
'}] ')
fp = StringIO(s)
StaticVirtualMachine.ReadStaticVirtualMachineFile(fp)
vm_pool = StaticVirtualMachine.vm_pool
self.assertEqual(2, len(vm_pool))
self._AssertStaticVMsEqual(
StaticVirtualMachine(
StaticVmSpec(_COMPONENT,
ip_address='174.12.14.1',
user_name='perfkitbenchmarker',
ssh_private_key='perfkitbenchmarker.pem')),
vm_pool[0])
self._AssertStaticVMsEqual(
StaticVirtualMachine(
StaticVmSpec(_COMPONENT,
ip_address='174.12.14.121',
user_name='ubuntu',
ssh_private_key='rackspace.pem',
internal_ip='10.10.10.2',
zone='rackspace_dallas')), vm_pool[1])
def testDiskSpecs(self):
s = """
[{
"ip_address": "174.12.14.1",
"user_name": "ubuntu",
"keyfile_path": "test_keyfile_path",
"local_disks": ["/test_local_disk_0", "/test_local_disk_1"],
"scratch_disk_mountpoints": ["/test_scratch_disk_0",
"/test_scratch_disk_1"]
}]
"""
expected_paths_and_mount_points = ((None, '/test_scratch_disk_0'),
(None, '/test_scratch_disk_1'),
('/test_local_disk_0', None),
('/test_local_disk_1', None))
fp = StringIO(s)
StaticVirtualMachine.ReadStaticVirtualMachineFile(fp)
self.assertEqual(1, len(StaticVirtualMachine.vm_pool))
vm = StaticVirtualMachine.GetStaticVirtualMachine()
self.assertTrue(vm.from_pool)
self.assertEqual(len(vm.disk_specs), 4)
for disk_spec, expected_paths in zip(vm.disk_specs,
expected_paths_and_mount_points):
expected_device_path, expected_mount_point = expected_paths
self.assertEqual(disk_spec.device_path, expected_device_path)
self.assertEqual(disk_spec.mount_point, expected_mount_point)
def testReadFromFile_NoErr(self):
s = ('[{'
' "ip_address": "174.12.14.1", '
' "user_name": "perfkitbenchmarker", '
' "keyfile_path": "perfkitbenchmarker.pem" '
'}, '
'{ '
' "ip_address": "174.12.14.121", '
' "user_name": "ubuntu", '
' "keyfile_path": "rackspace.pem", '
' "internal_ip": "10.10.10.2", '
' "zone": "rackspace_dallas" '
'}] ')
fp = BytesIO(s)
StaticVirtualMachine.ReadStaticVirtualMachineFile(fp)
vm_pool = StaticVirtualMachine.vm_pool
self.assertEqual(2, len(vm_pool))
self._AssertStaticVMsEqual(
StaticVirtualMachine(
'174.12.14.1', 'perfkitbenchmarker',
'perfkitbenchmarker.pem'), vm_pool[0])
self._AssertStaticVMsEqual(
StaticVirtualMachine('174.12.14.121', 'ubuntu', 'rackspace.pem',
'10.10.10.2', 'rackspace_dallas'),
vm_pool[1])
def testReadFromFile_UnknownOsTypeDefaultsToLinuxRequiredKeys(self):
FLAGS.os_type = 'unknown_os_type'
s = ('[{'
' "ip_address": "174.12.14.1", '
' "user_name": "perfkitbenchmarker", '
' "keyfile_path": "perfkitbenchmarker.pem"'
'}]')
fp = StringIO(s)
StaticVirtualMachine.ReadStaticVirtualMachineFile(fp)
vm_pool = StaticVirtualMachine.vm_pool
self.assertEqual(1, len(vm_pool))
self._AssertStaticVMsEqual(
StaticVirtualMachine(
StaticVmSpec(_COMPONENT,
ip_address='174.12.14.1',
user_name='perfkitbenchmarker',
ssh_private_key='perfkitbenchmarker.pem')),
vm_pool[0])
def testCreateReturn(self):
s = ('[{'
' "ip_address": "174.12.14.1", '
' "user_name": "perfkitbenchmarker", '
' "keyfile_path": "perfkitbenchmarker.pem" '
'}, '
'{ '
' "ip_address": "174.12.14.121", '
' "user_name": "ubuntu", '
' "keyfile_path": "rackspace.pem", '
' "internal_ip": "10.10.10.2", '
' "zone": "rackspace_dallas" '
'}] ')
fp = StringIO(s)
StaticVirtualMachine.ReadStaticVirtualMachineFile(fp)
self.assertEqual(2, len(StaticVirtualMachine.vm_pool))
vm0 = StaticVirtualMachine.GetStaticVirtualMachine()
self.assertTrue(vm0.from_pool)
self.assertEqual(1, len(StaticVirtualMachine.vm_pool))
vm0.Delete()
self.assertEqual(2, len(StaticVirtualMachine.vm_pool))
vm1 = StaticVirtualMachine.GetStaticVirtualMachine()
self.assertIs(vm0, vm1)
def testReadFromFile_Empty(self):
fp = StringIO('[]')
StaticVirtualMachine.ReadStaticVirtualMachineFile(fp)
self.assertEqual([], list(StaticVirtualMachine.vm_pool))
def testReadFromFile_Empty(self):
fp = BytesIO('[]')
StaticVirtualMachine.ReadStaticVirtualMachineFile(fp)
self.assertEqual([], StaticVirtualMachine.vm_pool)