From 4d0545405f8e053f8493d51ffaa1237fa7d5a6e9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fr=C3=A9d=C3=A9ric=20Pierret=20=28fepitre=29?= Date: Tue, 6 Aug 2019 14:41:23 +0200 Subject: [PATCH] tests/devices: add test for handling listing device classes --- qubesadmin/tests/devices.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/qubesadmin/tests/devices.py b/qubesadmin/tests/devices.py index 6edafd2..164da6f 100644 --- a/qubesadmin/tests/devices.py +++ b/qubesadmin/tests/devices.py @@ -306,3 +306,13 @@ class TC_00_DeviceCollection(qubesadmin.tests.QubesTestCase): self.app.domains['test-vm2'], 'test', 'dev1') self.vm.devices['test'].update_persistent(dev, False) self.assertAllCalled() + + def test_072_list(self): + self.app.expected_calls[ + ('dom0', 'admin.deviceclass.List', None, None)] = \ + b'0\x00block\nmic\nusb\n' + seen = set() + for devclass in self.app.domains['test-vm'].devices: + self.assertNotIn(devclass, seen) + seen.add(devclass) + self.assertEqual(seen, {'block', 'mic', 'usb'})