def testValidMachineTypeWithSpecifiedRootVolumeSize(self):
util.IssueRetryableCommand.side_effect = [(self.describeImageOutput, None)]
desired_root_volume_size_gb = 35
machine_type = 'c1.medium'
image_id = 'ami-a9d276c9'
region = 'us-west-2'
expected = [{'DeviceName': '/dev/sda1',
'Ebs': {'SnapshotId': 'snap-826344d5',
'DeleteOnTermination': True,
'VolumeType': 'gp2',
'VolumeSize': 35}},
{'DeviceName': '/dev/xvdb',
'VirtualName': 'ephemeral0'}]
actual = json.loads(aws_virtual_machine.GetBlockDeviceMap(
machine_type, desired_root_volume_size_gb, image_id, region))
self.assertEqual(actual, expected)
def testValidMachineTypeWithNoRootVolumeSize(self):
expected = [{'DeviceName': '/dev/xvdb', 'VirtualName': 'ephemeral0'}]
actual = json.loads(aws_virtual_machine.GetBlockDeviceMap('c1.medium'))
self.assertEqual(actual, expected)
def testInvalidMachineType(self):
self.assertEqual(aws_virtual_machine.GetBlockDeviceMap('invalid'),
None)