Gentoo Archives: gentoo-portage-dev

From: "Michał Górny" <mgorny@g.o>
To: gentoo-portage-dev@l.g.o
Cc: "Michał Górny" <mgorny@g.o>
Subject: [gentoo-portage-dev] [PATCH 05/14] tests: Add minimal safety checks for checksums
Date: Sun, 12 Mar 2017 19:03:02
Message-Id: 20170312190011.3234-6-mgorny@gentoo.org
In Reply to: [gentoo-portage-dev] [PATCHES] portage.checksum hacking, pt. 4 by "Michał Górny"
1 ---
2 pym/portage/tests/util/test_checksum.py | 88 +++++++++++++++++++++++++++++++++
3 1 file changed, 88 insertions(+)
4 create mode 100644 pym/portage/tests/util/test_checksum.py
5
6 diff --git a/pym/portage/tests/util/test_checksum.py b/pym/portage/tests/util/test_checksum.py
7 new file mode 100644
8 index 000000000..72b8cef3f
9 --- /dev/null
10 +++ b/pym/portage/tests/util/test_checksum.py
11 @@ -0,0 +1,88 @@
12 +# Copyright 2011-2017 Gentoo Foundation
13 +# Distributed under the terms of the GNU General Public License v2
14 +
15 +from portage.tests import TestCase
16 +
17 +from portage.checksum import checksum_str
18 +from portage.exception import DigestException
19 +
20 +class ChecksumTestCase(TestCase):
21 + text = b'Some test string used to check if the hash works'
22 +
23 + def test_md5(self):
24 + self.assertEqual(checksum_str(b'', 'MD5'),
25 + 'd41d8cd98f00b204e9800998ecf8427e')
26 + self.assertEqual(checksum_str(self.text, 'MD5'),
27 + '094c3bf4732f59b39d577e9726f1e934')
28 +
29 + def test_sha1(self):
30 + self.assertEqual(checksum_str(b'', 'SHA1'),
31 + 'da39a3ee5e6b4b0d3255bfef95601890afd80709')
32 + self.assertEqual(checksum_str(self.text, 'SHA1'),
33 + '5c572017d4e4d49e4aa03a2eda12dbb54a1e2e4f')
34 +
35 + def test_sha256(self):
36 + self.assertEqual(checksum_str(b'', 'SHA256'),
37 + 'e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855')
38 + self.assertEqual(checksum_str(self.text, 'SHA256'),
39 + 'e3d4a1135181fe156d61455615bb6296198e8ca5b2f20ddeb85cb4cd27f62320')
40 +
41 + def test_sha512(self):
42 + self.assertEqual(checksum_str(b'', 'SHA512'),
43 + 'cf83e1357eefb8bdf1542850d66d8007d620e4050b5715dc83f4a921d36ce9ce47d0d13c5d85f2b0ff8318d2877eec2f63b931bd47417a81a538327af927da3e')
44 + self.assertEqual(checksum_str(self.text, 'SHA512'),
45 + 'c8eaa902d48a2c82c2185a92f1c8bab8115c63c8d7a9966a8e8e81b07abcb9762f4707a6b27075e9d720277ba9fec072a59840d6355dd2ee64681d8f39a50856')
46 +
47 + def test_rmd160(self):
48 + try:
49 + self.assertEqual(checksum_str(b'', 'RMD160'),
50 + '9c1185a5c5e9fc54612808977ee8f548b2258d31')
51 + self.assertEqual(checksum_str(self.text, 'RMD160'),
52 + 'fc453174f63fc011d6f64abd2c45fb6a53c8239b')
53 + except DigestException:
54 + self.skipTest('RMD160 implementation not available')
55 +
56 + def test_whirlpool(self):
57 + try:
58 + self.assertEqual(checksum_str(b'', 'WHIRLPOOL'),
59 + '19fa61d75522a4669b44e39c1d2e1726c530232130d407f89afee0964997f7a73e83be698b288febcf88e3e03c4f0757ea8964e59b63d93708b138cc42a66eb3')
60 + self.assertEqual(checksum_str(self.text, 'WHIRLPOOL'),
61 + '8f556a079b87057f19e0880eed6d833e40c916f4b133196f6842281a2517873074d399832470c11ee251696b4844a10197714a069ba3e3415c8a4eced8f91b48')
62 + except DigestException:
63 + self.skipTest('WHIRLPOOL implementation not available')
64 +
65 + def test_blake2b(self):
66 + try:
67 + self.assertEqual(checksum_str(b'', 'BLAKE2B'),
68 + '786a02f742015903c6c6fd852552d272912f4740e15847618a86e217f71f5419d25e1031afee585313896444934eb04b903a685b1448b755d56f701afe9be2ce')
69 + self.assertEqual(checksum_str(self.text, 'BLAKE2B'),
70 + '84cb3c88838c7147bc9797c6525f812adcdcb40137f9c075963e3a3ed1fe06aaeeb4d2bb5589bad286864dc1aa834cfc4d66b8d7e4d4a246d91d45ce3a6eee43')
71 + except DigestException:
72 + self.skipTest('BLAKE2B implementation not available')
73 +
74 + def test_blake2s(self):
75 + try:
76 + self.assertEqual(checksum_str(b'', 'BLAKE2S'),
77 + '69217a3079908094e11121d042354a7c1f55b6482ca1a51e1b250dfd1ed0eef9')
78 + self.assertEqual(checksum_str(self.text, 'BLAKE2S'),
79 + '823ab2429f27690450efe888b0404d092fe2ee72a9bd63d5342c251b4dbb373d')
80 + except DigestException:
81 + self.skipTest('BLAKE2S implementation not available')
82 +
83 + def test_sha3_256(self):
84 + try:
85 + self.assertEqual(checksum_str(b'', 'SHA3_256'),
86 + 'a7ffc6f8bf1ed76651c14756a061d662f580ff4de43b49fa82d80a4b80f8434a')
87 + self.assertEqual(checksum_str(self.text, 'SHA3_256'),
88 + '932fc0498ebb865406f9b6606280939283aa8a148562e39fd095a5d22bdec5c6')
89 + except DigestException:
90 + self.skipTest('SHA3_256 implementation not available')
91 +
92 + def test_sha3_512(self):
93 + try:
94 + self.assertEqual(checksum_str(b'', 'SHA3_512'),
95 + 'a69f73cca23a9ac5c8b567dc185a756e97c982164fe25859e0d1dcc1475c80a615b2123af1f5f94c11e3e9402c3ac558f500199d95b6d3e301758586281dcd26')
96 + self.assertEqual(checksum_str(self.text, 'SHA3_512'),
97 + '6634c004dc31822fa65c2f1e2e3bbf0cfa35085653cca1ca9ca42f8f3f13c908405e0b665918146181c9fc9a9d793fc05429d669c35a55517820dfaa071425ca')
98 + except DigestException:
99 + self.skipTest('SHA3_512 implementation not available')
100 --
101 2.12.0