tests: fix signature of device.recover in expect-to-fail tests

6 years ago · 8618f44272
parent 1d3fa77ab6
commit 8618f44272
2 changed files with 20 additions and 4 deletions
--- 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,
+            )
--- 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,
+            )