From b70517d6d97578a6cd048cd2bcaf786f9b130b3e Mon Sep 17 00:00:00 2001
From: M1nd3r <petrsedlacek.km@seznam.cz>
Date: Tue, 4 Feb 2025 15:13:35 +0100
Subject: [PATCH] chore(tests): add protocol pytest marker [no changelog]

---
 tests/REGISTERED_MARKERS |  1 +
 tests/conftest.py        | 24 ++++++++++++++++++++++++
 2 files changed, 25 insertions(+)

diff --git a/tests/REGISTERED_MARKERS b/tests/REGISTERED_MARKERS
index fab4ec8b3a..bec85ca898 100644
--- a/tests/REGISTERED_MARKERS
+++ b/tests/REGISTERED_MARKERS
@@ -11,6 +11,7 @@ multisig
 nem
 ontology
 peercoin
+protocol
 ripple
 sd_card
 solana
diff --git a/tests/conftest.py b/tests/conftest.py
index 6cb5d1b6f2..bb462a11c7 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -29,6 +29,7 @@ from _pytest.python import IdMaker
 from _pytest.reports import TestReport
 
 from trezorlib import debuglink, log, models
+from trezorlib.client import ProtocolVersion
 from trezorlib.debuglink import SessionDebugWrapper
 from trezorlib.debuglink import TrezorClientDebugLink as Client
 from trezorlib.device import apply_settings
@@ -270,6 +271,29 @@ def _client_unlocked(
     if _raw_client.model not in models_filter:
         pytest.skip(f"Skipping test for model {_raw_client.model.internal_name}")
 
+    protocol_marker: Mark | None = request.node.get_closest_marker("protocol")
+    if protocol_marker:
+        args = protocol_marker.args
+        protocol_version = _raw_client.protocol_version
+
+        if (
+            protocol_version == ProtocolVersion.PROTOCOL_V1
+            and "protocol_v1" not in args
+        ):
+            pytest.skip(
+                f"Skipping test for device/emulator with protocol_v{protocol_version} - the protocol is not supported."
+            )
+
+        if (
+            protocol_version == ProtocolVersion.PROTOCOL_V2
+            and "protocol_v2" not in args
+        ):
+            pytest.skip(
+                f"Skipping test for device/emulator with protocol_v{protocol_version} - the protocol is not supported."
+            )
+
+    if _raw_client.protocol_version is ProtocolVersion.PROTOCOL_V2:
+        pass
     sd_marker = request.node.get_closest_marker("sd_card")
     if sd_marker and not _raw_client.features.sd_card_present:
         raise RuntimeError(