creyD/creyPY
1
0
Fork 0
mirror of https://github.com/creyD/creyPY.git synced 2026-04-16 13:20:31 +02:00
Code Issues Packages Projects Releases Wiki Activity

Minor bugfix

Browse Source
This commit is contained in:
Conrad
2024-04-02 11:40:26 +02:00
parent 6fc0d01189
commit 140c6e4678
2 changed files with 21 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
creyPY/fastapi/testing.py
View File

@@ -7,21 +7,23 @@ class GenericClient(TestClient):
def __init__(self, client): def __init__(self, client):
self.c = TestClient(client) self.c = TestClient(client)
def get(self, url: str, r_code:int =200, parse_json=True): def get(self, url: str, r_code: int = 200, parse_json=True):
re = self.c.get(url) re = self.c.get(url)
if re.status_code != r_code: if re.status_code != r_code:
print(re.content) print(re.content)
assert r_code == re.status_code assert r_code == re.status_code
return re.json() if parse_json else re.content return re.json() if parse_json else re.content
def delete(self, url: str, r_code:int =204): def delete(self, url: str, r_code: int = 204):
re = self.c.delete(url) re = self.c.delete(url)
if re.status_code != r_code: if re.status_code != r_code:
print(re.content) print(re.content)
assert r_code == re.status_code assert r_code == re.status_code
return re.json() if r_code != 204 else None return re.json() if r_code != 204 else None
def post(self, url: str, obj: dict ={}, r_code:int =201, raw_response=False, *args, **kwargs): def post(
self, url: str, obj: dict = {}, r_code: int = 201, raw_response=False, *args, **kwargs
):
re = self.c.post( re = self.c.post(
url, data=json.dumps(obj), headers={"Content-Type": "application/json"}, *args, **kwargs url, data=json.dumps(obj), headers={"Content-Type": "application/json"}, *args, **kwargs
) )
@@ -30,14 +32,16 @@ class GenericClient(TestClient):
assert r_code == re.status_code assert r_code == re.status_code
return re.json() if not raw_response else re return re.json() if not raw_response else re
def post_file(self, url: str, file, r_code:int =201, raw_response=False, *args, **kwargs): def post_file(self, url: str, file, r_code: int = 201, raw_response=False, *args, **kwargs):
re = self.c.post(url, files={"file": file}, *args, **kwargs) re = self.c.post(url, files={"file": file}, *args, **kwargs)
if re.status_code != r_code: if re.status_code != r_code:
print(re.content) print(re.content)
assert r_code == re.status_code assert r_code == re.status_code
return re.json() if not raw_response else re return re.json() if not raw_response else re
def patch(self, url: str, obj: dict ={}, r_code:int =200, raw_response=False, *args, **kwargs): def patch(
self, url: str, obj: dict = {}, r_code: int = 200, raw_response=False, *args, **kwargs
):
re = self.c.patch( re = self.c.patch(
url, data=json.dumps(obj), headers={"Content-Type": "application/json"}, *args, **kwargs url, data=json.dumps(obj), headers={"Content-Type": "application/json"}, *args, **kwargs
) )
@@ -46,7 +50,7 @@ class GenericClient(TestClient):
assert r_code == re.status_code assert r_code == re.status_code
return re.json() if not raw_response else re return re.json() if not raw_response else re
def put(self, url: str, obj: dict = {}, r_code:int =200, raw_response=False, *args, **kwargs): def put(self, url: str, obj: dict = {}, r_code: int = 200, raw_response=False, *args, **kwargs):
re = self.c.put( re = self.c.put(
url, url,
data=json.dumps(obj), data=json.dumps(obj),
@@ -103,4 +107,4 @@ class GenericClient(TestClient):
assert len(re) == 0 assert len(re) == 0
# GET # GET
self.get(f"{url}{obj_id}", parse_json=False, r_code:int =404) self.get(f"{url}{obj_id}", parse_json=False, r_code=404)

10
setup.py
View File

@@ -33,4 +33,14 @@ setup(
license="MIT", license="MIT",
python_requires=">=3.12", python_requires=">=3.12",
install_requires=requirements, install_requires=requirements,
keywords=[
"creyPY",
"Python",
"FastAPI",
"shortcuts",
"snippets",
"utils",
"personal library",
],
platforms="any",
) )