2019-10-21 14:49:34 +00:00
|
|
|
import os
|
|
|
|
|
|
|
|
|
|
import pytest
|
|
|
|
|
|
|
|
|
|
from ..emulators import EmulatorWrapper
|
|
|
|
|
|
|
|
|
|
SELECTED_GENS = [
|
|
|
|
|
gen.strip() for gen in os.environ.get("TREZOR_UPGRADE_TEST", "").split(",") if gen
|
|
|
|
|
]
|
|
|
|
|
|
|
|
|
|
if SELECTED_GENS:
|
|
|
|
|
# if any gens were selected via the environment variable, force enable all selected
|
|
|
|
|
LEGACY_ENABLED = "legacy" in SELECTED_GENS
|
|
|
|
|
CORE_ENABLED = "core" in SELECTED_GENS
|
|
|
|
|
|
|
|
|
|
else:
|
|
|
|
|
# if no selection was provided, select those for which we have emulators
|
|
|
|
|
try:
|
|
|
|
|
EmulatorWrapper("legacy")
|
|
|
|
|
LEGACY_ENABLED = True
|
|
|
|
|
except Exception:
|
|
|
|
|
LEGACY_ENABLED = False
|
|
|
|
|
|
|
|
|
|
try:
|
|
|
|
|
EmulatorWrapper("core")
|
|
|
|
|
CORE_ENABLED = True
|
|
|
|
|
except Exception:
|
|
|
|
|
CORE_ENABLED = False
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
legacy_only = pytest.mark.skipif(
|
|
|
|
|
not LEGACY_ENABLED, reason="This test requires legacy emulator"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
core_only = pytest.mark.skipif(
|
|
|
|
|
not CORE_ENABLED, reason="This test requires core emulator"
|
|
|
|
|
)
|
