from __future__ import annotations
import enum
import logging
import pathlib
import time
import typing
from .lib_tentacle import TentacleBase
from .usb_tentacle.usb_constants import Switch
from .usb_tentacle.usb_tentacle import UsbTentacles
from .util_baseclasses import OctoprobeAppExitException
from .util_pyudev import UdevPoller
logger = logging.getLogger(__file__)
FULL_POWERCYCLE_ALL_TENTACLES = False
[docs]
class CtxTestRun:
"""
The context of a test run
"""
def __init__(self, connected_tentacles: typing.Sequence[TentacleBase]) -> None:
assert isinstance(connected_tentacles, list)
self.connected_tentacles = connected_tentacles
[docs]
@staticmethod
def session_powercycle_tentacles() -> UsbTentacles:
"""
Powers all Pico infra.
Finds all tentacle by finding pico_unique_id of the Pico infra.
"""
# We have to reset the power for all pico-infra to become visible
return UsbTentacles.query(poweron=FULL_POWERCYCLE_ALL_TENTACLES)
def session_teardown(self) -> None:
pass
def function_setup_infa_and_dut(
self,
udev_poller: UdevPoller,
tentacle: TentacleBase,
directory_logs: pathlib.Path,
) -> None:
self.function_setup_infra(udev_poller=udev_poller, tentacle=tentacle)
self.function_prepare_dut(tentacle=tentacle)
self.function_setup_dut_flash(
udev_poller=udev_poller,
tentacle=tentacle,
directory_logs=directory_logs,
)
[docs]
def function_setup_infra(
self,
udev_poller: UdevPoller,
tentacle: TentacleBase,
) -> None:
"""
Power off all other known usb power switchs.
For each active tentacle:
* Power on infa
* Flash firmware
* Get serial numbers and assert if it does not match the config
* Return tty
"""
# Instantiate poller BEFORE switching on power to avoid a race condition
tentacle.infra.setup_infra(udev_poller)
# tentacle.infra.mcu_infra.active_led(on=False)
tentacle.infra.switches.dut = False
tentacle.verify_hw_version()
if not FULL_POWERCYCLE_ALL_TENTACLES:
# As the tentacle infra has NOT been powercycled, we
# have to reset the relays
tentacle.infra.switches.relays(
relays_close=[],
relays_open=tentacle.infra.list_all_relays,
)
def function_prepare_dut(self, tentacle: TentacleBase) -> None:
# Why close the infra mp_remote?
if tentacle.is_mcu:
tentacle.dut.mp_remote_close()
changed = tentacle.switches[Switch.DUT].set(on=False)
if changed:
# Give the DUT some time to power off
time.sleep(0.5)
def function_setup_dut_flash(
self,
udev_poller: UdevPoller,
tentacle: TentacleBase,
directory_logs: pathlib.Path,
) -> None:
if not tentacle.is_mcu:
return
# Flash the MCU
tentacle.flash_dut(
udev_poller=udev_poller,
directory_logs=directory_logs,
firmware_spec=tentacle.tentacle_state.firmware_spec,
)
def function_teardown(
self, active_tentacles: typing.Sequence[TentacleBase]
) -> None:
if False:
for tentacle in active_tentacles:
tentacle.power_dut_off_and_wait()
def ping_tentacle_infra(tentacle: TentacleBase, tag: str) -> None:
try:
tentacle.infra.mp_remote.read_bool("True")
logger.debug(f"{tentacle.label_short}: {tag}: Ping succeeded")
except Exception as e:
logger.warning(
f"{tentacle.label_short}: {tag}: Ping failed!", exc_info=e
)
do_ping = False
for tentacle in active_tentacles:
try:
tentacle.dut.mp_remote_close()
if do_ping:
ping_tentacle_infra(tentacle=tentacle, tag="a")
tentacle.power.dut = False
if do_ping:
ping_tentacle_infra(tentacle=tentacle, tag="b")
tentacle.power.led_error = False
if do_ping:
ping_tentacle_infra(tentacle=tentacle, tag="c")
# Free mp_remote
if not tentacle.is_mcu:
continue
if do_ping:
ping_tentacle_infra(tentacle=tentacle, tag="d")
# Before we can switch the relays: Connect to infra, power off and free mp_remote
tentacle.infra.load_base_code_if_needed()
ping_tentacle_infra(tentacle=tentacle, tag="e")
try:
tentacle.infra.switches.relays(
relays_open=tentacle.infra.list_all_relays
)
except Exception as e:
raise OctoprobeAppExitException(
f"{tentacle.infra.label}: Failed to control relays: {e!r}"
) from e
ping_tentacle_infra(tentacle=tentacle, tag="f")
tentacle.infra.mp_remote_close()
except Exception as e:
logger.error(e)
def setup_relays(
self, tentacles: typing.Sequence[TentacleBase], futs: tuple[enum.StrEnum]
) -> None:
assert isinstance(tentacles, list | tuple)
assert isinstance(futs, list | tuple)
assert len(futs) == 1
fut = futs[0]
for tentacle in tentacles:
tentacle.set_relays_by_FUT(fut=fut)