1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-02 04:18:20 +00:00

fixup! fix(python): fix device tests for test_session [no changelog]

This commit is contained in:
M1nd3r 2024-11-25 13:50:02 +01:00
parent 8cd9b03468
commit f916aa747f

View File

@ -22,7 +22,6 @@ from trezorlib.debuglink import TrezorClientDebugLink as Client
from trezorlib.debuglink import SessionDebugWrapper as Session from trezorlib.debuglink import SessionDebugWrapper as Session
from trezorlib.exceptions import TrezorFailure from trezorlib.exceptions import TrezorFailure
from trezorlib.tools import parse_path from trezorlib.tools import parse_path
from trezorlib.transport.session import SessionV1
from ..common import get_test_address from ..common import get_test_address
@ -33,8 +32,8 @@ PIN4 = "1234"
def test_thp_end_session(client: Client): def test_thp_end_session(client: Client):
session = client.get_session() session = Session(client.get_session())
if isinstance(session, SessionV1): if session.session_version == Session.CODEC_V1:
# TODO: This test should be skipped on non-THP builds # TODO: This test should be skipped on non-THP builds
return return