Coverage for docs_src / tutorial / fastapi / app_testing / tutorial001_py39 / test_main_004.py: 100%

1from fastapi.testclient import TestClient 9 ctx1stuvwxyzA

2from sqlmodel import Session, SQLModel, create_engine 9 ctx1stuvwxyzA

3from sqlmodel.pool import StaticPool # (1)! 9 ctx1stuvwxyzA

4 

5from .main import app, get_session 9 ctx1stuvwxyzA

6 

7 

8def test_create_hero(): 9 ctx1stuvwxyzA

9 engine = create_engine( 18 ctx1abcdefghijklmnopqr

10 "sqlite://", # (2)! 

11 connect_args={"check_same_thread": False}, 

12 poolclass=StaticPool, # (3)! 

13 ) 

14 SQLModel.metadata.create_all(engine) 18 ctx1abcdefghijklmnopqr

15 

16 with Session(engine) as session: 18 ctx1abcdefghijklmnopqr

17 

18 def get_session_override(): 18 ctx1abcdefghijklmnopqr

19 return session 18 ctx1abcdefghijklmnopqr

20 

21 app.dependency_overrides[get_session] = get_session_override 18 ctx1abcdefghijklmnopqr

22 

23 client = TestClient(app) 18 ctx1abcdefghijklmnopqr

24 

25 response = client.post( 18 ctx1abcdefghijklmnopqr

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

27 ) 

28 app.dependency_overrides.clear() 18 ctx1abcdefghijklmnopqr

29 data = response.json() 18 ctx1abcdefghijklmnopqr

30 

31 assert response.status_code == 200 18 ctx1abcdefghijklmnopqr

32 assert data["name"] == "Deadpond" 18 ctx1abcdefghijklmnopqr

33 assert data["secret_name"] == "Dive Wilson" 18 ctx1abcdefghijklmnopqr

34 assert data["age"] is None 18 ctx1abcdefghijklmnopqr

35 assert data["id"] is not None 18 ctx1abcdefghijklmnopqr