Prodream/ormar
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix headers for fastapi 0.65.2+

Browse Source
This commit is contained in:
collerek
2021-06-09 12:01:15 +02:00
parent 6b6dcb396e
commit 2d0ba2383a
4 changed files with 20 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
requirements.txt
View File

@ -24,7 +24,7 @@ pytest
pytest-cov pytest-cov
codecov codecov
pytest-asyncio pytest-asyncio
fastapi<=0.65.1 fastapi
flake8 flake8
flake8-black flake8-black
flake8-bugbear flake8-bugbear

6
tests/test_fastapi/test_binary_fields.py
View File

@ -20,6 +20,8 @@ database = databases.Database(DATABASE_URL, force_rollback=True)
metadata = sqlalchemy.MetaData() metadata = sqlalchemy.MetaData()
app.state.database = database app.state.database = database
headers = {"content-type": "application/json"}
@app.on_event("startup") @app.on_event("startup")
async def startup() -> None: async def startup() -> None:
@ -82,7 +84,9 @@ def test_read_main():
client = TestClient(app) client = TestClient(app)
with client as client: with client as client:
response = client.post( response = client.post(
"/things", data=json.dumps({"bt": base64.b64encode(blob3).decode()}) "/things",
data=json.dumps({"bt": base64.b64encode(blob3).decode()}),
headers=headers,
) )
assert response.status_code == 200 assert response.status_code == 200
response = client.get("/things") response = client.get("/things")

6
tests/test_fastapi/test_nested_saving.py
View File

@ -16,6 +16,8 @@ metadata = sqlalchemy.MetaData()
database = databases.Database(DATABASE_URL, force_rollback=True) database = databases.Database(DATABASE_URL, force_rollback=True)
app.state.database = database app.state.database = database
headers = {"content-type": "application/json"}
@app.on_event("startup") @app.on_event("startup")
async def startup() -> None: async def startup() -> None:
@ -151,7 +153,9 @@ def test_saving_related_in_fastapi():
}, },
], ],
} }
response = client.post("/departments/", data=json.dumps(payload)) response = client.post(
"/departments/", data=json.dumps(payload), headers=headers
)
department = Department(**response.json()) department = Department(**response.json())
assert department.id is not None assert department.id is not None

12
tests/test_fastapi/test_skip_reverse_models.py
View File

@ -15,6 +15,8 @@ metadata = sqlalchemy.MetaData()
database = databases.Database(DATABASE_URL, force_rollback=True) database = databases.Database(DATABASE_URL, force_rollback=True)
app.state.database = database app.state.database = database
headers = {"content-type": "application/json"}
@app.on_event("startup") @app.on_event("startup")
async def startup() -> None: async def startup() -> None:
@ -106,7 +108,9 @@ def test_queries():
wrong_category = {"name": "Test category2", "posts": [{"title": "Test Post"}]} wrong_category = {"name": "Test category2", "posts": [{"title": "Test Post"}]}
# cannot add posts if skipped, will be ignored (with extra=ignore by default) # cannot add posts if skipped, will be ignored (with extra=ignore by default)
response = client.post("/categories/", data=json.dumps(wrong_category)) response = client.post(
"/categories/", data=json.dumps(wrong_category), headers=headers
)
assert response.status_code == 200 assert response.status_code == 200
response = client.get("/categories/") response = client.get("/categories/")
assert response.status_code == 200 assert response.status_code == 200
@ -115,7 +119,9 @@ def test_queries():
assert categories[0] is not None assert categories[0] is not None
assert categories[0].name == "Test category2" assert categories[0].name == "Test category2"
response = client.post("/categories/", data=json.dumps(right_category)) response = client.post(
"/categories/", data=json.dumps(right_category), headers=headers
)
assert response.status_code == 200 assert response.status_code == 200
response = client.get("/categories/") response = client.get("/categories/")
@ -129,7 +135,7 @@ def test_queries():
"author": {"first_name": "John", "last_name": "Smith"}, "author": {"first_name": "John", "last_name": "Smith"},
"categories": [{"name": "New cat"}], "categories": [{"name": "New cat"}],
} }
response = client.post("/posts/", data=json.dumps(right_post)) response = client.post("/posts/", data=json.dumps(right_post), headers=headers)
assert response.status_code == 200 assert response.status_code == 200
Category.__config__.extra = "allow" Category.__config__.extra = "allow"