oak/benchmarks

from __future__ import annotations

import hashlib

import sys

import tempfile

import unittest

from pathlib import Path

sys.path.insert(0, str(Path(__file__).resolve().parents[1] / "scripts"))

from oakbench.subjects import binary_sha256

class SubjectFingerprintTest(unittest.TestCase):

    def test_binary_sha256_hashes_existing_file(self) -> None:

        with tempfile.TemporaryDirectory() as tmp:

            path = Path(tmp) / "bin"

            path.write_bytes(b"oak")

            self.assertEqual(binary_sha256(path), hashlib.sha256(b"oak").hexdigest())

    def test_binary_sha256_missing_is_none(self) -> None:

        self.assertIsNone(binary_sha256(Path("/definitely/not/here/oak")))

if __name__ == "__main__":

    unittest.main()