1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2025-08-04 21:05:29 +00:00

fixup! test: update device tests

This commit is contained in:
M1nd3r 2025-03-24 15:26:42 +01:00
parent 9af0a05de8
commit 03e0eb552d

View File

@ -15,27 +15,10 @@
# If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>. # If not, see <https://www.gnu.org/licenses/lgpl-3.0.html>.
from trezorlib import device, messages, models from trezorlib import device, messages, models
from trezorlib.client import ProtocolVersion
from trezorlib.debuglink import SessionDebugWrapper as Session from trezorlib.debuglink import SessionDebugWrapper as Session
from trezorlib.debuglink import TrezorClientDebugLink as Client from trezorlib.debuglink import TrezorClientDebugLink as Client
def test_features(client: Client):
session = client.get_session()
f0 = session.features
if client.protocol_version == ProtocolVersion.V1:
# session erases session_id from its features
f0.session_id = session.id
f1 = session.call(messages.Initialize(session_id=session.id))
assert f0 == f1
else:
session2 = client.resume_session(session)
f1: messages.Features = session2.call(messages.GetFeatures())
assert f1.session_id is None
assert f0 == f1
def test_capabilities(session: Session): def test_capabilities(session: Session):
assert (messages.Capability.Translations in session.features.capabilities) == ( assert (messages.Capability.Translations in session.features.capabilities) == (
session.model is not models.T1B1 session.model is not models.T1B1