diff --git a/tests/test_protect_call.py b/tests/test_protect_call.py index 1efc0f9ab..12066b433 100644 --- a/tests/test_protect_call.py +++ b/tests/test_protect_call.py @@ -1,7 +1,7 @@ import unittest import common -from bitkeylib.client import CallException, PinException, OtpException +from bitkeylib.client import CallException, PinException from bitkeylib import proto class TestProtectCall(common.BitkeyTest): @@ -12,52 +12,24 @@ class TestProtectCall(common.BitkeyTest): self.assertEqual(len(entropy), entropy_len) def test_no_protection(self): - self.bitkey.load_device(algo=proto.ELECTRUM, + self.bitkey.load_device( seed='beyond neighbor scratch swirl embarrass doll cause also stick softly physical nice', - otp=False, pin='', spv=False) + pin='') - self.assertEqual(self.bitkey.features.has_otp, False) - self.assertEqual(self.bitkey.features.pin, False) + self.assertEqual(self.bitkey.debuglink.read_pin()[0], '') self._some_protected_call() - def test_otp_only(self): - self.bitkey.load_device(algo=proto.ELECTRUM, + def test_pin(self): + self.bitkey.load_device( seed='beyond neighbor scratch swirl embarrass doll cause also stick softly physical nice', - otp=True, pin='', spv=False) + pin='2345') - self.assertEqual(self.bitkey.features.has_otp, True) - self.assertEqual(self.bitkey.features.pin, False) + self.assertEqual(self.bitkey.debuglink.read_pin()[0], '2345') self._some_protected_call() - def test_pin_only(self): - self.bitkey.load_device(algo=proto.ELECTRUM, - seed='beyond neighbor scratch swirl embarrass doll cause also stick softly physical nice', - otp=False, pin='2345', spv=False) - - self.assertEqual(self.bitkey.features.has_otp, False) - self.assertEqual(self.bitkey.features.pin, True) - self._some_protected_call() - - def test_both(self): - self.bitkey.load_device(algo=proto.ELECTRUM, - seed='beyond neighbor scratch swirl embarrass doll cause also stick softly physical nice', - otp=True, pin='3456', spv=False) - - self.assertEqual(self.bitkey.features.has_otp, True) - self.assertEqual(self.bitkey.features.pin, True) - self._some_protected_call() - def test_incorrect_pin(self): - self.bitkey.setup_debuglink(button=True, pin_correct=False, otp_correct=True) + self.bitkey.setup_debuglink(button=True, pin_correct=False) self.assertRaises(PinException, self._some_protected_call) - - def test_incorrect_otp(self): - self.bitkey.setup_debuglink(button=True, pin_correct=True, otp_correct=False) - self.assertRaises(OtpException, self._some_protected_call) - - def test_incorrect_both(self): - self.bitkey.setup_debuglink(button=True, pin_correct=False, otp_correct=False) - self.assertRaises(CallException, self._some_protected_call) if __name__ == '__main__': unittest.main() \ No newline at end of file