|
# retoor <retoor@molodetz.nl>
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from fastapi import FastAPI
|
|
from starlette.testclient import TestClient
|
|
|
|
from devplacepy_services.base.auth import set_internal_gateway_key
|
|
|
|
|
|
class ServiceTestClient:
|
|
def __init__(
|
|
self,
|
|
app: FastAPI,
|
|
*,
|
|
internal_key: str = "test-internal-key",
|
|
authenticated_user: str | None = None,
|
|
) -> None:
|
|
set_internal_gateway_key(internal_key)
|
|
headers: dict[str, str] = {"X-Internal-Key": internal_key}
|
|
if authenticated_user:
|
|
headers["X-Authenticated-User"] = authenticated_user
|
|
self._client = TestClient(app, headers=headers)
|
|
|
|
def get(self, path: str, **kwargs: Any):
|
|
return self._client.get(path, **kwargs)
|
|
|
|
def post(self, path: str, **kwargs: Any):
|
|
return self._client.post(path, **kwargs)
|
|
|
|
def put(self, path: str, **kwargs: Any):
|
|
return self._client.put(path, **kwargs)
|
|
|
|
def delete(self, path: str, **kwargs: Any):
|
|
return self._client.delete(path, **kwargs)
|
|
|
|
def close(self) -> None:
|
|
self._client.close() |