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

tests: mgmt.Events

Browse Source 
QubesOS/qubes-issues#2622
This commit is contained in:
Marek Marczykowski-Górecki 2017-04-13 15:31:19 +02:00
parent 73df1b8ea7
commit 8cc0af1eda
No known key found for this signature in database
GPG Key ID: 063938BA42CFA724
1 changed files with 27 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
qubes/tests/mgmt.py
View File

@ -21,6 +21,7 @@
''' Tests for management calls endpoints '''
import asyncio
import libvirt
import unittest.mock
@ -836,6 +837,28 @@ class TC_00_VMs(MgmtTestCase):
self.assertIsNone(value)
func_mock.assert_called_once_with()
def test_270_events(self):
send_event = unittest.mock.Mock()
mgmt_obj = qubes.mgmt.QubesMgmt(self.app, b'dom0', b'mgmt.Events',
b'dom0', b'', send_event=send_event)
@asyncio.coroutine
def fire_event():
self.vm.fire_event('test-event', arg1='abc')
mgmt_obj.cancel()
loop = asyncio.get_event_loop()
execute_task = asyncio.ensure_future(
mgmt_obj.execute(untrusted_payload=b''))
asyncio.ensure_future(fire_event())
loop.run_until_complete(execute_task)
self.assertIsNone(execute_task.result())
self.assertEventFired(self.emitter,
'mgmt-permission:' + 'mgmt.Events')
self.assertEqual(send_event.mock_calls,
[unittest.mock.call(self.vm, 'test-event', arg1='abc')])
def test_990_vm_unexpected_payload(self):
methods_with_no_payload = [
b'mgmt.vm.List',
@ -871,6 +894,7 @@ class TC_00_VMs(MgmtTestCase):
b'mgmt.vm.Pause',
b'mgmt.vm.Unpause',
b'mgmt.vm.Kill',
b'mgmt.Events',
]
# make sure also no methods on actual VM gets called
vm_mock = unittest.mock.MagicMock()
@ -915,6 +939,7 @@ class TC_00_VMs(MgmtTestCase):
b'mgmt.vm.Pause',
b'mgmt.vm.Unpause',
b'mgmt.vm.Kill',
b'mgmt.Events',
]
# make sure also no methods on actual VM gets called
vm_mock = unittest.mock.MagicMock()
@ -955,6 +980,7 @@ class TC_00_VMs(MgmtTestCase):
b'mgmt.pool.Info',
b'mgmt.pool.Remove',
b'mgmt.backup.Execute',
b'mgmt.Events',
]
# make sure also no methods on actual VM gets called
vm_mock = unittest.mock.MagicMock()
@ -986,6 +1012,7 @@ class TC_00_VMs(MgmtTestCase):
b'mgmt.property.List',
b'mgmt.pool.List',
b'mgmt.pool.ListDrivers',
b'mgmt.Events',
]
# make sure also no methods on actual VM gets called
vm_mock = unittest.mock.MagicMock()