From 93c7f6f6cb54166e6b2340f44480ec95eb74acbf Mon Sep 17 00:00:00 2001 From: vikynoah Date: Wed, 11 Dec 2024 17:14:59 +0100 Subject: [PATCH] fix: Async Testing (#26) * fix: httpx fix as per latest version * fix: Fix Async Testing client --- creyPY/fastapi/testing_async.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) 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)