Coverage for docs_src/tutorial/fastapi/app_testing/tutorial001/test_main.py: 100%

1import pytest 20 ctx1aubvcwdxeyfzgAhBiCjD

2from fastapi.testclient import TestClient 20 ctx1aubvcwdxeyfzgAhBiCjD

3from sqlmodel import Session, SQLModel, create_engine 20 ctx1aubvcwdxeyfzgAhBiCjD

4from sqlmodel.pool import StaticPool 20 ctx1aubvcwdxeyfzgAhBiCjD

5 

6from .main import Hero, app, get_session 20 ctx1aubvcwdxeyfzgAhBiCjD

7 

8 

9@pytest.fixture(name="session") 20 ctx1aubvcwdxeyfzgAhBiCjD

10def session_fixture(): 20 ctx1aubvcwdxeyfzgAhBiCjD

11 engine = create_engine( 10 ctx1abcdefghij

12 "sqlite://", connect_args={"check_same_thread": False}, poolclass=StaticPool 

13 ) 

14 SQLModel.metadata.create_all(engine) 10 ctx1abcdefghij

15 with Session(engine) as session: 10 ctx1abcdefghij

16 yield session 10 ctx1abcdefghij

17 

18 

19@pytest.fixture(name="client") 20 ctx1aubvcwdxeyfzgAhBiCjD

20def client_fixture(session: Session): 20 ctx1aubvcwdxeyfzgAhBiCjD

21 def get_session_override(): 10 ctx1abcdefghij

22 return session 100 ctx2bbcbdbY + , Z E k F ebfbgb0 - . 1 G l H hbibjb2 / : 3 I m J kblbmb4 ; = 5 K n L nbobpb6 ? @ 7 M o N qbrbsb8 [ ] 9 O p P tbubvb! ^ _ # Q q R wbxbyb$ ` { % S r T zbAbBb' | } ( U s V CbDbEb) ~ ab* W t X

23 

24 app.dependency_overrides[get_session] = get_session_override 10 ctx1abcdefghij

25 client = TestClient(app) 10 ctx1abcdefghij

26 yield client 10 ctx1abcdefghij

27 app.dependency_overrides.clear() 10 ctx1abcdefghij

28 

29 

30def test_create_hero(client: TestClient): 20 ctx1aubvcwdxeyfzgAhBiCjD

31 response = client.post( 10 ctx1Y02468!$')

32 "/heroes/", json={"name": "Deadpond", "secret_name": "Dive Wilson"} 

33 ) 

34 data = response.json() 10 ctx1Y02468!$')

35 

36 assert response.status_code == 200 10 ctx1Y02468!$')

37 assert data["name"] == "Deadpond" 10 ctx1Y02468!$')

38 assert data["secret_name"] == "Dive Wilson" 10 ctx1Y02468!$')

39 assert data["age"] is None 10 ctx1Y02468!$')

40 assert data["id"] is not None 10 ctx1Y02468!$')

41 

42 

43def test_create_hero_incomplete(client: TestClient): 20 ctx1aubvcwdxeyfzgAhBiCjD

44 # No secret_name 

45 response = client.post("/heroes/", json={"name": "Deadpond"}) 10 ctx1+-/;?[^`|~

46 assert response.status_code == 422 10 ctx1+-/;?[^`|~

47 

48 

49def test_create_hero_invalid(client: TestClient): 20 ctx1aubvcwdxeyfzgAhBiCjD

50 # secret_name has an invalid type 

51 response = client.post( 10 ctx2, . : = @ ] _ { } ab

52 "/heroes/", 

53 json={ 

54 "name": "Deadpond", 

55 "secret_name": {"message": "Do you wanna know my secret identity?"}, 

56 }, 

57 ) 

58 assert response.status_code == 422 10 ctx2, . : = @ ] _ { } ab

59 

60 

61def test_read_heroes(session: Session, client: TestClient): 20 ctx1aubvcwdxeyfzgAhBiCjD

62 hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson") 10 ctx1klmnopqrst

