class TestConfigGenerator(unittest.TestCase): """Tests the config generator.""" def setUp(self): self.instanceConfig = InstanceConfig("test-config") self.instanceConfig.setKernel("/vmlinuz") self.instanceConfig.addDisk("/tmp/foo.img", "sda1") self.generator = LibVirtXMLConfigGenerator() def test_simple_config(self): expected = "<?xml version='1.0' encoding='UTF-8'?><domain type='xen'><name>test-config</name><os><type>linux</type><kernel>/vmlinuz</kernel><root>/dev/sda1</root></os><memory>131072</memory><vcpu>1</vcpu><devices><disk type='file'><source file='/tmp/foo.img'/><target dev='sda1'/></disk></devices><on_reboot>restart</on_reboot><on_poweroff>destroy</on_poweroff><on_crash>rename-restart</on_crash></domain>" xml = self.generator.generate(self.instanceConfig, pretty=False) self.assertEqual(expected, xml)
def setUp(self): self.instanceConfig = InstanceConfig("test-config") self.instanceConfig.setKernel("/vmlinuz") self.instanceConfig.addDisk("/tmp/foo.img", "sda1") self.generator = LibVirtXMLConfigGenerator()