fix up USB/SPI race condition in flasher script

commaai · Dec 4, 2024 · 4c3fb53 · 4c3fb53
1 parent c7cc2de
commit 4c3fb53
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/reflash_internal_panda.py b/tests/reflash_internal_panda.py
@@ -30,18 +30,18 @@ def gpio_set(pin, high):
   print("resetting into DFU")
   gpio_set(GPIO.STM_RST_N, 1)
   gpio_set(GPIO.STM_BOOT0, 1)
-  time.sleep(1)
+  time.sleep(0.5)
   gpio_set(GPIO.STM_RST_N, 0)
   gpio_set(GPIO.STM_BOOT0, 0)
-  time.sleep(1)
+  assert Panda.wait_for_dfu(None, timeout=10)
 
   print("flashing bootstub")
   PandaDFU(None).recover()
 
   gpio_set(GPIO.STM_RST_N, 1)
   time.sleep(0.5)
   gpio_set(GPIO.STM_RST_N, 0)
-  time.sleep(1)
+  assert Panda.wait_for_panda(None, timeout=10)
 
   print("flashing app")
   p = Panda()