63 hero_2 = Hero(name="Rusty-Man", secret_name="Tommy Sharp", age=48) 10 ctx1klmnopqrst

64 session.add(hero_1) 10 ctx1klmnopqrst

65 session.add(hero_2) 10 ctx1klmnopqrst

66 session.commit() 10 ctx1klmnopqrst

67 

68 response = client.get("/heroes/") 10 ctx1klmnopqrst

69 data = response.json() 10 ctx1klmnopqrst

70 

71 assert response.status_code == 200 10 ctx1klmnopqrst

72 

73 assert len(data) == 2 10 ctx1klmnopqrst

74 assert data[0]["name"] == hero_1.name 10 ctx1klmnopqrst

75 assert data[0]["secret_name"] == hero_1.secret_name 10 ctx1klmnopqrst

76 assert data[0]["age"] == hero_1.age 10 ctx1klmnopqrst

77 assert data[0]["id"] == hero_1.id 10 ctx1klmnopqrst

78 assert data[1]["name"] == hero_2.name 10 ctx1klmnopqrst

79 assert data[1]["secret_name"] == hero_2.secret_name 10 ctx1klmnopqrst

80 assert data[1]["age"] == hero_2.age 10 ctx1klmnopqrst

81 assert data[1]["id"] == hero_2.id 10 ctx1klmnopqrst

82 

83 

84def test_read_hero(session: Session, client: TestClient): 20 ctx1aubvcwdxeyfzgAhBiCjD

85 hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson") 10 ctx1EGIKMOQSUW

86 session.add(hero_1) 10 ctx1EGIKMOQSUW

87 session.commit() 10 ctx1EGIKMOQSUW

88 

89 response = client.get(f"/heroes/{hero_1.id}") 10 ctx1EGIKMOQSUW

90 data = response.json() 10 ctx1EGIKMOQSUW

91 

92 assert response.status_code == 200 10 ctx1EGIKMOQSUW

93 assert data["name"] == hero_1.name 10 ctx1EGIKMOQSUW

94 assert data["secret_name"] == hero_1.secret_name 10 ctx1EGIKMOQSUW

95 assert data["age"] == hero_1.age 10 ctx1EGIKMOQSUW

96 assert data["id"] == hero_1.id 10 ctx1EGIKMOQSUW

97 

98 

99def test_update_hero(session: Session, client: TestClient): 20 ctx1aubvcwdxeyfzgAhBiCjD

100 hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson") 10 ctx1FHJLNPRTVX

101 session.add(hero_1) 10 ctx1FHJLNPRTVX

102 session.commit() 10 ctx1FHJLNPRTVX

103 

104 response = client.patch(f"/heroes/{hero_1.id}", json={"name": "Deadpuddle"}) 10 ctx1FHJLNPRTVX

105 data = response.json() 10 ctx1FHJLNPRTVX

106 

107 assert response.status_code == 200 10 ctx1FHJLNPRTVX

108 assert data["name"] == "Deadpuddle" 10 ctx1FHJLNPRTVX

109 assert data["secret_name"] == "Dive Wilson" 10 ctx1FHJLNPRTVX

110 assert data["age"] is None 10 ctx1FHJLNPRTVX

111 assert data["id"] == hero_1.id 10 ctx1FHJLNPRTVX

112 

113 

114def test_delete_hero(session: Session, client: TestClient): 20 ctx1aubvcwdxeyfzgAhBiCjD

115 hero_1 = Hero(name="Deadpond", secret_name="Dive Wilson") 10 ctx1Z13579#%(*

116 session.add(hero_1) 10 ctx1Z13579#%(*

117 session.commit() 10 ctx1Z13579#%(*

118 

119 response = client.delete(f"/heroes/{hero_1.id}") 10 ctx1Z13579#%(*

120 

121 hero_in_db = session.get(Hero, hero_1.id) 10 ctx1Z13579#%(*

122 

123 assert response.status_code == 200 10 ctx1Z13579#%(*

124 

125 assert hero_in_db is None 10 ctx1Z13579#%(*