test/stubgen: add for static checks
This commit is contained in:
@@ -0,0 +1,70 @@
|
||||
"""Verify generated .pyi stubs are consistent with pycapnp runtime.
|
||||
|
||||
Test file lives alongside the ``.capnp`` schemas + generated ``.pyi``
|
||||
stubs. pycapnp's import hook handles runtime loading.
|
||||
basedpyright type-checks this file together with the project.
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
# Ensure this directory is importable
|
||||
_HERE = Path(__file__).parent
|
||||
if str(_HERE) not in sys.path:
|
||||
sys.path.insert(0, str(_HERE))
|
||||
|
||||
import capnp # pyright: ignore[reportMissingTypeStubs, reportUnusedImport] # activates import hook
|
||||
import test_simple_capnp
|
||||
|
||||
|
||||
class TestSimpleStub:
|
||||
"""test_simple.capnp — basic struct."""
|
||||
|
||||
def test_new_message(self) -> None:
|
||||
p = test_simple_capnp.Person.new_message(
|
||||
name="Alice", age=30, email="alice@example.com"
|
||||
)
|
||||
assert p.name == "Alice"
|
||||
assert p.age == 30
|
||||
|
||||
def test_field_assignment(self) -> None:
|
||||
p = test_simple_capnp.Person.new_message()
|
||||
p.name = "Bob"
|
||||
p.age = 25
|
||||
assert p.name == "Bob"
|
||||
|
||||
def test_to_dict(self) -> None:
|
||||
p = test_simple_capnp.Person.new_message(name="Carol", age=28)
|
||||
d = p.to_dict()
|
||||
assert d["name"] == "Carol"
|
||||
|
||||
def test_from_dict(self) -> None:
|
||||
p = test_simple_capnp.Person.new_message()
|
||||
_ = p.from_dict({"name": "Dave", "age": 35})
|
||||
assert p.name == "Dave"
|
||||
|
||||
def test_serialize_roundtrip(self) -> None:
|
||||
p = test_simple_capnp.Person.new_message(name="Eve", age=22)
|
||||
data = p.to_bytes()
|
||||
p2 = test_simple_capnp.Person.from_bytes(data)
|
||||
assert p2 is not None
|
||||
|
||||
|
||||
class TestNestedStub:
|
||||
"""test_nested.capnp — nested types + lists."""
|
||||
|
||||
def test_nested_message(self) -> None:
|
||||
import test_nested_capnp
|
||||
ab = test_nested_capnp.AddressBook.new_message()
|
||||
assert ab.people is not None
|
||||
|
||||
|
||||
class TestGenericsStub:
|
||||
"""test_generics.capnp — generic structs."""
|
||||
|
||||
def test_holder_loads(self) -> None:
|
||||
import test_generics_capnp
|
||||
c = test_generics_capnp.Container.new_message()
|
||||
assert c is not None
|
||||
Reference in New Issue
Block a user