test/stubgen: add tests for schema transpiling

tests/test_utils.py

@@ -0,0 +1,61 @@
+"""Tests for capnp_stubgen.utils."""
+
+import pytest
+from capnp_stubgen.utils import (
+    filename_to_module_name,
+    get_display_name,
+    get_node_kind,
+    type_id_to_int,
+)
+
+
+class TestFilenameToModuleName:
+    def test_simple(self):
+        assert filename_to_module_name("foo.capnp") == "foo_capnp"
+
+    def test_path(self):
+        assert filename_to_module_name("path/to/bar.capnp") == "bar_capnp"
+
+    def test_no_extension(self):
+        assert filename_to_module_name("foo") == "foo_capnp"
+
+
+class TestGetDisplayName:
+    def test_extracts_name(self):
+        node = {
+            "displayName": "path/to/file.capnp:MyType",
+            "displayNamePrefixLength": 19,  # "path/to/file.capnp:" = 19 chars
+        }
+        assert get_display_name(node) == "MyType"
+
+    def test_nested_type(self):
+        node = {
+            "displayName": "path/to/file.capnp:OuterType.InnerType",
+            "displayNamePrefixLength": 19,  # "path/to/file.capnp:" = 19 chars
+        }
+        assert get_display_name(node) == "OuterType.InnerType"
+
+
+class TestGetNodeKind:
+    def test_struct(self):
+        assert get_node_kind({"struct": {}, "id": "1"}) == "struct"
+
+    def test_enum(self):
+        assert get_node_kind({"enum": {}, "id": "1"}) == "enum"
+
+    def test_file(self):
+        assert get_node_kind({"file": {}, "id": "1"}) == "file"
+
+    def test_interface(self):
+        assert get_node_kind({"interface": {}, "id": "1"}) == "interface"
+
+    def test_unknown(self):
+        assert get_node_kind({"id": "1"}) == "unknown"
+
+
+class TestTypeIdToInt:
+    def test_converts(self):
+        assert type_id_to_int("12345") == 12345
+
+    def test_large_id(self):
+        assert type_id_to_int("15491878201996422075") == 15491878201996422075