mirror of
https://github.com/capnproto/pycapnp.git
synced 2025-03-04 16:35:04 +01:00
db63fd1880
Any module with underscores in it will now attempt to look for 3 files, first the original, then with dashes, then with spaces, ie: import addressbook_v_2_capnp will search for 'addressbook_v_2.capnp', 'addressbook-v-2.capnp', and 'addressbook v 2.capnp'.
89 lines
2.1 KiB
Python
89 lines
2.1 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(ValueError):
|
|
bar.foo = foo
|
|
|
|
def test_defualt_import_hook():
|
|
import addressbook_capnp
|
|
|
|
def test_dash_import():
|
|
import addressbook_with_dashes_capnp
|
|
|
|
def test_spaces_import():
|
|
import addressbook_with_spaces_capnp
|
|
|
|
def test_add_import_hook():
|
|
capnp.add_import_hook([this_dir])
|
|
|
|
import addressbook_capnp
|
|
addressbook_capnp.AddressBook.new_message()
|
|
|
|
def test_multiple_add_import_hook():
|
|
capnp.add_import_hook()
|
|
capnp.add_import_hook()
|
|
capnp.add_import_hook([this_dir])
|
|
|
|
import addressbook_capnp
|
|
addressbook_capnp.AddressBook.new_message()
|
|
|
|
def test_remove_import_hook():
|
|
capnp.add_import_hook([this_dir])
|
|
capnp.remove_import_hook()
|
|
|
|
if 'addressbook_capnp' in sys.modules:
|
|
del sys.modules['addressbook_capnp'] # hack to deal with it being imported already
|
|
|
|
with pytest.raises(ImportError):
|
|
import addressbook_capnp
|