Skip to content

Commit 2b6dfb8

Browse files
committed
q-dev: update admin api device tests
1 parent 4539f63 commit 2b6dfb8

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

qubes/tests/api_admin.py

+9-2
Original file line numberDiff line numberDiff line change
@@ -1874,16 +1874,22 @@ def test_474_vm_device_list_attached_specific(self):
18741874
"mode='manual' frontend_domain='test-vm1' _attach_opt='value'\n")
18751875
self.assertFalse(self.app.save.called)
18761876

1877+
def get_dev(self, *args, **kwargs):
1878+
yield DeviceInfo(Port(
1879+
self.vm, '1234', 'testclass'),
1880+
device_id='0000:0000::t000000',)
1881+
18771882
def test_480_vm_device_attach(self):
18781883
self.vm.add_handler('device-list:testclass', self.device_list_testclass)
1884+
self.vm.add_handler('device-get:testclass', self.get_dev)
18791885
mock_action = unittest.mock.Mock()
18801886
mock_action.return_value = None
18811887
del mock_action._is_coroutine
18821888
self.vm.add_handler(f'device-attach:testclass', mock_action)
18831889
with unittest.mock.patch.object(qubes.vm.qubesvm.QubesVM,
18841890
'is_halted', lambda _: False):
18851891
value = self.call_mgmt_func(b'admin.vm.device.testclass.Attach',
1886-
b'test-vm1', b'test-vm1+1234:0000:0000::?******')
1892+
b'test-vm1', b'test-vm1+1234:0000:0000::t000000')
18871893
self.assertIsNone(value)
18881894
mock_action.assert_called_once_with(
18891895
self.vm, f'device-attach:testclass',
@@ -1991,14 +1997,15 @@ def test_486_vm_device_assign_required_not_running(self):
19911997

19921998
def test_487_vm_device_attach_options(self):
19931999
self.vm.add_handler('device-list:testclass', self.device_list_testclass)
2000+
self.vm.add_handler('device-get:testclass', self.get_dev)
19942001
mock_attach = unittest.mock.Mock()
19952002
mock_attach.return_value = None
19962003
del mock_attach._is_coroutine
19972004
self.vm.add_handler('device-attach:testclass', mock_attach)
19982005
with unittest.mock.patch.object(qubes.vm.qubesvm.QubesVM,
19992006
'is_halted', lambda _: False):
20002007
value = self.call_mgmt_func(b'admin.vm.device.testclass.Attach',
2001-
b'test-vm1', b'test-vm1+1234:0000:0000::?******',
2008+
b'test-vm1', b'test-vm1+1234:0000:0000::t000000',
20022009
b"_option1='value2'")
20032010
self.assertIsNone(value)
20042011
dev = self.vm.devices['testclass']['1234']

0 commit comments

Comments
 (0)