test suite

tests/test_scenegraph.py

@@ -0,0 +1,157 @@
+"""Tests for SceneGraph node selection and geometry/material tools."""
+
+import pytest
+import time
+
+
+class TestSceneGraph:
+    """Verify selectScenegraphNode, getNodeVertices, getNodeAdjacency, etc."""
+
+    @pytest.mark.probe
+    def test_tools_listed(self, bridge):
+        tools = bridge.list_tools()
+        names = [t["name"] for t in tools]
+        for expected in [
+            "selectScenegraphNode", "getNodeVertices", "getNodeAdjacency",
+            "getMaterialShaders", "getShaderSource", "getMaterialProperties",
+            "setRenderMode", "setSlowMode",
+        ]:
+            assert expected in names, f"Missing tool: {expected}"
+
+    @pytest.mark.probe
+    def test_select_scenegraph_node(self, connected_bridge):
+        connected_bridge.call_tool("selectQuickWindow", {"index": 0})
+        time.sleep(1)
+        nodes = connected_bridge.call_tool("listScenegraphNodes")
+
+        # Find first Geometry Node
+        def find_geo(items_list):
+            for item in items_list:
+                if item.get("type") == "Geometry Node":
+                    return item.get("address", "")
+                found = find_geo(item.get("children", []))
+                if found:
+                    return found
+            return None
+
+        addr = find_geo(nodes)
+        if not addr:
+            pytest.skip("No Geometry Node found")
+
+        result = connected_bridge.call_tool("selectScenegraphNode", {"address": addr})
+        assert isinstance(result, dict), f"selectScenegraphNode failed: {result}"
+        assert result.get("type") == "Geometry Node"
+        assert "hasGeometry" in result
+        assert "hasMaterial" in result
+
+    @pytest.mark.probe
+    def test_get_node_vertices(self, connected_bridge):
+        connected_bridge.call_tool("selectQuickWindow", {"index": 0})
+        time.sleep(1)
+        nodes = connected_bridge.call_tool("listScenegraphNodes")
+
+        def find_geo(items_list):
+            for item in items_list:
+                if item.get("type") == "Geometry Node":
+                    return item.get("address", "")
+                found = find_geo(item.get("children", []))
+                if found:
+                    return found
+            return None
+
+        addr = find_geo(nodes)
+        if not addr:
+            pytest.skip("No Geometry Node found")
+
+        result = connected_bridge.call_tool("getNodeVertices", {"address": addr})
+        assert isinstance(result, dict)
+        # May return "no vertex data" for software renderer -- that's OK
+        assert "error" in result or "vertexCount" in result
+
+    @pytest.mark.probe
+    def test_get_node_adjacency(self, connected_bridge):
+        connected_bridge.call_tool("selectQuickWindow", {"index": 0})
+        time.sleep(1)
+        nodes = connected_bridge.call_tool("listScenegraphNodes")
+
+        def find_geo(items_list):
+            for item in items_list:
+                if item.get("type") == "Geometry Node":
+                    return item.get("address", "")
+                found = find_geo(item.get("children", []))
+                if found:
+                    return found
+            return None
+
+        addr = find_geo(nodes)
+        if not addr:
+            pytest.skip("No Geometry Node found")
+
+        result = connected_bridge.call_tool("getNodeAdjacency", {"address": addr})
+        assert isinstance(result, dict)
+        assert "error" in result or "drawingMode" in result
+
+    @pytest.mark.probe
+    def test_get_material_shaders(self, connected_bridge):
+        connected_bridge.call_tool("selectQuickWindow", {"index": 0})
+        time.sleep(1)
+        nodes = connected_bridge.call_tool("listScenegraphNodes")
+
+        def find_geo(items_list):
+            for item in items_list:
+                if item.get("type") == "Geometry Node":
+                    return item.get("address", "")
+                found = find_geo(item.get("children", []))
+                if found:
+                    return found
+            return None
+
+        addr = find_geo(nodes)
+        if not addr:
+            pytest.skip("No Geometry Node found")
+
+        result = connected_bridge.call_tool("getMaterialShaders", {"address": addr})
+        assert isinstance(result, dict)
+        # Software renderer -> no shaders; that's expected
+        assert "error" in result or isinstance(result, list)
+
+    @pytest.mark.probe
+    def test_get_material_properties(self, connected_bridge):
+        connected_bridge.call_tool("selectQuickWindow", {"index": 0})
+        time.sleep(1)
+        nodes = connected_bridge.call_tool("listScenegraphNodes")
+
+        def find_geo(items_list):
+            for item in items_list:
+                if item.get("type") == "Geometry Node":
+                    return item.get("address", "")
+                found = find_geo(item.get("children", []))
+                if found:
+                    return found
+            return None
+
+        addr = find_geo(nodes)
+        if not addr:
+            pytest.skip("No Geometry Node found")
+
+        result = connected_bridge.call_tool("getMaterialProperties", {"address": addr})
+        assert isinstance(result, dict)
+        assert "error" in result or "properties" in result
+
+    @pytest.mark.probe
+    def test_set_render_mode(self, connected_bridge):
+        result = connected_bridge.call_tool("setRenderMode",
+                                            {"mode": "VisualizeOverdraw"})
+        assert isinstance(result, dict)
+        assert result.get("renderMode") == "VisualizeOverdraw"
+
+        # Reset
+        connected_bridge.call_tool("setRenderMode", {"mode": "NormalRendering"})
+
+    @pytest.mark.probe
+    def test_set_slow_mode(self, connected_bridge):
+        result = connected_bridge.call_tool("setSlowMode", {"enabled": True})
+        assert isinstance(result, dict)
+        assert result.get("slowMode") is True
+
+        connected_bridge.call_tool("setSlowMode", {"enabled": False})