Add a bit of infrastructure for testing transforms

TeamSpen210 · Dec 25, 2022 · 0e4a00d · 0e4a00d
1 parent a145912
commit 0e4a00d
Show file tree

Hide file tree

Showing 8 changed files with 65 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -40,9 +40,12 @@ Icon
 *.vmx
 *.log
 *.prt
-*.bsp
 *.cache
 
+# Ignore BSPs, but only in some folders.
+instances/**/*.bsp
+examples/**/*.bsp
+
 # Output of "compiled" FGDs
 build/
 build_tmp/

diff --git a/pytest.ini b/pytest.ini
@@ -0,0 +1,2 @@
+[pytest]
+trio_mode = true
diff --git a/src/hammeraddons/bsp_transform/__init__.py b/src/hammeraddons/bsp_transform/__init__.py
@@ -15,6 +15,7 @@
 __all__ = [
     'check_control_enabled',
     'Context', 'trans', 'run_transformations',
+    'TransFunc', 'TRANSFORMS',
 ]
 
 

diff --git a/test-requirements.txt b/test-requirements.txt
@@ -0,0 +1,4 @@
+pytest-trio>=0.8.0
+pytest>>7.0.0
+pytest-datadir>=1.3.1
+pytest-regressions>=2.2.0
diff --git a/tests/__init__.py b/tests/__init__.py
diff --git a/tests/test_transforms/__init__.py b/tests/test_transforms/__init__.py
@@ -0,0 +1,40 @@
+import importlib
+import sys
+from pathlib import Path
+from typing import Awaitable, Callable
+
+import pytest
+
+from hammeraddons.bsp_transform import Context, TransFunc, TRANSFORMS
+from srctools.bsp import BSP
+from srctools.filesys import VirtualFileSystem
+from srctools.game import Game
+from srctools.packlist import PackList
+
+
+@pytest.fixture
+def blank_ctx(shared_datadir: Path) -> Context:
+    """Build a blank context."""
+    bsp = BSP(shared_datadir / 'blank.bsp')
+    game = Game(shared_datadir)
+    fsys = VirtualFileSystem({})
+    return Context(
+        fsys,
+        bsp.ents,
+        PackList(fsys),
+        bsp,
+        game,
+    )
+
+
+def get_transform_func(module_name: str, transform: str) -> Callable[[Context], Awaitable[None]]:
+    """Import the builtin transforms, then fetch this context."""
+    folder = str(Path(__file__, '..', '..', '..', 'transforms').resolve())
+    print(f'Adding "{folder}" to path.')
+
+    sys.path.append(folder)
+    try:
+        importlib.import_module(module_name)
+        return TRANSFORMS[transform]
+    finally:
+        sys.path.remove(folder)
diff --git a/tests/test_transforms/data/blank.bsp b/tests/test_transforms/data/blank.bsp
diff --git a/tests/test_transforms/data/gameinfo.txt b/tests/test_transforms/data/gameinfo.txt
@@ -0,0 +1,14 @@
+"GameInfo"
+{
+	game 		"HALF-LIFE 2"
+	title 		"HALF-LIFE'"
+	type		singleplayer_only
+
+	FileSystem
+	{
+		SteamAppId				220
+		SearchPaths
+		{
+		}
+	}
+}