mirror of
https://gitlab.com/veilid/veilid.git
synced 2024-10-01 01:26:08 -04:00
88 lines
3.1 KiB
Python
88 lines
3.1 KiB
Python
# Routing context veilid tests
|
|
|
|
import veilid
|
|
import pytest
|
|
import asyncio
|
|
import json
|
|
from . import *
|
|
|
|
##################################################################
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_routing_contexts():
|
|
async def func(api: veilid.VeilidAPI):
|
|
rc = await api.new_routing_context()
|
|
rcp = await rc.with_privacy()
|
|
rcps = await rcp.with_sequencing(veilid.Sequencing.ENSURE_ORDERED)
|
|
rcpsr = await rcps.with_custom_privacy(veilid.Stability.RELIABLE)
|
|
await simple_connect_and_run(func)
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_routing_context_app_message_loopback():
|
|
|
|
app_message_queue = asyncio.Queue()
|
|
|
|
async def app_message_queue_update_callback(update: veilid.VeilidUpdate):
|
|
if update.kind == veilid.VeilidUpdateKind.APP_MESSAGE:
|
|
await app_message_queue.put(update)
|
|
|
|
api = await veilid.json_api_connect(VEILID_SERVER, VEILID_SERVER_PORT, app_message_queue_update_callback)
|
|
async with api:
|
|
|
|
# make a routing context that uses a safety route
|
|
rc = await (await api.new_routing_context()).with_privacy()
|
|
|
|
# make a new local private route
|
|
prl, blob = await api.new_private_route()
|
|
|
|
# import it as a remote route as well so we can send to it
|
|
prr = await api.import_remote_private_route(blob)
|
|
|
|
# send an app message to our own private route
|
|
message = b"abcd1234"
|
|
await rc.app_message(prr, message)
|
|
|
|
# we should get the same message back
|
|
update: veilid.VeilidUpdate = await asyncio.wait_for(app_message_queue.get(), timeout=10)
|
|
appmsg: veilid.VeilidAppMessage = update.detail
|
|
assert appmsg.message == message
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_routing_context_app_call_loopback():
|
|
|
|
app_call_queue = asyncio.Queue()
|
|
|
|
async def app_call_queue_update_callback(update: veilid.VeilidUpdate):
|
|
if update.kind == veilid.VeilidUpdateKind.APP_CALL:
|
|
await app_call_queue.put(update)
|
|
|
|
api = await veilid.json_api_connect(VEILID_SERVER, VEILID_SERVER_PORT, app_call_queue_update_callback)
|
|
async with api:
|
|
|
|
# make a routing context that uses a safety route
|
|
rc = await (await api.new_routing_context()).with_privacy()
|
|
|
|
# make a new local private route
|
|
prl, blob = await api.new_private_route()
|
|
|
|
# import it as a remote route as well so we can send to it
|
|
prr = await api.import_remote_private_route(blob)
|
|
|
|
# send an app message to our own private route
|
|
request = b"abcd1234"
|
|
app_call_task = asyncio.create_task(rc.app_call(prr, request), name = "app call task")
|
|
|
|
# we should get the same request back
|
|
update: veilid.VeilidUpdate = await asyncio.wait_for(app_call_queue.get(), timeout=10)
|
|
appcall: veilid.VeilidAppCall = update.detail
|
|
assert appcall.message == request
|
|
|
|
# now we reply to the request
|
|
reply = b"qwer5678"
|
|
await api.app_call_reply(appcall.call_id, reply)
|
|
|
|
# now we should get the reply from the call
|
|
result = await app_call_task
|
|
assert result == reply
|