#!/usr/bin/env python3
"""Focused regression tests for production code-shape classification."""
import importlib.util
import unittest
from pathlib import Path


SCRIPT = Path(__file__).with_name("check-code-shape.py")
SPEC = importlib.util.spec_from_file_location("check_code_shape", SCRIPT)
MODULE = importlib.util.module_from_spec(SPEC)
assert SPEC.loader is not None
SPEC.loader.exec_module(MODULE)


class CodeShapeClassificationTest(unittest.TestCase):
    def test_internal_testdata_is_excluded(self) -> None:
        fixture = MODULE.REPO / "internal/service/kb/testdata/fixture.cjs"
        self.assertTrue(MODULE.is_excluded(fixture))

    def test_internal_cjs_outside_testdata_remains_production(self) -> None:
        source = MODULE.REPO / "internal/service/kb/runtime.cjs"
        self.assertFalse(MODULE.is_excluded(source))


if __name__ == "__main__":
    unittest.main()