chore(tests): fix deprecation warnings

3 years ago · e5a481ded5
parent b1ca6ca848
commit e5a481ded5
1 changed files with 5 additions and 5 deletions
--- a/tests/device_tests/test_session_id_and_passphrase.py
+++ b/tests/device_tests/test_session_id_and_passphrase.py
@ -55,13 +55,13 @@ def _get_xpub(client, passphrase=None):
    """Get XPUB and check that the appropriate passphrase flow has happened."""
    if passphrase is not None:
        expected_responses = [
-            messages.PassphraseRequest(),
-            messages.ButtonRequest(),
-            messages.ButtonRequest(),
-            messages.PublicKey(),
+            messages.PassphraseRequest,
+            messages.ButtonRequest,
+            messages.ButtonRequest,
+            messages.PublicKey,
        ]
    else:
-        expected_responses = [messages.PublicKey()]
+        expected_responses = [messages.PublicKey]

    with client:
        client.use_passphrase(passphrase or "")