123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809 |
- # pylint: disable=protected-access,pointless-statement
- #
- # The Qubes OS Project, https://www.qubes-os.org/
- #
- # Copyright (C) 2014-2015 Joanna Rutkowska <joanna@invisiblethingslab.com>
- # Copyright (C) 2014-2015 Wojtek Porczyk <woju@invisiblethingslab.com>
- #
- # This library is free software; you can redistribute it and/or
- # modify it under the terms of the GNU Lesser General Public
- # License as published by the Free Software Foundation; either
- # version 2.1 of the License, or (at your option) any later version.
- #
- # This library is distributed in the hope that it will be useful,
- # but WITHOUT ANY WARRANTY; without even the implied warranty of
- # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- # Lesser General Public License for more details.
- #
- # You should have received a copy of the GNU Lesser General Public
- # License along with this library; if not, see <https://www.gnu.org/licenses/>.
- #
- import os
- import unittest.mock as mock
- import lxml.etree
- import qubes
- import qubes.events
- import qubes.tests
- import qubes.tests.init
- import qubes.tests.storage_reflink
- class TestApp(qubes.tests.TestEmitter):
- pass
- class TC_20_QubesHost(qubes.tests.QubesTestCase):
- sample_xc_domain_getinfo = [
- {'paused': 0, 'cpu_time': 243951379111104, 'ssidref': 0,
- 'hvm': 0, 'shutdown_reason': 255, 'dying': 0,
- 'mem_kb': 3733212, 'domid': 0, 'max_vcpu_id': 7,
- 'crashed': 0, 'running': 1, 'maxmem_kb': 3734236,
- 'shutdown': 0, 'online_vcpus': 8,
- 'handle': [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
- 'cpupool': 0, 'blocked': 0},
- {'paused': 0, 'cpu_time': 2849496569205, 'ssidref': 0,
- 'hvm': 0, 'shutdown_reason': 255, 'dying': 0,
- 'mem_kb': 303916, 'domid': 1, 'max_vcpu_id': 0,
- 'crashed': 0, 'running': 0, 'maxmem_kb': 308224,
- 'shutdown': 0, 'online_vcpus': 1,
- 'handle': [116, 174, 229, 207, 17, 1, 79, 39, 191, 37, 41,
- 186, 205, 158, 219, 8],
- 'cpupool': 0, 'blocked': 1},
- {'paused': 0, 'cpu_time': 249658663079978, 'ssidref': 0,
- 'hvm': 0, 'shutdown_reason': 255, 'dying': 0,
- 'mem_kb': 3782668, 'domid': 11, 'max_vcpu_id': 7,
- 'crashed': 0, 'running': 0, 'maxmem_kb': 3783692,
- 'shutdown': 0, 'online_vcpus': 8,
- 'handle': [169, 95, 55, 127, 140, 94, 79, 220, 186, 210,
- 117, 5, 148, 11, 185, 206],
- 'cpupool': 0, 'blocked': 1}]
- def setUp(self):
- super(TC_20_QubesHost, self).setUp()
- self.app = TestApp()
- self.app.vmm = mock.Mock()
- self.qubes_host = qubes.app.QubesHost(self.app)
- def test_000_get_vm_stats_single(self):
- self.app.vmm.configure_mock(**{
- 'xc.domain_getinfo.return_value': self.sample_xc_domain_getinfo
- })
- info_time, info = self.qubes_host.get_vm_stats()
- self.assertEqual(self.app.vmm.mock_calls, [
- ('xc.domain_getinfo', (0, 1024), {}),
- ])
- self.assertIsNotNone(info_time)
- expected_info = {
- 0: {
- 'cpu_time': 243951379111104,
- 'cpu_usage': 0,
- 'cpu_usage_raw': 0,
- 'memory_kb': 3733212,
- },
- 1: {
- 'cpu_time': 2849496569205,
- 'cpu_usage': 0,
- 'cpu_usage_raw': 0,
- 'memory_kb': 303916,
- },
- 11: {
- 'cpu_time': 249658663079978,
- 'cpu_usage': 0,
- 'cpu_usage_raw': 0,
- 'memory_kb': 3782668,
- },
- }
- self.assertEqual(info, expected_info)
- def test_001_get_vm_stats_twice(self):
- self.app.vmm.configure_mock(**{
- 'xc.domain_getinfo.return_value': self.sample_xc_domain_getinfo
- })
- prev_time, prev_info = self.qubes_host.get_vm_stats()
- prev_time -= 1
- prev_info[0]['cpu_time'] -= 8 * 10 ** 8 # 0.8s
- prev_info[1]['cpu_time'] -= 10 ** 9 # 1s
- prev_info[11]['cpu_time'] -= 10 ** 9 # 1s
- info_time, info = self.qubes_host.get_vm_stats(prev_time, prev_info)
- self.assertIsNotNone(info_time)
- expected_info = {
- 0: {
- 'cpu_time': 243951379111104,
- 'cpu_usage': 9,
- 'cpu_usage_raw': 79,
- 'memory_kb': 3733212,
- },
- 1: {
- 'cpu_time': 2849496569205,
- 'cpu_usage': 99,
- 'cpu_usage_raw': 99,
- 'memory_kb': 303916,
- },
- 11: {
- 'cpu_time': 249658663079978,
- 'cpu_usage': 12,
- 'cpu_usage_raw': 99,
- 'memory_kb': 3782668,
- },
- }
- self.assertEqual(info, expected_info)
- self.assertEqual(self.app.vmm.mock_calls, [
- ('xc.domain_getinfo', (0, 1024), {}),
- ('xc.domain_getinfo', (0, 1024), {}),
- ])
- def test_002_get_vm_stats_one_vm(self):
- self.app.vmm.configure_mock(**{
- 'xc.domain_getinfo.return_value': [self.sample_xc_domain_getinfo[1]]
- })
- vm = mock.Mock
- vm.xid = 1
- vm.name = 'somevm'
- info_time, info = self.qubes_host.get_vm_stats(only_vm=vm)
- self.assertIsNotNone(info_time)
- self.assertEqual(self.app.vmm.mock_calls, [
- ('xc.domain_getinfo', (1, 1), {}),
- ])
- def test_010_iommu_supported(self):
- self.app.vmm.configure_mock(**{
- 'xc.physinfo.return_value': {
- 'hw_caps': '...',
- 'scrub_memory': 0,
- 'virt_caps': 'hvm hvm_directio',
- 'nr_cpus': 4,
- 'threads_per_core': 1,
- 'cpu_khz': 3400001,
- 'nr_nodes': 1,
- 'free_memory': 234752,
- 'cores_per_socket': 4,
- 'total_memory': 16609720
- }
- })
- self.assertEqual(self.qubes_host.is_iommu_supported(), True)
- def test_011_iommu_supported(self):
- self.app.vmm.configure_mock(**{
- 'xc.physinfo.return_value': {
- 'hw_caps': '...',
- 'scrub_memory': 0,
- 'virt_caps': 'hvm hvm_directio pv pv_directio',
- 'nr_cpus': 4,
- 'threads_per_core': 1,
- 'cpu_khz': 3400001,
- 'nr_nodes': 1,
- 'free_memory': 234752,
- 'cores_per_socket': 4,
- 'total_memory': 16609720
- }
- })
- self.assertEqual(self.qubes_host.is_iommu_supported(), True)
- def test_010_iommu_supported(self):
- self.app.vmm.configure_mock(**{
- 'xc.physinfo.return_value': {
- 'hw_caps': '...',
- 'scrub_memory': 0,
- 'virt_caps': 'hvm pv',
- 'nr_cpus': 4,
- 'threads_per_core': 1,
- 'cpu_khz': 3400001,
- 'nr_nodes': 1,
- 'free_memory': 234752,
- 'cores_per_socket': 4,
- 'total_memory': 16609720
- }
- })
- self.assertEqual(self.qubes_host.is_iommu_supported(), False)
- class TC_30_VMCollection(qubes.tests.QubesTestCase):
- def setUp(self):
- super().setUp()
- self.app = TestApp()
- self.vms = qubes.app.VMCollection(self.app)
- self.testvm1 = qubes.tests.init.TestVM(
- None, None, qid=1, name='testvm1')
- self.testvm2 = qubes.tests.init.TestVM(
- None, None, qid=2, name='testvm2')
- self.addCleanup(self.cleanup_vmcollection)
- def cleanup_vmcollection(self):
- self.testvm1.close()
- self.testvm2.close()
- self.vms.close()
- del self.testvm1
- del self.testvm2
- del self.vms
- del self.app
- def test_000_contains(self):
- self.vms._dict = {1: self.testvm1}
- self.assertIn(1, self.vms)
- self.assertIn('testvm1', self.vms)
- self.assertIn(self.testvm1, self.vms)
- self.assertNotIn(2, self.vms)
- self.assertNotIn('testvm2', self.vms)
- self.assertNotIn(self.testvm2, self.vms)
- def test_001_getitem(self):
- self.vms._dict = {1: self.testvm1}
- self.assertIs(self.vms[1], self.testvm1)
- self.assertIs(self.vms['testvm1'], self.testvm1)
- self.assertIs(self.vms[self.testvm1], self.testvm1)
- def test_002_add(self):
- self.vms.add(self.testvm1)
- self.assertIn(1, self.vms)
- self.assertEventFired(self.app, 'domain-add',
- kwargs={'vm': self.testvm1})
- with self.assertRaises(TypeError):
- self.vms.add(object())
- testvm_qid_collision = qubes.tests.init.TestVM(
- None, None, name='testvm2', qid=1)
- testvm_name_collision = qubes.tests.init.TestVM(
- None, None, name='testvm1', qid=2)
- with self.assertRaises(ValueError):
- self.vms.add(testvm_qid_collision)
- with self.assertRaises(ValueError):
- self.vms.add(testvm_name_collision)
- def test_003_qids(self):
- self.vms.add(self.testvm1)
- self.vms.add(self.testvm2)
- self.assertCountEqual(self.vms.qids(), [1, 2])
- self.assertCountEqual(self.vms.keys(), [1, 2])
- def test_004_names(self):
- self.vms.add(self.testvm1)
- self.vms.add(self.testvm2)
- self.assertCountEqual(self.vms.names(), ['testvm1', 'testvm2'])
- def test_005_vms(self):
- self.vms.add(self.testvm1)
- self.vms.add(self.testvm2)
- self.assertCountEqual(self.vms.vms(), [self.testvm1, self.testvm2])
- self.assertCountEqual(self.vms.values(), [self.testvm1, self.testvm2])
- def test_006_items(self):
- self.vms.add(self.testvm1)
- self.vms.add(self.testvm2)
- self.assertCountEqual(self.vms.items(),
- [(1, self.testvm1), (2, self.testvm2)])
- def test_007_len(self):
- self.vms.add(self.testvm1)
- self.vms.add(self.testvm2)
- self.assertEqual(len(self.vms), 2)
- def test_008_delitem(self):
- self.vms.add(self.testvm1)
- self.vms.add(self.testvm2)
- del self.vms['testvm2']
- self.assertCountEqual(self.vms.vms(), [self.testvm1])
- self.assertEventFired(self.app, 'domain-delete',
- kwargs={'vm': self.testvm2})
- def test_100_get_new_unused_qid(self):
- self.vms.add(self.testvm1)
- self.vms.add(self.testvm2)
- self.vms.get_new_unused_qid()
- # def test_200_get_vms_based_on(self):
- # pass
- # def test_201_get_vms_connected_to(self):
- # pass
- class TC_80_QubesInitialPools(qubes.tests.QubesTestCase):
- def setUp(self):
- super().setUp()
- self.app = qubes.Qubes('/tmp/qubestest.xml', load=False,
- offline_mode=True)
- self.test_dir = '/var/tmp/test-varlibqubes'
- self.test_patch = mock.patch.dict(
- qubes.config.defaults['pool_configs']['varlibqubes'],
- {'dir_path': self.test_dir})
- self.test_patch.start()
- def tearDown(self):
- self.test_patch.stop()
- self.app.close()
- del self.app
- def get_driver(self, fs_type, accessible):
- qubes.tests.storage_reflink.mkdir_fs(self.test_dir, fs_type,
- accessible=accessible,
- cleanup_via=self.addCleanup)
- self.app.load_initial_values()
- varlibqubes = self.app.pools['varlibqubes']
- self.assertEqual(varlibqubes.dir_path, self.test_dir)
- return varlibqubes.driver
- def test_100_varlibqubes_btrfs_accessible(self):
- self.assertEqual(self.get_driver('btrfs', True), 'file-reflink')
- def test_101_varlibqubes_btrfs_inaccessible(self):
- self.assertEqual(self.get_driver('btrfs', False), 'file')
- def test_102_varlibqubes_ext4_accessible(self):
- self.assertEqual(self.get_driver('ext4', True), 'file')
- def test_103_varlibqubes_ext4_inaccessible(self):
- self.assertEqual(self.get_driver('ext4', False), 'file')
- class TC_89_QubesEmpty(qubes.tests.QubesTestCase):
- def tearDown(self):
- try:
- os.unlink('/tmp/qubestest.xml')
- except:
- pass
- try:
- self.app.close()
- del self.app
- except AttributeError:
- pass
- super().tearDown()
- @qubes.tests.skipUnlessDom0
- def test_000_init_empty(self):
- # pylint: disable=no-self-use,unused-variable,bare-except
- try:
- os.unlink('/tmp/qubestest.xml')
- except FileNotFoundError:
- pass
- qubes.Qubes.create_empty_store('/tmp/qubestest.xml').close()
- def test_100_property_migrate_default_fw_netvm(self):
- xml_template = """<?xml version="1.0" encoding="utf-8" ?>
- <qubes version="3.0">
- <properties>
- <property name="default_netvm">{default_netvm}</property>
- <property name="default_fw_netvm">{default_fw_netvm}</property>
- </properties>
- <labels>
- <label id="label-1" color="#cc0000">red</label>
- </labels>
- <pools>
- <pool driver="file" dir_path="/tmp/qubes-test" name="default"/>
- </pools>
- <domains>
- <domain class="StandaloneVM" id="domain-1">
- <properties>
- <property name="qid">1</property>
- <property name="name">sys-net</property>
- <property name="provides_network">True</property>
- <property name="label" ref="label-1" />
- <property name="netvm"></property>
- <property name="uuid">2fcfc1f4-b2fe-4361-931a-c5294b35edfa</property>
- </properties>
- <features/>
- <devices class="pci"/>
- </domain>
- <domain class="StandaloneVM" id="domain-2">
- <properties>
- <property name="qid">2</property>
- <property name="name">sys-firewall</property>
- <property name="provides_network">True</property>
- <property name="label" ref="label-1" />
- <property name="uuid">9a6d9689-25f7-48c9-a15f-8205d6c5b7c6</property>
- </properties>
- </domain>
- <domain class="StandaloneVM" id="domain-3">
- <properties>
- <property name="qid">3</property>
- <property name="name">appvm</property>
- <property name="label" ref="label-1" />
- <property name="uuid">1d6aab41-3262-400a-b3d3-21aae8fdbec8</property>
- </properties>
- </domain>
- </domains>
- </qubes>
- """
- with self.subTest('default_setup'):
- with open('/tmp/qubestest.xml', 'w') as xml_file:
- xml_file.write(xml_template.format(
- default_netvm='sys-firewall',
- default_fw_netvm='sys-net'))
- self.app = qubes.Qubes('/tmp/qubestest.xml', offline_mode=True)
- self.assertEqual(
- self.app.domains['sys-net'].netvm, None)
- self.assertEqual(
- self.app.domains['sys-firewall'].netvm,
- self.app.domains['sys-net'])
- # property is no longer "default"
- self.assertFalse(
- self.app.domains['sys-firewall'].property_is_default('netvm'))
- # verify that appvm.netvm is unaffected
- self.assertTrue(
- self.app.domains['appvm'].property_is_default('netvm'))
- self.assertEqual(
- self.app.domains['appvm'].netvm,
- self.app.domains['sys-firewall'])
- with self.assertRaises(AttributeError):
- self.app.default_fw_netvm
- self.app.close()
- del self.app
- with self.subTest('same'):
- with open('/tmp/qubestest.xml', 'w') as xml_file:
- xml_file.write(xml_template.format(
- default_netvm='sys-net',
- default_fw_netvm='sys-net'))
- self.app = qubes.Qubes('/tmp/qubestest.xml', offline_mode=True)
- self.assertEqual(
- self.app.domains['sys-net'].netvm, None)
- self.assertEqual(
- self.app.domains['sys-firewall'].netvm,
- self.app.domains['sys-net'])
- self.assertTrue(
- self.app.domains['sys-firewall'].property_is_default('netvm'))
- # verify that appvm.netvm is unaffected
- self.assertTrue(
- self.app.domains['appvm'].property_is_default('netvm'))
- self.assertEqual(
- self.app.domains['appvm'].netvm,
- self.app.domains['sys-net'])
- with self.assertRaises(AttributeError):
- self.app.default_fw_netvm
- self.app.close()
- del self.app
- with self.subTest('loop'):
- with open('/tmp/qubestest.xml', 'w') as xml_file:
- xml_file.write(xml_template.format(
- default_netvm='sys-firewall',
- default_fw_netvm='sys-firewall'))
- self.app = qubes.Qubes('/tmp/qubestest.xml', offline_mode=True)
- self.assertEqual(
- self.app.domains['sys-net'].netvm, None)
- # this was netvm loop, better set to none, to not crash qubesd
- self.assertEqual(
- self.app.domains['sys-firewall'].netvm, None)
- self.assertFalse(
- self.app.domains['sys-firewall'].property_is_default('netvm'))
- # verify that appvm.netvm is unaffected
- self.assertTrue(
- self.app.domains['appvm'].property_is_default('netvm'))
- self.assertEqual(
- self.app.domains['appvm'].netvm,
- self.app.domains['sys-firewall'])
- with self.assertRaises(AttributeError):
- self.app.default_fw_netvm
- self.app.close()
- del self.app
- def test_101_property_migrate_label(self):
- xml_template = """<?xml version="1.0" encoding="utf-8" ?>
- <qubes version="3.0">
- <labels>
- <label id="label-1" color="{old_gray}">gray</label>
- </labels>
- <pools>
- <pool driver="file" dir_path="/tmp/qubes-test" name="default"/>
- </pools>
- <domains>
- <domain class="StandaloneVM" id="domain-1">
- <properties>
- <property name="qid">1</property>
- <property name="name">sys-net</property>
- <property name="provides_network">True</property>
- <property name="label" ref="label-1" />
- <property name="netvm"></property>
- <property name="uuid">2fcfc1f4-b2fe-4361-931a-c5294b35edfa</property>
- </properties>
- <features/>
- <devices class="pci"/>
- </domain>
- </domains>
- </qubes>
- """
- with self.subTest('replace_label'):
- with open('/tmp/qubestest.xml', 'w') as xml_file:
- xml_file.write(xml_template.format(old_gray='0x555753'))
- self.app = qubes.Qubes('/tmp/qubestest.xml', offline_mode=True)
- self.assertEqual(
- self.app.get_label('gray').color, '0x555555')
- self.app.close()
- del self.app
- with self.subTest('dont_replace_label'):
- with open('/tmp/qubestest.xml', 'w') as xml_file:
- xml_file.write(xml_template.format(old_gray='0x123456'))
- self.app = qubes.Qubes('/tmp/qubestest.xml', offline_mode=True)
- self.assertEqual(
- self.app.get_label('gray').color, '0x123456')
- self.app.close()
- del self.app
- class TC_90_Qubes(qubes.tests.QubesTestCase):
- def tearDown(self):
- try:
- os.unlink('/tmp/qubestest.xml')
- except:
- pass
- super().tearDown()
- def setUp(self):
- super(TC_90_Qubes, self).setUp()
- self.app = qubes.Qubes('/tmp/qubestest.xml', load=False,
- offline_mode=True)
- self.addCleanup(self.cleanup_qubes)
- self.app.load_initial_values()
- self.template = self.app.add_new_vm('TemplateVM', name='test-template',
- label='green')
- def cleanup_qubes(self):
- self.app.close()
- del self.app
- try:
- del self.template
- except AttributeError:
- pass
- def test_100_clockvm(self):
- appvm = self.app.add_new_vm('AppVM', name='test-vm',
- template=self.template,
- label='red')
- self.assertIsNone(self.app.clockvm)
- self.assertNotIn('service.clocksync', appvm.features)
- self.assertNotIn('service.clocksync', self.template.features)
- self.app.clockvm = appvm
- self.assertIn('service.clocksync', appvm.features)
- self.assertTrue(appvm.features['service.clocksync'])
- self.app.clockvm = self.template
- self.assertNotIn('service.clocksync', appvm.features)
- self.assertIn('service.clocksync', self.template.features)
- self.assertTrue(self.template.features['service.clocksync'])
- def test_110_netvm_loop(self):
- """Netvm loop through default_netvm"""
- netvm = self.app.add_new_vm('AppVM', name='test-net',
- template=self.template, label='red')
- try:
- self.app.default_netvm = None
- netvm.netvm = qubes.property.DEFAULT
- with self.assertRaises(ValueError):
- self.app.default_netvm = netvm
- finally:
- del netvm
- def test_111_netvm_loop(self):
- """Netvm loop through default_netvm"""
- netvm = self.app.add_new_vm('AppVM', name='test-net',
- template=self.template, label='red')
- try:
- netvm.netvm = None
- self.app.default_netvm = netvm
- with self.assertRaises(ValueError):
- netvm.netvm = qubes.property.DEFAULT
- finally:
- del netvm
- def test_112_default_guivm(self):
- class MyTestHolder(qubes.tests.TestEmitter, qubes.PropertyHolder):
- default_guivm = qubes.property('default_guivm',
- default=(lambda self: 'dom0'))
- holder = MyTestHolder(None)
- guivm = self.app.add_new_vm('AppVM', name='sys-gui', guivm='dom0',
- template=self.template, label='red')
- appvm = self.app.add_new_vm('AppVM', name='test-vm',
- template=self.template, label='red')
- holder.default_guivm = 'sys-gui'
- self.assertEqual(holder.default_guivm, 'sys-gui')
- self.assertIsNotNone(self.app.default_guivm)
- self.assertTrue(appvm.property_is_default('guivm'))
- self.app.default_guivm = guivm
- self.assertEventFired(holder, 'property-set:default_guivm',
- kwargs={'name': 'default_guivm',
- 'newvalue': 'sys-gui'})
- self.assertIn('guivm-sys-gui', appvm.tags)
- def test_113_guivm(self):
- class MyTestHolder(qubes.tests.TestEmitter, qubes.PropertyHolder):
- guivm = qubes.property('guivm',
- default=(lambda self: 'dom0'))
- holder = MyTestHolder(None)
- guivm = self.app.add_new_vm('AppVM', name='sys-gui', guivm='dom0',
- template=self.template, label='red')
- vncvm = self.app.add_new_vm('AppVM', name='sys-vnc', guivm='dom0',
- template=self.template, label='red')
- appvm = self.app.add_new_vm('AppVM', name='test-vm', guivm='dom0',
- template=self.template, label='red')
- holder.guivm = 'sys-gui'
- self.assertEqual(holder.guivm, 'sys-gui')
- self.assertEventFired(holder, 'property-set:guivm',
- kwargs={'name': 'guivm',
- 'newvalue': 'sys-gui'})
- # Set GuiVM
- self.assertFalse(appvm.property_is_default('guivm'))
- appvm.guivm = guivm
- self.assertIn('guivm-sys-gui', appvm.tags)
- # Change GuiVM
- appvm.guivm = vncvm
- self.assertIn('guivm-sys-vnc', appvm.tags)
- self.assertNotIn('guivm-sys-gui', appvm.tags)
- # Empty GuiVM
- del appvm.guivm
- self.assertNotIn('guivm-sys-vnc', appvm.tags)
- self.assertNotIn('guivm-sys-gui', appvm.tags)
- self.assertNotIn('guivm-', appvm.tags)
- def test_114_default_audiovm(self):
- class MyTestHolder(qubes.tests.TestEmitter, qubes.PropertyHolder):
- default_audiovm = qubes.property('default_audiovm',
- default=(lambda self: 'dom0'))
- holder = MyTestHolder(None)
- audiovm = self.app.add_new_vm('AppVM', name='sys-audio', audiovm='dom0',
- template=self.template, label='red')
- appvm = self.app.add_new_vm('AppVM', name='test-vm',
- template=self.template, label='red')
- holder.default_audiovm = 'sys-audio'
- self.assertEqual(holder.default_audiovm, 'sys-audio')
- self.assertIsNotNone(self.app.default_audiovm)
- self.assertTrue(appvm.property_is_default('audiovm'))
- self.app.default_audiovm = audiovm
- self.assertEventFired(holder, 'property-set:default_audiovm',
- kwargs={'name': 'default_audiovm',
- 'newvalue': 'sys-audio'})
- self.assertIn('audiovm-sys-audio', appvm.tags)
- def test_115_audiovm(self):
- class MyTestHolder(qubes.tests.TestEmitter, qubes.PropertyHolder):
- audiovm = qubes.property('audiovm',
- default=(lambda self: 'dom0'))
- holder = MyTestHolder(None)
- audiovm = self.app.add_new_vm('AppVM', name='sys-audio', audiovm='dom0',
- template=self.template, label='red')
- guivm = self.app.add_new_vm('AppVM', name='sys-gui', audiovm='dom0',
- template=self.template, label='red')
- appvm = self.app.add_new_vm('AppVM', name='test-vm', audiovm='dom0',
- template=self.template, label='red')
- holder.audiovm = 'sys-audio'
- self.assertEqual(holder.audiovm, 'sys-audio')
- self.assertEventFired(holder, 'property-set:audiovm',
- kwargs={'name': 'audiovm',
- 'newvalue': 'sys-audio'})
- # Set AudioVM
- self.assertFalse(appvm.property_is_default('audiovm'))
- appvm.audiovm = audiovm
- self.assertIn('audiovm-sys-audio', appvm.tags)
- # Change AudioVM
- appvm.audiovm = guivm
- self.assertIn('audiovm-sys-gui', appvm.tags)
- self.assertNotIn('audiovm-sys-audio', appvm.tags)
- # Empty AudioVM
- del appvm.audiovm
- self.assertNotIn('audiovm-sys-gui', appvm.tags)
- self.assertNotIn('audiovm-sys-audio', appvm.tags)
- self.assertNotIn('audiovm-', appvm.tags)
- def test_200_remove_template(self):
- appvm = self.app.add_new_vm('AppVM', name='test-vm',
- template=self.template,
- label='red')
- with mock.patch.object(self.app, 'vmm'):
- with self.assertRaises(qubes.exc.QubesException):
- del self.app.domains[self.template]
- def test_201_remove_netvm(self):
- netvm = self.app.add_new_vm('AppVM', name='test-netvm',
- template=self.template,
- provides_network=True,
- label='red')
- appvm = self.app.add_new_vm('AppVM', name='test-vm',
- template=self.template,
- label='red')
- appvm.netvm = netvm
- with mock.patch.object(self.app, 'vmm'):
- with self.assertRaises(qubes.exc.QubesVMInUseError):
- del self.app.domains[netvm]
- def test_202_remove_default_netvm(self):
- netvm = self.app.add_new_vm('AppVM', name='test-netvm',
- template=self.template,
- provides_network=True,
- label='red')
- netvm.netvm = None
- self.app.default_netvm = netvm
- with mock.patch.object(self.app, 'vmm'):
- with self.assertRaises(qubes.exc.QubesVMInUseError):
- del self.app.domains[netvm]
- def test_203_remove_default_dispvm(self):
- appvm = self.app.add_new_vm('AppVM', name='test-appvm',
- template=self.template,
- label='red')
- self.app.default_dispvm = appvm
- with mock.patch.object(self.app, 'vmm'):
- with self.assertRaises(qubes.exc.QubesVMInUseError):
- del self.app.domains[appvm]
- def test_204_remove_appvm_dispvm(self):
- dispvm = self.app.add_new_vm('AppVM', name='test-appvm',
- template=self.template,
- label='red')
- appvm = self.app.add_new_vm('AppVM', name='test-appvm2',
- template=self.template,
- default_dispvm=dispvm,
- label='red')
- with mock.patch.object(self.app, 'vmm'):
- with self.assertRaises(qubes.exc.QubesVMInUseError):
- del self.app.domains[dispvm]
- def test_205_remove_appvm_dispvm(self):
- appvm = self.app.add_new_vm('AppVM', name='test-appvm',
- template=self.template,
- template_for_dispvms=True,
- label='red')
- dispvm = self.app.add_new_vm('DispVM', name='test-dispvm',
- template=appvm,
- label='red')
- with mock.patch.object(self.app, 'vmm'):
- with self.assertRaises(qubes.exc.QubesVMInUseError):
- del self.app.domains[appvm]
- def test_206_remove_attached(self):
- # See also qubes.tests.api_admin.
- vm = self.app.add_new_vm(
- 'AppVM', name='test-vm', template=self.template, label='red')
- assignment = mock.Mock(ident='1234')
- vm.get_provided_assignments = lambda: [assignment]
- with self.assertRaises(qubes.exc.QubesVMInUseError):
- del self.app.domains[vm]
- @qubes.tests.skipUnlessGit
- def test_900_example_xml_in_doc(self):
- self.assertXMLIsValid(
- lxml.etree.parse(open(
- os.path.join(qubes.tests.in_git, 'doc/example.xml'), 'rb')),
- 'qubes.rng')
|