mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-07-12 00:40:07 +08:00
Jack Michaud
16 lines
469 B
Python
16 lines
469 B
Python
import pytest
|
|
from bot.api_client import OasstApiClient
|
|
|
|
from oasst_shared.schemas import protocol as protocol_schema
|
|
|
|
|
|
@pytest.fixture
|
|
def oasst_api_client_mocked():
|
|
client = OasstApiClient(backend_url="http://localhost:8080", api_key="123")
|
|
yield client
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_fetch_task(oasst_api_client_mocked: OasstApiClient):
|
|
assert await oasst_api_client_mocked.fetch_task(task_type=protocol_schema.TaskRequestType.random) is not None
|