From 46eb223b2a7192cd8d21a2077981f420ddac384f Mon Sep 17 00:00:00 2001 From: matejcik Date: Tue, 25 Aug 2020 16:28:32 +0200 Subject: [PATCH] tests: use new trezorlib locking/sessioning API where appropriate --- tests/device_tests/test_msg_changepin.py | 4 ++-- tests/device_tests/test_msg_changepin_t2.py | 4 ++-- tests/device_tests/test_passphrase_slip39_advanced.py | 4 ++-- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/device_tests/test_msg_changepin.py b/tests/device_tests/test_msg_changepin.py index 095a41fcd..7ad6c7daa 100644 --- a/tests/device_tests/test_msg_changepin.py +++ b/tests/device_tests/test_msg_changepin.py @@ -27,7 +27,7 @@ pytestmark = pytest.mark.skip_t2 def _check_pin(client, pin): - client.clear_session() + client.lock() with client: client.use_pin_sequence([pin]) client.set_expected_responses([messages.PinMatrixRequest(), messages.Address()]) @@ -35,7 +35,7 @@ def _check_pin(client, pin): def _check_no_pin(client): - client.clear_session() + client.lock() with client: client.set_expected_responses([messages.Address()]) client.call(messages.GetAddress()) diff --git a/tests/device_tests/test_msg_changepin_t2.py b/tests/device_tests/test_msg_changepin_t2.py index 61b3f665f..e830e9574 100644 --- a/tests/device_tests/test_msg_changepin_t2.py +++ b/tests/device_tests/test_msg_changepin_t2.py @@ -27,7 +27,7 @@ pytestmark = pytest.mark.skip_t1 def _check_pin(client, pin): - client.clear_session() + client.lock() assert client.features.pin_protection is True assert client.features.unlocked is False @@ -38,7 +38,7 @@ def _check_pin(client, pin): def _check_no_pin(client): - client.clear_session() + client.lock() assert client.features.pin_protection is False with client: diff --git a/tests/device_tests/test_passphrase_slip39_advanced.py b/tests/device_tests/test_passphrase_slip39_advanced.py index 47c40f6a9..d90a44997 100644 --- a/tests/device_tests/test_passphrase_slip39_advanced.py +++ b/tests/device_tests/test_passphrase_slip39_advanced.py @@ -35,7 +35,7 @@ def test_128bit_passphrase(client): client.use_passphrase("TREZOR") address = get_test_address(client) assert address == "mkKDUMRR1CcK8eLAzCZAjKnNbCquPoWPxN" - client.state = None + client.clear_session() client.use_passphrase("ROZERT") address_compare = get_test_address(client) @@ -54,7 +54,7 @@ def test_256bit_passphrase(client): client.use_passphrase("TREZOR") address = get_test_address(client) assert address == "mxVtGxUJ898WLzPMmy6PT1FDHD1GUCWGm7" - client.state = None + client.clear_session() client.use_passphrase("ROZERT") address_compare = get_test_address(client)