mirror of https://github.com/commaai/panda.git
43 lines
1.2 KiB
Python
Executable File
43 lines
1.2 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import concurrent.futures
|
|
|
|
from panda import PandaJungle, PandaJungleDFU, McuType
|
|
from panda.tests.libs.resetter import Resetter
|
|
|
|
SERIALS = {'180019001451313236343430', '1d0017000c50435635333720'}
|
|
|
|
def recover(s):
|
|
with PandaJungleDFU(s) as pd:
|
|
pd.recover()
|
|
|
|
def flash(s):
|
|
with PandaJungle(s) as p:
|
|
p.flash()
|
|
return p.get_mcu_type()
|
|
|
|
# Reset + flash all CI hardware to get it into a consistent state
|
|
# * port 1: jungles-under-test
|
|
# * port 2: USB hubs
|
|
# * port 3: HITL pandas and their jungles
|
|
if __name__ == "__main__":
|
|
with Resetter() as r:
|
|
# everything off
|
|
for i in range(1, 4):
|
|
r.enable_power(i, 0)
|
|
r.cycle_power(ports=[1, 2], dfu=True)
|
|
|
|
dfu_serials = PandaJungleDFU.list()
|
|
print(len(dfu_serials), len(SERIALS))
|
|
assert len(dfu_serials) == len(SERIALS)
|
|
|
|
with concurrent.futures.ProcessPoolExecutor(max_workers=len(dfu_serials)) as exc:
|
|
list(exc.map(recover, dfu_serials, timeout=30))
|
|
|
|
# power cycle for H7 bootloader bug
|
|
r.cycle_power(ports=[1, 2])
|
|
|
|
serials = PandaJungle.list()
|
|
assert set(PandaJungle.list()) >= SERIALS
|
|
mcu_types = list(exc.map(flash, SERIALS, timeout=20))
|
|
assert set(mcu_types) == {McuType.F4, McuType.H7}
|