1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
import subprocess
import sys
import tempfile
import unittest
from pathlib import Path
SCRIPT = Path(__file__).with_name("digest_compare.py")
def run(a: str, b: str) -> subprocess.CompletedProcess:
with tempfile.TemporaryDirectory() as d:
pa, pb = Path(d, "a.log"), Path(d, "b.log")
pa.write_text(a)
pb.write_text(b)
return subprocess.run([sys.executable, str(SCRIPT), str(pa), str(pb)], capture_output=True, text=True)
class DigestCompareTests(unittest.TestCase):
def test_equal(self):
r = run("x\ndigest 1 0 00000000000000aa\n", "y\ndigest 1 0 00000000000000aa\n")
self.assertEqual(0, r.returncode)
self.assertIn("1 digests equal", r.stdout)
def test_first_difference(self):
r = run("digest 1 0 00000000000000aa\ndigest 2 0 00000000000000bb\n",
"digest 1 0 00000000000000aa\ndigest 2 0 00000000000000cc\n")
self.assertEqual(1, r.returncode)
self.assertIn("digest 2 differs", r.stdout)
def test_count_mismatch(self):
r = run("digest 1 0 00000000000000aa\n", "digest 1 0 00000000000000aa\ndigest 2 0 00000000000000bb\n")
self.assertEqual(1, r.returncode)
self.assertIn("1 vs 2 digests", r.stdout)
def test_none(self):
r = run("x\n", "y\n")
self.assertEqual(1, r.returncode)
self.assertIn("no digest lines", r.stdout)
if __name__ == "__main__":
unittest.main()
|
