#!/usr/bin/python2 -O # vim: fileencoding=utf-8 # # The Qubes OS Project, https://www.qubes-os.org/ # # Copyright (C) 2014-2015 Joanna Rutkowska # Copyright (C) 2014-2015 Wojtek Porczyk # # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation; either version 2 of the License, or # (at your option) any later version. # # This program 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 General Public License for more details. # # You should have received a copy of the GNU General Public License along # with this program; if not, write to the Free Software Foundation, Inc., # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. # import os import unittest import lxml.etree import qubes import qubes.events import qubes.vm import qubes.tests class TC_00_Label(qubes.tests.QubesTestCase): def test_000_constructor(self): label = qubes.Label(1, '#cc0000', 'red') self.assertEqual(label.index, 1) self.assertEqual(label.color, '#cc0000') self.assertEqual(label.name, 'red') self.assertEqual(label.icon, 'appvm-red') self.assertEqual(label.icon_dispvm, 'dispvm-red') def test_001_fromxml(self): xml = lxml.etree.XML(''' ''') node = xml.xpath('//label')[0] label = qubes.Label.fromxml(node) self.assertEqual(label.index, 1) self.assertEqual(label.color, '#cc0000') self.assertEqual(label.name, 'red') self.assertEqual(label.icon, 'appvm-red') self.assertEqual(label.icon_dispvm, 'dispvm-red') class TC_10_property(qubes.tests.QubesTestCase): def setUp(self): try: class TestHolder(qubes.tests.TestEmitter, qubes.PropertyHolder): testprop1 = qubes.property('testprop1') except: self.skipTest('TestHolder class definition failed') self.holder = TestHolder(None) def test_000_init(self): pass def test_001_hash(self): hash(self.holder.__class__.testprop1) def test_002_eq(self): self.assertEquals(qubes.property('testprop2'), qubes.property('testprop2')) def test_010_set(self): self.holder.testprop1 = 'testvalue' self.assertEventFired(self.holder, 'property-pre-set:testprop1') self.assertEventFired(self.holder, 'property-set:testprop1') def test_020_get(self): self.holder.testprop1 = 'testvalue' self.assertEqual(self.holder.testprop1, 'testvalue') def test_021_get_unset(self): with self.assertRaises(AttributeError): self.holder.testprop1 def test_022_get_default(self): class TestHolder(qubes.tests.TestEmitter, qubes.PropertyHolder): testprop1 = qubes.property('testprop1', default='defaultvalue') holder = TestHolder(None) self.assertEqual(holder.testprop1, 'defaultvalue') def test_023_get_default_func(self): class TestHolder(qubes.tests.TestEmitter, qubes.PropertyHolder): testprop1 = qubes.property('testprop1', default=(lambda self: 'defaultvalue')) holder = TestHolder(None) self.assertEqual(holder.testprop1, 'defaultvalue') holder.testprop1 = 'testvalue' self.assertEqual(holder.testprop1, 'testvalue') def test_030_set_setter(self): def setter(self2, prop, value): self.assertIs(self2, holder) self.assertIs(prop, TestHolder.testprop1) self.assertEquals(value, 'testvalue') return 'settervalue' class TestHolder(qubes.tests.TestEmitter, qubes.PropertyHolder): testprop1 = qubes.property('testprop1', setter=setter) holder = TestHolder(None) holder.testprop1 = 'testvalue' self.assertEqual(holder.testprop1, 'settervalue') def test_031_set_type(self): class TestHolder(qubes.tests.TestEmitter, qubes.PropertyHolder): testprop1 = qubes.property('testprop1', type=int) holder = TestHolder(None) holder.testprop1 = '5' self.assertEqual(holder.testprop1, 5) self.assertNotEqual(holder.testprop1, '5') def test_090_delete(self): self.holder.testprop1 = 'testvalue' try: if self.holder.testprop1 != 'testvalue': self.skipTest('testprop1 value is wrong') except AttributeError: self.skipTest('testprop1 value is wrong') del self.holder.testprop1 with self.assertRaises(AttributeError): self.holder.testprop def test_090_delete_by_assign(self): self.holder.testprop1 = 'testvalue' try: if self.holder.testprop1 != 'testvalue': self.skipTest('testprop1 value is wrong') except AttributeError: self.skipTest('testprop1 value is wrong') self.holder.testprop1 = qubes.property.DEFAULT with self.assertRaises(AttributeError): self.holder.testprop def test_092_delete_default(self): class TestHolder(qubes.tests.TestEmitter, qubes.PropertyHolder): testprop1 = qubes.property('testprop1', default='defaultvalue') holder = TestHolder(None) holder.testprop1 = 'testvalue' try: if holder.testprop1 != 'testvalue': self.skipTest('testprop1 value is wrong') except AttributeError: self.skipTest('testprop1 value is wrong') del holder.testprop1 self.assertEqual(holder.testprop1, 'defaultvalue') class TestHolder(qubes.tests.TestEmitter, qubes.PropertyHolder): testprop1 = qubes.property('testprop1', order=0) testprop2 = qubes.property('testprop2', order=1, save_via_ref=True) testprop3 = qubes.property('testprop3', order=2, default='testdefault') testprop4 = qubes.property('testprop4', order=3) class TC_20_PropertyHolder(qubes.tests.QubesTestCase): def setUp(self): xml = lxml.etree.XML(''' testvalue1 ''') self.holder = TestHolder(xml) @unittest.expectedFailure def test_000_load_properties(self): self.holder.load_properties() self.assertEventFired(self.holder, 'property-loaded') self.assertEventNotFired(self.holder, 'property-set:testprop1') self.assertEquals(self.holder.testprop1, 'testvalue1') self.assertEquals(self.holder.testprop2, 'testref2') self.assertEquals(self.holder.testprop3, 'testdefault') with self.assertRaises(AttributeError): self.holder.testprop4 def test_001_save_properties(self): self.holder.load_properties() elements = self.holder.xml_properties() elements_with_defaults = self.holder.xml_properties(with_defaults=True) self.assertEqual(len(elements), 2) self.assertEqual(len(elements_with_defaults), 3) expected_prop1 = lxml.etree.Element('property', name='testprop1') expected_prop1.text = 'testvalue1' self.assertXMLEqual(elements_with_defaults[0], expected_prop1) expected_prop2 = lxml.etree.Element('property', name='testprop2', ref='testref2') self.assertXMLEqual(elements_with_defaults[1], expected_prop2) expected_prop3 = lxml.etree.Element('property', name='testprop3') expected_prop3.text = 'testdefault' self.assertXMLEqual(elements_with_defaults[2], expected_prop3) class TestVM(qubes.vm.BaseVM): qid = qubes.property('qid', type=int) name = qubes.property('name') netid = qid class TestApp(qubes.tests.TestEmitter): pass class TC_30_VMCollection(qubes.tests.QubesTestCase): def setUp(self): self.app = TestApp() self.vms = qubes.VMCollection(self.app) self.testvm1 = TestVM(None, None, qid=1, name='testvm1') self.testvm2 = TestVM(None, None, qid=2, name='testvm2') 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-added', args=[self.testvm1]) with self.assertRaises(TypeError): self.vms.add(object()) testvm_qid_collision = TestVM(None, None, name='testvm2', qid=1) testvm_name_collision = 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.assertItemsEqual(self.vms.qids(), [1, 2]) self.assertItemsEqual(self.vms.keys(), [1, 2]) def test_004_names(self): self.vms.add(self.testvm1) self.vms.add(self.testvm2) self.assertItemsEqual(self.vms.names(), ['testvm1', 'testvm2']) def test_005_vms(self): self.vms.add(self.testvm1) self.vms.add(self.testvm2) self.assertItemsEqual(self.vms.vms(), [self.testvm1, self.testvm2]) self.assertItemsEqual(self.vms.values(), [self.testvm1, self.testvm2]) def test_006_items(self): self.vms.add(self.testvm1) self.vms.add(self.testvm2) self.assertItemsEqual(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.assertItemsEqual(self.vms.vms(), [self.testvm1]) self.assertEventFired(self.app, 'domain-deleted', args=[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_101_get_new_unused_netid(self): self.vms.add(self.testvm1) self.vms.add(self.testvm2) self.vms.get_new_unused_netid() # def test_200_get_vms_based_on(self): # pass # def test_201_get_vms_connected_to(self): # pass class TC_90_Qubes(qubes.tests.QubesTestCase): @qubes.tests.skipUnlessDom0 def test_000_init_empty(self): try: os.unlink('/tmp/qubestest.xml') except: pass app = qubes.Qubes('/tmp/qubestest.xml') @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')