def setUp(self):
"""Test case setup function called automatically prior to each test."""
super(TestCOTAddFile, self).setUp()
self.instance = COTAddFile(UI())
self.instance.output = self.temp_file
class TestCOTAddFile(COT_UT):
"""Test cases for the COTAddFile module."""
def setUp(self):
"""Test case setup function called automatically prior to each test."""
super(TestCOTAddFile, self).setUp()
self.instance = COTAddFile(UI())
self.instance.output = self.temp_file
def test_readiness(self):
"""Test ready_to_run() under various combinations of parameters."""
ready, reason = self.instance.ready_to_run()
self.assertFalse(ready)
self.assertEqual("FILE is a mandatory argument!", reason)
self.assertRaises(InvalidInputError, self.instance.run)
self.instance.file = self.iosv_ovf
ready, reason = self.instance.ready_to_run()
self.assertFalse(ready)
self.assertEqual("PACKAGE is a mandatory argument!", reason)
self.assertRaises(InvalidInputError, self.instance.run)
self.instance.package = self.input_ovf
ready, reason = self.instance.ready_to_run()
self.assertTrue(ready)
def test_add_file(self):
"""Basic file addition."""
self.instance.package = self.input_ovf
self.instance.file = self.iosv_ovf
self.instance.run()
self.instance.finished()
self.check_diff("""
<ovf:File ovf:href="input.iso" ovf:id="file2" ovf:size="{iso_size}" />
+ <ovf:File ovf:href="iosv.ovf" ovf:id="iosv.ovf" ovf:size="{ovf_size}" />
</ovf:References>
""".format(iso_size=self.FILE_SIZE['input.iso'],
ovf_size=os.path.getsize(self.iosv_ovf)))
def test_add_file_with_id(self):
"""Add a file with explicit 'file_id' argument."""
self.instance.package = self.input_ovf
self.instance.file = self.iosv_ovf
self.instance.file_id = "myfile"
self.instance.run()
self.instance.finished()
self.check_diff("""
<ovf:File ovf:href="input.iso" ovf:id="file2" ovf:size="{iso_size}" />
+ <ovf:File ovf:href="iosv.ovf" ovf:id="myfile" ovf:size="{ovf_size}" />
</ovf:References>
""".format(iso_size=self.FILE_SIZE['input.iso'],
ovf_size=os.path.getsize(self.iosv_ovf)))
def test_overwrite_file(self):
"""Overwrite a file implicitly."""
self.instance.package = self.input_ovf
self.instance.file = resource_filename(__name__, 'input.iso')
self.instance.run()
self.assertLogged(**self.OVERWRITING_FILE)
self.instance.finished()
self.check_diff("")
def test_add_file_then_change_to_disk(self):
"""Add a disk as a file, then make it a proper disk."""
self.instance.package = self.minimal_ovf
intermediate_ovf = os.path.join(self.temp_dir, "mid.ovf")
self.instance.output = intermediate_ovf
self.instance.file = self.blank_vmdk
self.instance.file_id = "mydisk"
self.instance.run()
self.instance.finished()
self.check_diff(file1=self.minimal_ovf,
file2=intermediate_ovf,
expected="""
<ovf:Envelope xmlns:ovf="http://schemas.dmtf.org/ovf/envelope/1">
- <ovf:References />
+ <ovf:References>
+ <ovf:File ovf:href="blank.vmdk" ovf:id="mydisk" ovf:size="{0}" />
+ </ovf:References>
<ovf:VirtualSystem ovf:id="x">
""".format(self.FILE_SIZE['blank.vmdk']))
from COT.add_disk import COTAddDisk
ad = COTAddDisk(UI())
ad.package = intermediate_ovf
ad.output = self.temp_file
ad.disk_image = self.blank_vmdk
ad.file_id = "mydisk"
ad.run()
self.assertLogged(**self.TYPE_NOT_SPECIFIED_GUESS_HARDDISK)
self.assertLogged(**self.CONTROLLER_NOT_SPECIFIED_GUESS_IDE)
self.assertLogged(**self.OVERWRITING_FILE)
self.assertLogged(**self.ADDRESS_ON_PARENT_NOT_SPECIFIED)
ad.finished()
ad.destroy()
self.check_diff(file1=intermediate_ovf, expected="""
<?xml version='1.0' encoding='utf-8'?>
-<ovf:Envelope xmlns:ovf="http://schemas.dmtf.org/ovf/envelope/1">
+<ovf:Envelope xmlns:ovf="http://schemas.dmtf.org/ovf/envelope/1" \
xmlns:rasd="http://schemas.dmtf.org/wbem/wscim/1/cim-schema/2/\
CIM_ResourceAllocationSettingData">
<ovf:References>
...
</ovf:References>
+ <ovf:DiskSection>
+ <ovf:Info>Virtual disk information</ovf:Info>
+ <ovf:Disk ovf:capacity="512" ovf:capacityAllocationUnits="byte * 2^20" \
ovf:diskId="mydisk" ovf:fileRef="mydisk" ovf:format="http://www.vmware.com/\
interfaces/specifications/vmdk.html#streamOptimized" />
+ </ovf:DiskSection>
<ovf:VirtualSystem ovf:id="x">
...
<ovf:Info />
+ <ovf:Item>
+ <rasd:Address>0</rasd:Address>
+ <rasd:Description>IDE Controller 0</rasd:Description>
+ <rasd:ElementName>IDE Controller</rasd:ElementName>
+ <rasd:InstanceID>1</rasd:InstanceID>
+ <rasd:ResourceType>5</rasd:ResourceType>
+ </ovf:Item>
+ <ovf:Item>
+ <rasd:AddressOnParent>0</rasd:AddressOnParent>
+ <rasd:ElementName>Hard Disk Drive</rasd:ElementName>
+ <rasd:HostResource>ovf:/disk/mydisk</rasd:HostResource>
+ <rasd:InstanceID>2</rasd:InstanceID>
+ <rasd:Parent>1</rasd:Parent>
+ <rasd:ResourceType>17</rasd:ResourceType>
+ </ovf:Item>
</ovf:VirtualHardwareSection>
""")
