usbstoragedriver: Allow to define a different waiting medium timeout

Nice to have a way to define a timeout for usb storage driver if
it used after a usb switch. This will make this code more simpler

mux_driver = target.get_driver('LXAUSBMuxDriver')
mux_driver.set_links(['host-device'])
time.sleep(3) <-- avoid to use a new sleep here

usb = target.get_driver('USBStorageDriver')
assert usb.get_size() > 0

Signed-off-by: Michael Trimarchi <[email protected]>

Author: panicking

labgrid/driver/usbstoragedriver.py

@@ -39,15 +39,26 @@ class USBStorageDriver(Driver):
             "NetworkUSBSDWireDevice",
         },
     }
+
+    def __attrs_post_init__(self):
+        super().__attrs_post_init__()
+        self._expect = PtxExpect(self)
+
     image = attr.ib(
         default=None,
         validator=attr.validators.optional(attr.validators.instance_of(str))
     )
-    WAIT_FOR_MEDIUM_TIMEOUT = 10.0 # s
+    timeout = attr.ib(default=0.0, validator=attr.validators.instance_of(float))
+        default=10.0,
+        validator=attr.validators.instance_of(float))
+    )
+
     WAIT_FOR_MEDIUM_SLEEP = 0.5 # s
     MOUNT_RETRIES = 5
 
     def __attrs_post_init__(self):
+        if self.timeout < 0.0:
+            raise ValueError("timeout must be positive")
         super().__attrs_post_init__()
         self.wrapper = None
         self.proxy = None
@@ -208,7 +219,7 @@ def _get_devpath(self, partition):
 
     @Driver.check_active
     def _wait_for_medium(self, partition):
-        timeout = Timeout(self.WAIT_FOR_MEDIUM_TIMEOUT)
+        timeout = Timeout(self.timeout)
         while not timeout.expired:
             if self.get_size(partition) > 0:
                 break