From f916aa747f126bf7258c3ec1e723fb31df484979 Mon Sep 17 00:00:00 2001 From: M1nd3r Date: Mon, 25 Nov 2024 13:50:02 +0100 Subject: [PATCH] fixup! fix(python): fix device tests for test_session [no changelog] --- tests/device_tests/test_session.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tests/device_tests/test_session.py b/tests/device_tests/test_session.py index b602e07084..c88f85167c 100644 --- a/tests/device_tests/test_session.py +++ b/tests/device_tests/test_session.py @@ -22,7 +22,6 @@ from trezorlib.debuglink import TrezorClientDebugLink as Client from trezorlib.debuglink import SessionDebugWrapper as Session from trezorlib.exceptions import TrezorFailure from trezorlib.tools import parse_path -from trezorlib.transport.session import SessionV1 from ..common import get_test_address @@ -33,8 +32,8 @@ PIN4 = "1234" def test_thp_end_session(client: Client): - session = client.get_session() - if isinstance(session, SessionV1): + session = Session(client.get_session()) + if session.session_version == Session.CODEC_V1: # TODO: This test should be skipped on non-THP builds return