From 8618f44272cd1c508ad953977c23b0f34926f80d Mon Sep 17 00:00:00 2001 From: matejcik Date: Wed, 3 Oct 2018 14:19:53 +0200 Subject: [PATCH] tests: fix signature of device.recover in expect-to-fail tests --- .../tests/device_tests/test_msg_recoverydevice.py | 12 ++++++++++-- .../tests/device_tests/test_msg_recoverydevice_t2.py | 12 ++++++++++-- 2 files changed, 20 insertions(+), 4 deletions(-) diff --git a/trezorlib/tests/device_tests/test_msg_recoverydevice.py b/trezorlib/tests/device_tests/test_msg_recoverydevice.py index ecb7eae55..e1e7ff5ed 100644 --- a/trezorlib/tests/device_tests/test_msg_recoverydevice.py +++ b/trezorlib/tests/device_tests/test_msg_recoverydevice.py @@ -204,5 +204,13 @@ class TestMsgRecoverydevice(TrezorTest): def test_already_initialized(self): self.setup_mnemonic_nopin_nopassphrase() - with pytest.raises(Exception): - device.recover(self.client, 12, False, False, "label", "english") + with pytest.raises(RuntimeError): + device.recover( + self.client, + 12, + False, + False, + "label", + "english", + self.client.mnemonic_callback, + ) diff --git a/trezorlib/tests/device_tests/test_msg_recoverydevice_t2.py b/trezorlib/tests/device_tests/test_msg_recoverydevice_t2.py index 149388f13..b906fe43b 100644 --- a/trezorlib/tests/device_tests/test_msg_recoverydevice_t2.py +++ b/trezorlib/tests/device_tests/test_msg_recoverydevice_t2.py @@ -109,5 +109,13 @@ class TestMsgRecoverydeviceT2(TrezorTest): def test_already_initialized(self): self.setup_mnemonic_nopin_nopassphrase() - with pytest.raises(Exception): - device.recover(self.client, 12, False, False, "label", "english") + with pytest.raises(RuntimeError): + device.recover( + self.client, + 12, + False, + False, + "label", + "english", + self.client.mnemonic_callback, + )