import unittest
from pathlib import Path


PROJECT_ROOT = Path(__file__).resolve().parents[1]
SOURCE_ROOT = PROJECT_ROOT / "src" / "anyimage"


class NodeAssetTest(unittest.TestCase):
    def test_production_code_does_not_create_node_groups(self):
        offenders = []
        for path in SOURCE_ROOT.rglob("*.py"):
            if "bpy.data.node_groups.new" in path.read_text(encoding="utf-8"):
                offenders.append(path.relative_to(PROJECT_ROOT).as_posix())
        self.assertEqual(offenders, [])

    def test_runtime_node_group_entries_load_assets(self):
        material_source = (
            SOURCE_ROOT / "operators" / "common" / "mesh_result.py"
        ).read_text(encoding="utf-8")
        plane_source = (
            SOURCE_ROOT / "operators" / "convert_to_plane" / "nodes.py"
        ).read_text(encoding="utf-8")
        self.assertIn("return load_asset_node_group(group_name)", material_source)
        self.assertIn(
            "return load_asset_node_group(DEPTH_PLANE_NODE_GROUP_NAME)",
            plane_source,
        )


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