1
0
mirror of https://github.com/trezor/trezor-firmware.git synced 2024-12-05 05:58:23 +00:00
trezor-firmware/core/tests/test_trezor.crypto.hashlib.sha1.py

57 lines
1.7 KiB
Python
Raw Normal View History

from common import * # isort:skip
from trezor.crypto import hashlib
2017-12-16 01:11:26 +00:00
class TestCryptoSha1(unittest.TestCase):
# vectors from https://www.di-mgt.com.au/sha_testvectors.html
vectors = [
2023-06-28 10:46:29 +00:00
(b"", "da39a3ee5e6b4b0d3255bfef95601890afd80709"),
(b"abc", "a9993e364706816aba3e25717850c26c9cd0d89d"),
(
b"abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq",
"84983e441c3bd26ebaae4aa1f95129e5e54670f1",
),
(
b"abcdefghbcdefghicdefghijdefghijkefghijklfghijklmghijklmnhijklmnoijklmnopjklmnopqklmnopqrlmnopqrsmnopqrstnopqrstu",
"a49b2446a02c645bf419f995b67091253a04a259",
),
]
def test_digest(self):
for b, d in self.vectors:
self.assertEqual(hashlib.sha1(b).digest(), unhexlify(d))
def test_update(self):
for b, d in self.vectors:
x = hashlib.sha1()
x.update(b)
self.assertEqual(x.digest(), unhexlify(d))
x = hashlib.sha1()
for i in range(1000000):
2023-06-28 10:46:29 +00:00
x.update(b"a")
self.assertEqual(
x.digest(), unhexlify("34aa973cd4c4daa4f61eeb2bdbad27316534016f")
)
2023-06-28 10:46:29 +00:00
"""
x = hashlib.sha1()
for i in range(16777216):
x.update(b'abcdefghbcdefghicdefghijdefghijkefghijklfghijklmghijklmnhijklmno')
self.assertEqual(x.digest(), unhexlify('7789f0c9ef7bfc40d93311143dfbe69e2017f592'))
2023-06-28 10:46:29 +00:00
"""
def test_digest_multi(self):
x = hashlib.sha1()
d0 = x.digest()
d1 = x.digest()
d2 = x.digest()
self.assertEqual(d0, d1)
self.assertEqual(d0, d2)
2017-12-16 01:11:26 +00:00
2023-06-28 10:46:29 +00:00
if __name__ == "__main__":
unittest.main()