tests: Allow claiming in secondary bus helper

This saves an extra step when creating tests that disconnect.
This commit is contained in:
Benjamin Berg
2020-12-16 13:39:20 +01:00
parent f75e800d5c
commit 8057e49d31

View File

@ -464,7 +464,7 @@ class FPrintdVirtualDeviceBaseTest(FPrintdTest):
return (enrolled, enroll_map)
def get_secondary_bus_and_device(self):
def get_secondary_bus_and_device(self, claim=None):
addr = os.environ['DBUS_SYSTEM_BUS_ADDRESS']
# Get a separat bus connection
@ -482,6 +482,9 @@ class FPrintdVirtualDeviceBaseTest(FPrintdTest):
'net.reactivated.Fprint.Device',
None)
if claim is not None:
dev.Claim('(s)', claim)
return bus, dev