From ee05baf5081241a5c7b7b9d794b006fc8dd2e99f Mon Sep 17 00:00:00 2001
From: Colin Basnett <5035660+cmbasnett@users.noreply.github.com>
Date: Mon, 2 Mar 2026 01:17:21 -0800
Subject: [PATCH] Add some simple PSK export tests

---
 tests/data/psk_export_tests.blend |  3 ++
 tests/psk_export_test.py          | 62 +++++++++++++++++++++++++++++++
 2 files changed, 65 insertions(+)
 create mode 100644 tests/data/psk_export_tests.blend
 create mode 100644 tests/psk_export_test.py

diff --git a/tests/data/psk_export_tests.blend b/tests/data/psk_export_tests.blend
new file mode 100644
index 0000000..a1ca947
--- /dev/null
+++ b/tests/data/psk_export_tests.blend
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7fada8e55579162537186be2a0f04548dc4f0dc63fbc2fb38ac20b010bd0ba8d
+size 103797
diff --git a/tests/psk_export_test.py b/tests/psk_export_test.py
new file mode 100644
index 0000000..f9da8fc
--- /dev/null
+++ b/tests/psk_export_test.py
@@ -0,0 +1,62 @@
+import bpy
+import pytest
+import tempfile
+from psk_psa_py.psk.reader import read_psk_from_file
+
+
+@pytest.fixture(autouse=True)
+def run_before_and_after_tests(tmpdir):
+    # Setup: Run before the tests
+    bpy.ops.wm.read_homefile(filepath='tests/data/psk_export_tests.blend')
+
+    yield
+    # Teardown: Run after the tests
+    pass
+
+
+def export_psk_and_read_back(collection_name: str):
+    collection = bpy.data.collections.get(collection_name, None)
+    assert collection is not None, f"Collection {collection_name} not found in the scene."
+
+    # Select the collection to make it the active collection.
+    view_layer = bpy.context.view_layer
+    assert view_layer is not None, "No active view layer found."
+    view_layer.active_layer_collection = view_layer.layer_collection.children[collection_name]
+
+    filepath = str(tempfile.gettempdir() + f'/{collection_name}.psk')
+
+    collection.exporters[0].filepath = filepath
+
+    assert bpy.ops.collection.exporter_export() == {'FINISHED'}, "PSK export failed."
+
+    # Now load the exported PSK file and return its contents.
+    psk = read_psk_from_file(filepath)
+    return psk
+
+
+def test_psk_export_cube_no_bones():
+    psk = export_psk_and_read_back('cube_no_bones')
+
+    # There should be one bone when no armature is present, this is added automatically to serve as the root bone for the mesh.
+    assert len(psk.bones) == 1, f"Expected 1 bone, but found {len(psk.bones)}."
+    assert len(psk.points) == 8, f"Expected 8 points, but found {len(psk.points)}."
+    assert len(psk.faces) == 12, f"Expected 12 faces, but found {len(psk.faces)}."
+    assert len(psk.materials) == 1, f"Expected 1 material, but found {len(psk.materials)}."
+
+
+def test_cube_edge_split():
+    # The cube has all the edges set to split with a modifier.
+    psk = export_psk_and_read_back('cube_edge_split')
+
+    assert len(psk.bones) == 1, f"Expected 1 bone, but found {len(psk.bones)}."
+    assert len(psk.points) == 24, f"Expected 24 points, but found {len(psk.points)}."
+    assert len(psk.faces) == 12, f"Expected 12 faces, but found {len(psk.faces)}."
+    assert len(psk.materials) == 1, f"Expected 1 material, but found {len(psk.materials)}."
+
+
+def test_cube_with_simple_armature():
+    # The cube has all the edges set to split with a modifier.
+    psk = export_psk_and_read_back('cube_with_simple_armature')
+
+    assert len(psk.bones) == 1, f"Expected 1 bone, but found {len(psk.bones)}."
+    assert psk.bones[0].name == b'ROOT', f"Expected bone name 'ROOT', but found {psk.bones[0].name}."