"""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