tests/__init__ => conftest

This commit is contained in:
Teknique 2023-06-17 11:37:39 -07:00
parent 553ea7a3e0
commit f1aad14099
No known key found for this signature in database
GPG Key ID: 54D9906A294B7FF3
5 changed files with 39 additions and 46 deletions

View File

@ -1,37 +0,0 @@
import os
from functools import cache
from typing import AsyncGenerator
import pytest_asyncio
import veilid
from veilid.json_api import _JsonVeilidAPI
pytest_plugins = ("pytest_asyncio",)
@cache
def server_info() -> tuple[str, int]:
"""Return the hostname and port of the test server."""
VEILID_SERVER = os.getenv("VEILID_SERVER")
if VEILID_SERVER is None:
return "localhost", 5959
hostname, *rest = VEILID_SERVER.split(":")
if rest:
return hostname, int(rest[0])
return hostname, 5959
async def simple_update_callback(update: veilid.VeilidUpdate):
print(f"VeilidUpdate: {update}")
@pytest_asyncio.fixture
async def api_connection() -> AsyncGenerator[_JsonVeilidAPI, None]:
hostname, port = server_info()
api = await veilid.json_api_connect(hostname, port, simple_update_callback)
async with api:
# purge routes to ensure we start fresh
await api.debug("purge routes")
yield api

View File

@ -0,0 +1,37 @@
import os
from functools import cache
from typing import AsyncGenerator
import pytest_asyncio
import veilid
from veilid.json_api import _JsonVeilidAPI
pytest_plugins = ("pytest_asyncio",)
@cache
def server_info() -> tuple[str, int]:
"""Return the hostname and port of the test server."""
VEILID_SERVER = os.getenv("VEILID_SERVER")
if VEILID_SERVER is None:
return "localhost", 5959
hostname, *rest = VEILID_SERVER.split(":")
if rest:
return hostname, int(rest[0])
return hostname, 5959
async def simple_update_callback(update: veilid.VeilidUpdate):
print(f"VeilidUpdate: {update}")
@pytest_asyncio.fixture
async def api_connection() -> AsyncGenerator[_JsonVeilidAPI, None]:
hostname, port = server_info()
api = await veilid.json_api_connect(hostname, port, simple_update_callback)
async with api:
# purge routes to ensure we start fresh
await api.debug("purge routes")
yield api

View File

@ -5,9 +5,7 @@ import socket
import pytest import pytest
import veilid import veilid
from . import api_connection, simple_update_callback from .conftest import simple_update_callback
##################################################################
@pytest.mark.asyncio @pytest.mark.asyncio

View File

@ -4,10 +4,6 @@ import pytest
import veilid import veilid
from veilid.api import CryptoSystem from veilid.api import CryptoSystem
from . import api_connection
##################################################################
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_best_crypto_system(api_connection): async def test_best_crypto_system(api_connection):

View File

@ -1,12 +1,11 @@
# Routing context veilid tests # Routing context veilid tests
import asyncio import asyncio
import json
import pytest import pytest
import veilid import veilid
from . import api_connection, server_info from .conftest import server_info
################################################################## ##################################################################