41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
import pytest
|
|
import capnp
|
|
import os
|
|
import socket
|
|
|
|
this_dir = os.path.dirname(__file__)
|
|
|
|
@pytest.fixture
|
|
def capability():
|
|
return capnp.load(os.path.join(this_dir, 'test_capability.capnp'))
|
|
|
|
class Server:
|
|
def __init__(self, val=1):
|
|
self.val = val
|
|
|
|
def foo(self, context):
|
|
context.results.x = str(context.params.i * 5 + self.val)
|
|
|
|
def test_simple_rpc(capability):
|
|
def _restore(ref_id):
|
|
return capability.TestInterface.new_server(Server(100))
|
|
|
|
loop = capnp.EventLoop()
|
|
|
|
read, write = socket.socketpair(socket.AF_UNIX)
|
|
read_stream = capnp.FdAsyncIoStream(read.fileno())
|
|
write_stream = capnp.FdAsyncIoStream(write.fileno())
|
|
|
|
restorer = capnp.Restorer(capability.TestSturdyRefObjectId, _restore)
|
|
server = capnp.RpcServer(loop, restorer, write_stream)
|
|
client = capnp.RpcClient(loop, read_stream)
|
|
|
|
ref = capability.TestSturdyRefObjectId.new_message()
|
|
cap = client.restore(ref)
|
|
cap = cap.cast_as(capability.TestInterface)
|
|
|
|
remote = cap.foo(i=5)
|
|
response = loop.wait_remote(remote)
|
|
|
|
assert response.x == '125'
|