privateGPT/tests/fixtures/fast_api_test_client.py
Javier Martinez 5851b02378
feat: update llama-index + dependencies (#2092)
* chore: update libraries

* fix: mypy

* chore: more updates

* fix: mypy/black

* chore: fix docker warnings

* fix: mypy

* fix: black
2024-09-26 16:29:52 +02:00

15 lines
465 B
Python

import pytest
from fastapi.testclient import TestClient
from private_gpt.launcher import create_app
from tests.fixtures.mock_injector import MockInjector
@pytest.fixture
def test_client(request: pytest.FixtureRequest, injector: MockInjector) -> TestClient:
if request is not None and hasattr(request, "param"):
injector.bind_settings(request.param or {})
app_under_test = create_app(injector.test_injector)
return TestClient(app_under_test)