diff --git a/creyPY/fastapi/testing_async.py b/creyPY/fastapi/testing_async.py index 1e1fb5d..d4a744d 100644 --- a/creyPY/fastapi/testing_async.py +++ b/creyPY/fastapi/testing_async.py @@ -1,13 +1,11 @@ import json -from httpx import AsyncClient, ASGITransport +from httpx import ASGITransport, AsyncClient class AsyncGenericClient: - def __init__(self, app): - self.c = AsyncClient( - transport=ASGITransport(app=app), base_url="http://testserver", follow_redirects=True - ) - self.default_headers = {} + def __init__(self, app, headers = {}): + self.c = AsyncClient(transport=ASGITransport(app=app), base_url="http://testserver", follow_redirects=True) + self.default_headers = headers async def get(self, url: str, r_code: int = 200, parse_json=True): re = await self.c.get(url, headers=self.default_headers)