From f59b3fdc911b4f747d212ec44523092b31bef99b Mon Sep 17 00:00:00 2001 From: DaneSlattery Date: Mon, 19 Jun 2023 12:21:38 +0200 Subject: [PATCH] adjust tests for async --- test/test_load.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/test/test_load.py b/test/test_load.py index 7d82aa6..0fde0c9 100644 --- a/test/test_load.py +++ b/test/test_load.py @@ -121,7 +121,7 @@ def test_bundled_import_hook(): import stream_capnp # noqa: F401 -def test_load_capnp(foo): +async def test_load_capnp(foo): # test dynamically loading loader = capnp.SchemaLoader() loader.load(foo.Baz.schema.get_proto()) @@ -133,7 +133,7 @@ def test_load_capnp(foo): assert schema.fields["qux"].proto.slot.type.which == "struct" class Wrapper(foo.Wrapper.Server): - def wrapped(self, object, **kwargs): + async def wrapped(self, object, **kwargs): assert isinstance(object, capnp.lib.capnp._DynamicObjectReader) baz_ = object.as_struct(schema) assert baz_.text == "test" @@ -145,4 +145,5 @@ def wrapped(self, object, **kwargs): baz_.qux.id = 2 wrapper = foo.Wrapper._new_client(Wrapper()) - wrapper.wrapped(baz_).wait() + remote = wrapper.wrapped(baz_) + await remote