Qubes/core-admin
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

qubes/tests: add event asserts for qubes module

Browse Source
This commit is contained in:
Wojtek Porczyk 2015-01-08 17:32:45 +01:00
parent 74c3126b80
commit 713ced8cd2
2 changed files with 28 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
qubes/tests/__init__.py
View File

@ -72,6 +72,21 @@ class QubesTestCase(unittest.TestCase):
self._testMethodName) self._testMethodName)
def assertXMLEqual(self, xml1, xml2):
'''Check for equality of two XML objects.
:param xml1: first element
:param xml2: second element
:type xml1: :py:class:`lxml.etree._Element`
:type xml2: :py:class:`lxml.etree._Element`
'''
self.assertEqual(xml1.tag, xml2.tag)
self.assertEqual(xml1.text, xml2.text)
self.assertItemsEqual(xml1.keys(), xml2.keys())
for key in xml1.keys():
self.assertEqual(xml1.get(key), xml2.get(key))
def assertEventFired(self, emitter, event, args=[], kwargs=[]): def assertEventFired(self, emitter, event, args=[], kwargs=[]):
'''Check whether event was fired on given emitter and fail if it did '''Check whether event was fired on given emitter and fail if it did
not. not.

24
qubes/tests/init.py
View File

@ -140,20 +140,14 @@ class TC_10_property(qubes.tests.QubesTestCase):
self.assertEqual(holder.testprop1, 'defaultvalue') self.assertEqual(holder.testprop1, 'defaultvalue')
class TestHolder(qubes.PropertyHolder): class TestHolder(qubes.tests.TestEmitter, qubes.PropertyHolder):
testprop1 = qubes.property('testprop1', order=0) testprop1 = qubes.property('testprop1', order=0)
testprop2 = qubes.property('testprop2', order=1, save_via_ref=True) testprop2 = qubes.property('testprop2', order=1, save_via_ref=True)
testprop3 = qubes.property('testprop3', order=2, default='testdefault') testprop3 = qubes.property('testprop3', order=2, default='testdefault')
testprop4 = qubes.property('testprop4', order=3) testprop4 = qubes.property('testprop4', order=3)
class TC_20_PropertyHolder(qubes.tests.QubesTestCase):
def assertXMLEqual(self, xml1, xml2):
self.assertEqual(xml1.tag, xml2.tag)
self.assertEqual(xml1.text, xml2.text)
self.assertEqual(sorted(xml1.keys()), sorted(xml2.keys()))
for key in xml1.keys():
self.assertEqual(xml1.get(key), xml2.get(key))
class TC_20_PropertyHolder(qubes.tests.QubesTestCase):
def setUp(self): def setUp(self):
xml = lxml.etree.XML(''' xml = lxml.etree.XML('''
<qubes version="3"> <qubes version="3">
@ -166,8 +160,13 @@ class TC_20_PropertyHolder(qubes.tests.QubesTestCase):
self.holder = TestHolder(xml) self.holder = TestHolder(xml)
def test_000_load_properties(self): def test_000_load_properties(self):
self.holder.load_properties() 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.testprop1, 'testvalue1')
self.assertEquals(self.holder.testprop2, 'testref2') self.assertEquals(self.holder.testprop2, 'testref2')
self.assertEquals(self.holder.testprop3, 'testdefault') self.assertEquals(self.holder.testprop3, 'testdefault')
@ -201,13 +200,13 @@ class TestVM(qubes.vm.BaseVM):
name = qubes.property('name') name = qubes.property('name')
netid = qid netid = qid
class TestApp(qubes.events.Emitter): class TestApp(qubes.tests.TestEmitter):
pass pass
class TC_30_VMCollection(qubes.tests.QubesTestCase): class TC_30_VMCollection(qubes.tests.QubesTestCase):
def setUp(self): def setUp(self):
# XXX passing None may be wrong in the future self.app = TestApp()
self.vms = qubes.VMCollection(TestApp()) self.vms = qubes.VMCollection(self.app)
self.testvm1 = TestVM(None, None, qid=1, name='testvm1') self.testvm1 = TestVM(None, None, qid=1, name='testvm1')
self.testvm2 = TestVM(None, None, qid=2, name='testvm2') self.testvm2 = TestVM(None, None, qid=2, name='testvm2')
@ -234,6 +233,8 @@ class TC_30_VMCollection(qubes.tests.QubesTestCase):
self.vms.add(self.testvm1) self.vms.add(self.testvm1)
self.assertIn(1, self.vms) self.assertIn(1, self.vms)
self.assertEventFired(self.app, 'domain-added', args=[self.testvm1])
with self.assertRaises(TypeError): with self.assertRaises(TypeError):
self.vms.add(object()) self.vms.add(object())
@ -284,6 +285,7 @@ class TC_30_VMCollection(qubes.tests.QubesTestCase):
del self.vms['testvm2'] del self.vms['testvm2']
self.assertItemsEqual(self.vms.vms(), [self.testvm1]) self.assertItemsEqual(self.vms.vms(), [self.testvm1])
self.assertEventFired(self.app, 'domain-deleted', args=[self.testvm2])
def test_100_get_new_unused_qid(self): def test_100_get_new_unused_qid(self):
self.vms.add(self.testvm1) self.vms.add(self.testvm1)