Browse Source

Simplify tests

pull/9704/head
Motov Yurii 1 week ago
committed by GitHub
parent
commit
00006bb244
No known key found for this signature in database GPG Key ID: B5690EEEBB952194
  1. 11
      tests/test_response_model_default_factory.py

11
tests/test_response_model_default_factory.py

@ -1,18 +1,13 @@
import functools
import random
from fastapi import FastAPI from fastapi import FastAPI
from fastapi.testclient import TestClient from fastapi.testclient import TestClient
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
app = FastAPI() app = FastAPI()
messages = ["操作成功.", "Successful operation."]
class ResponseModel(BaseModel): class ResponseModel(BaseModel):
code: int = 200 code: int = 200
message: str = Field(default_factory=functools.partial(random.choice, messages)) message: str = Field(default_factory=lambda: "Successful operation.")
@app.get( @app.get(
@ -40,7 +35,7 @@ def test_response_model_has_default_factory_return_dict():
assert response.status_code == 200, response.text assert response.status_code == 200, response.text
assert response.json()["code"] == 200 assert response.json()["code"] == 200
assert response.json()["message"] in messages assert response.json()["message"] == "Successful operation."
def test_response_model_has_default_factory_return_model(): def test_response_model_has_default_factory_return_model():
@ -49,4 +44,4 @@ def test_response_model_has_default_factory_return_model():
assert response.status_code == 200, response.text assert response.status_code == 200, response.text
assert response.json()["code"] == 200 assert response.json()["code"] == 200
assert response.json()["message"] in messages assert response.json()["message"] == "Successful operation."

Loading…
Cancel
Save