mirror of
https://github.com/capnproto/pycapnp.git
synced 2025-03-04 08:24:43 +01:00
3aade70bfa
- Stop adding the directory of every .capnp file to the import path. If a .capnp file wants to import a file in its own directory, it should use a relative import. Fixes #278 - Stop using /usr/include/capnp as an import path. This is incorrect. It should only be /usr/include. - Stop allowing additional paths to be specified for magic imports. This leads to inconsistencies. More specifically, the way that a nested import like `ma.mb.mc_capnp` gets imported by python, is to first import `ma`, then import `ma.mb`, and finally `ma.mb.mc_capnp`. Pycapnp's magic importing is only involved in the last step. So any additional paths specified don't work for nested imports. It is very confusing to only have this for non-nested imports. Users with folder layouts that don't follow pythons import paths can still use `capnp.load(.., .., imports=[blah])`.
154 lines
3.7 KiB
Python
154 lines
3.7 KiB
Python
import pytest
|
|
import capnp
|
|
import os
|
|
import sys
|
|
|
|
this_dir = os.path.dirname(__file__)
|
|
|
|
|
|
@pytest.fixture
|
|
def addressbook():
|
|
return capnp.load(os.path.join(this_dir, "addressbook.capnp"))
|
|
|
|
|
|
@pytest.fixture
|
|
def foo():
|
|
return capnp.load(os.path.join(this_dir, "foo.capnp"))
|
|
|
|
|
|
@pytest.fixture
|
|
def bar():
|
|
return capnp.load(os.path.join(this_dir, "bar.capnp"))
|
|
|
|
|
|
def test_basic_load():
|
|
capnp.load(os.path.join(this_dir, "addressbook.capnp"))
|
|
|
|
|
|
def test_constants(addressbook):
|
|
assert addressbook.qux == 123
|
|
|
|
|
|
def test_classes(addressbook):
|
|
assert addressbook.AddressBook
|
|
assert addressbook.Person
|
|
|
|
|
|
def test_import(foo, bar):
|
|
m = capnp._MallocMessageBuilder()
|
|
foo = m.init_root(foo.Foo)
|
|
m2 = capnp._MallocMessageBuilder()
|
|
bar = m2.init_root(bar.Bar)
|
|
|
|
foo.name = "foo"
|
|
bar.foo = foo
|
|
|
|
assert bar.foo.name == "foo"
|
|
|
|
|
|
def test_failed_import():
|
|
s = capnp.SchemaParser()
|
|
s2 = capnp.SchemaParser()
|
|
|
|
foo = s.load(os.path.join(this_dir, "foo.capnp"))
|
|
bar = s2.load(os.path.join(this_dir, "bar.capnp"))
|
|
|
|
m = capnp._MallocMessageBuilder()
|
|
foo = m.init_root(foo.Foo)
|
|
m2 = capnp._MallocMessageBuilder()
|
|
bar = m2.init_root(bar.Bar)
|
|
|
|
foo.name = "foo"
|
|
|
|
with pytest.raises(Exception):
|
|
bar.foo = foo
|
|
|
|
|
|
def test_defualt_import_hook():
|
|
# Make sure any previous imports of addressbook_capnp are gone
|
|
capnp.cleanup_global_schema_parser()
|
|
|
|
import addressbook_capnp # noqa: F401
|
|
|
|
|
|
def test_dash_import():
|
|
import addressbook_with_dashes_capnp # noqa: F401
|
|
|
|
|
|
def test_spaces_import():
|
|
import addressbook_with_spaces_capnp # noqa: F401
|
|
|
|
|
|
def test_add_import_hook():
|
|
capnp.add_import_hook()
|
|
|
|
# Make sure any previous imports of addressbook_capnp are gone
|
|
capnp.cleanup_global_schema_parser()
|
|
|
|
import addressbook_capnp
|
|
|
|
addressbook_capnp.AddressBook.new_message()
|
|
|
|
|
|
def test_multiple_add_import_hook():
|
|
capnp.add_import_hook()
|
|
capnp.add_import_hook()
|
|
|
|
# Make sure any previous imports of addressbook_capnp are gone
|
|
capnp.cleanup_global_schema_parser()
|
|
|
|
import addressbook_capnp
|
|
|
|
addressbook_capnp.AddressBook.new_message()
|
|
|
|
|
|
def test_remove_import_hook():
|
|
capnp.add_import_hook()
|
|
capnp.remove_import_hook()
|
|
|
|
if "addressbook_capnp" in sys.modules:
|
|
# hack to deal with it being imported already
|
|
del sys.modules["addressbook_capnp"]
|
|
|
|
with pytest.raises(ImportError):
|
|
import addressbook_capnp # noqa: F401
|
|
|
|
|
|
def test_bundled_import_hook():
|
|
# stream.capnp should be bundled, or provided by the system capnproto
|
|
capnp.add_import_hook()
|
|
from capnp import stream_capnp # noqa: F401
|
|
|
|
|
|
def test_nested_import():
|
|
import schemas.parent_capnp # noqa: F401
|
|
import schemas.child_capnp # noqa: F401
|
|
|
|
|
|
async def test_load_capnp(foo):
|
|
# test dynamically loading
|
|
loader = capnp.SchemaLoader()
|
|
loader.load(foo.Baz.schema.get_proto())
|
|
loader.load_dynamic(foo.Qux.schema.get_proto().node)
|
|
|
|
schema = loader.get(foo.Baz.schema.get_proto().node.id).as_struct()
|
|
assert "text" in schema.fieldnames
|
|
assert "qux" in schema.fieldnames
|
|
assert schema.fields["qux"].proto.slot.type.which == "struct"
|
|
|
|
class Wrapper(foo.Wrapper.Server):
|
|
async def wrapped(self, object, **kwargs):
|
|
assert isinstance(object, capnp.lib.capnp._DynamicObjectReader)
|
|
baz_ = object.as_struct(schema)
|
|
assert baz_.text == "test"
|
|
assert baz_.qux.id == 2
|
|
|
|
# test calling into the wrapper with a Baz message.
|
|
baz_ = foo.Baz.new_message()
|
|
baz_.text = "test"
|
|
baz_.qux.id = 2
|
|
|
|
async with capnp.kj_loop():
|
|
wrapper = foo.Wrapper._new_client(Wrapper())
|
|
remote = wrapper.wrapped(baz_)
|
|
await remote
|