mirror of
https://github.com/trezor/trezor-firmware.git
synced 2025-03-12 22:26:08 +00:00
32 lines
826 B
Python
32 lines
826 B
Python
# flake8: noqa: F403,F405
|
|
from common import * # isort:skip
|
|
|
|
if utils.USE_TROPIC:
|
|
from trezor.crypto import tropic
|
|
|
|
|
|
@unittest.skipUnless(utils.USE_TROPIC, "tropic")
|
|
class TestCryptoTropic(unittest.TestCase):
|
|
def test_ping(self):
|
|
self.assertEqual(tropic.ping(""), "")
|
|
self.assertEqual(tropic.ping("HeLlO!"), "HeLlO!")
|
|
|
|
def test_get_certificate(self):
|
|
self.assertEqual(len(tropic.get_certificate()), 512)
|
|
|
|
def test_sign(self):
|
|
try:
|
|
tropic.sign(0, "ASD")
|
|
assert False
|
|
except ValueError as e:
|
|
self.assertIn("invalid length", str(e).lower())
|
|
|
|
tropic.key_generate(0)
|
|
|
|
# signing should work now that we have a key
|
|
self.assertEqual(len(tropic.sign(0, "a" * 32)), 64)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|