From 98952cf4efcb9b160e2d6e73cb5e0e670ced373f Mon Sep 17 00:00:00 2001
From: M1nd3r <petrsedlacek.km@seznam.cz>
Date: Fri, 24 Jan 2025 13:03:48 +0100
Subject: [PATCH] test: skip thp device test on protocol_v1 emu [no changelog]

---
 tests/device_tests/test_session.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/tests/device_tests/test_session.py b/tests/device_tests/test_session.py
index 56b8ace996..36bca2d46b 100644
--- a/tests/device_tests/test_session.py
+++ b/tests/device_tests/test_session.py
@@ -32,11 +32,9 @@ XPUB = "xpub6BiVtCpG9fQPxnPmHXG8PhtzQdWC2Su4qWu6XW9tpWFYhxydCLJGrWBJZ5H6qTAHdPQ7
 PIN4 = "1234"
 
 
+@pytest.mark.protocol("protocol_v2")
 def test_thp_end_session(client: Client):
     session = Session(client.get_session())
-    if session.protocol_version == ProtocolVersion.PROTOCOL_V1:
-        # TODO: This test should be skipped on non-THP builds
-        return
 
     msg = session.call(messages.EndSession())
     assert isinstance(msg, messages.Success)