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()