FilesExpand file tree

master

/

test_serialize_response.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

57 lines (39 loc) · 1.38 KB

master

/

test_serialize_response.py

Copy path

Top

File metadata and controls

• Code
• Blame

57 lines (39 loc) · 1.38 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

from typing import List, Optional

from fastapi import FastAPI

from fastapi.testclient import TestClient

from pydantic import BaseModel

app = FastAPI()

class Item(BaseModel):

name: str

price: Optional[float] = None

owner_ids: Optional[List[int]] = None

@app.get("/items/valid", response_model=Item)

def get_valid():

return {"name": "valid", "price": 1.0}

@app.get("/items/coerce", response_model=Item)

def get_coerce():

return {"name": "coerce", "price": "1.0"}

@app.get("/items/validlist", response_model=List[Item])

def get_validlist():

return [

{"name": "foo"},

{"name": "bar", "price": 1.0},

{"name": "baz", "price": 2.0, "owner_ids": [1, 2, 3]},

]

client = TestClient(app)

def test_valid():

response = client.get("/items/valid")

response.raise_for_status()

assert response.json() == {"name": "valid", "price": 1.0, "owner_ids": None}

def test_coerce():

response = client.get("/items/coerce")

response.raise_for_status()

assert response.json() == {"name": "coerce", "price": 1.0, "owner_ids": None}

def test_validlist():

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]},

]