From 8057e49d3147514608a3221fa5ba43697bccbd16 Mon Sep 17 00:00:00 2001
From: Benjamin Berg <bberg@redhat.com>
Date: Wed, 16 Dec 2020 13:39:20 +0100
Subject: [PATCH] tests: Allow claiming in secondary bus helper

This saves an extra step when creating tests that disconnect.
---
 tests/fprintd.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/tests/fprintd.py b/tests/fprintd.py
index 5073dcf..1c635b9 100644
--- a/tests/fprintd.py
+++ b/tests/fprintd.py
@@ -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