From 984e4e214a38fa5537359511ce55a518f13e69fc Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ioan=20Biz=C4=83u?= Date: Wed, 25 Sep 2024 18:45:38 +0200 Subject: [PATCH] feat(core/tests): add smoke test for libtropic [no changelog] --- core/tests/test_trezor.crypto.tropic.py | 31 +++++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 core/tests/test_trezor.crypto.tropic.py diff --git a/core/tests/test_trezor.crypto.tropic.py b/core/tests/test_trezor.crypto.tropic.py new file mode 100644 index 0000000000..8c70989ac6 --- /dev/null +++ b/core/tests/test_trezor.crypto.tropic.py @@ -0,0 +1,31 @@ +# 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()