• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1"""Tests for the md5sum script in the Tools directory."""
2
3import sys
4import os
5import unittest
6from test.support import os_helper
7from test.support import hashlib_helper
8from test.support.script_helper import assert_python_ok, assert_python_failure
9
10from test.test_tools import scriptsdir, skip_if_missing
11
12skip_if_missing()
13
14@hashlib_helper.requires_hashdigest('md5', openssl=True)
15class MD5SumTests(unittest.TestCase):
16    @classmethod
17    def setUpClass(cls):
18        cls.script = os.path.join(scriptsdir, 'md5sum.py')
19        os.mkdir(os_helper.TESTFN_ASCII)
20        cls.fodder = os.path.join(os_helper.TESTFN_ASCII, 'md5sum.fodder')
21        with open(cls.fodder, 'wb') as f:
22            f.write(b'md5sum\r\ntest file\r\n')
23        cls.fodder_md5 = b'd38dae2eb1ab346a292ef6850f9e1a0d'
24        cls.fodder_textmode_md5 = b'a8b07894e2ca3f2a4c3094065fa6e0a5'
25
26    @classmethod
27    def tearDownClass(cls):
28        os_helper.rmtree(os_helper.TESTFN_ASCII)
29
30    def test_noargs(self):
31        rc, out, err = assert_python_ok(self.script)
32        self.assertEqual(rc, 0)
33        self.assertTrue(
34            out.startswith(b'd41d8cd98f00b204e9800998ecf8427e <stdin>'))
35        self.assertFalse(err)
36
37    def test_checksum_fodder(self):
38        rc, out, err = assert_python_ok(self.script, self.fodder)
39        self.assertEqual(rc, 0)
40        self.assertTrue(out.startswith(self.fodder_md5))
41        for part in self.fodder.split(os.path.sep):
42            self.assertIn(part.encode(), out)
43        self.assertFalse(err)
44
45    def test_dash_l(self):
46        rc, out, err = assert_python_ok(self.script, '-l', self.fodder)
47        self.assertEqual(rc, 0)
48        self.assertIn(self.fodder_md5, out)
49        parts = self.fodder.split(os.path.sep)
50        self.assertIn(parts[-1].encode(), out)
51        self.assertNotIn(parts[-2].encode(), out)
52
53    def test_dash_t(self):
54        rc, out, err = assert_python_ok(self.script, '-t', self.fodder)
55        self.assertEqual(rc, 0)
56        self.assertTrue(out.startswith(self.fodder_textmode_md5))
57        self.assertNotIn(self.fodder_md5, out)
58
59    def test_dash_s(self):
60        rc, out, err = assert_python_ok(self.script, '-s', '512', self.fodder)
61        self.assertEqual(rc, 0)
62        self.assertIn(self.fodder_md5, out)
63
64    def test_multiple_files(self):
65        rc, out, err = assert_python_ok(self.script, self.fodder, self.fodder)
66        self.assertEqual(rc, 0)
67        lines = out.splitlines()
68        self.assertEqual(len(lines), 2)
69        self.assertEqual(*lines)
70
71    def test_usage(self):
72        rc, out, err = assert_python_failure(self.script, '-h')
73        self.assertEqual(rc, 2)
74        self.assertEqual(out, b'')
75        self.assertGreater(err, b'')
76
77
78if __name__ == '__main__':
79    unittest.main()
80