ellar/tests/test_response/test_defined_response_model.py at cff8140adf0c2eb423327ac23a21da906115e894 · python-ellar/ellar · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
from typing import List, Optional

from ellar.app import AppFactory
from ellar.common import Inject, ModuleRouter
from ellar.core import ExecutionContext
from pydantic import BaseModel
from starlette.responses import PlainTextResponse

mr = ModuleRouter("")


class Item(BaseModel):
    name: str
    price: Optional[float] = None
    owner_ids: Optional[List[int]] = None


@mr.get("/items/valid", response=Item)
def get_valid():
    return {"name": "valid", "price": 1.0}


@mr.get("/items/coerce", response=Item)
def get_coerce():
    return {"name": "coerce", "price": "1.0"}


@mr.get("/items/validlist", response=List[Item])
def get_validlist():
    return [
        {"name": "foo"},
        {"name": "bar", "price": 1.0},
        {"name": "baz", "price": 2.0, "owner_ids": [1, 2, 3]},
    ]


@mr.get("/items/valid_tuple_return", response={200: List[Item], 201: Item})
def get_valid_tuple_return():
    return {"name": "baz", "price": 2.0, "owner_ids": [1, 2, 3]}, 201


@mr.get("/items/not_found_res_model", response={200: List[Item], 201: Item})
def get_not_found_res_model():
    return {"name": "baz", "price": 2.0, "owner_ids": [1, 2, 3]}, 301


@mr.get("/items/text-case-1", response=PlainTextResponse)
def get_plain_text_case_1():
    return '{"name": "baz", "price": 2.0, "owner_ids": [1, 2, 3]}, 301'


@mr.get("/items/text-case-2", response={200: PlainTextResponse, 201: Item})
def get_plain_text_case_2():
    return '{"name": "baz", "price": 2.0, "owner_ids": [1, 2, 3]}, 301'


@mr.get("/items/text-case-3", response={200: PlainTextResponse})
async def get_plain_text_case_3(ctx: Inject[ExecutionContext]):
    res = PlainTextResponse("Sent Response with going through response process")
    await res(*ctx.get_args())


app = AppFactory.create_app(routers=(mr,))


def test_valid_tuple_return(test_client_factory):
    client = test_client_factory(app)
    response = client.get("/items/valid_tuple_return")
    response.raise_for_status()
    assert response.json() == {"name": "baz", "price": 2.0, "owner_ids": [1, 2, 3]}


def test_get_not_found_res_model_uses_empty_model(test_client_factory):
    client = test_client_factory(app)
    res = client.get("/items/not_found_res_model")
    assert res.status_code == 301
    assert res.json() == {"name": "baz", "price": 2.0, "owner_ids": [1, 2, 3]}


def test_valid(test_client_factory):
    client = test_client_factory(app)
    response = client.get("/items/valid")
    response.raise_for_status()
    assert response.json() == {"name": "valid", "price": 1.0, "owner_ids": None}


def test_coerce(test_client_factory):
    client = test_client_factory(app)
    response = client.get("/items/coerce")
    response.raise_for_status()
    assert response.json() == {"name": "coerce", "price": 1.0, "owner_ids": None}


def test_validlist(test_client_factory):
    client = test_client_factory(app)
    response = client.get("/items/validlist")
    response.raise_for_status()
    assert response.json() == [
        {"name": "foo", "price": None, "owner_ids": None},
        {"name": "bar", "price": 1.0, "owner_ids": None},
        {"name": "baz", "price": 2.0, "owner_ids": [1, 2, 3]},
    ]


def test_plain_test_case_1(test_client_factory):
    client = test_client_factory(app)
    response = client.get("/items/text-case-1")
    response.raise_for_status()
    assert "text/plain" in str(response.headers["content-type"])
    assert response.text == '{"name": "baz", "price": 2.0, "owner_ids": [1, 2, 3]}, 301'


def test_plain_test_case_2(test_client_factory):
    client = test_client_factory(app)
    response = client.get("/items/text-case-2")
    assert "text/plain" in str(response.headers["content-type"])
    assert response.text == '{"name": "baz", "price": 2.0, "owner_ids": [1, 2, 3]}, 301'


def test_sent_without_response_model_response(test_client_factory):
    client = test_client_factory(app)
    response = client.get("/items/text-case-3")
    assert "text/plain" in str(response.headers["content-type"])
    assert response.text == "Sent Response with going through response process"