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

1from fastapi.testclient import TestClient 9 ctx1stuvwxyzA

2from sqlmodel import Session, SQLModel, create_engine 9 ctx1stuvwxyzA

3 

4from .main import app, get_session # (1)! 9 ctx1stuvwxyzA

5 

6 

7def test_create_hero(): 9 ctx1stuvwxyzA

8 engine = create_engine( 18 ctx1abcdefghijklmnopqr

9 "sqlite:///testing.db", connect_args={"check_same_thread": False} 

10 ) 

11 SQLModel.metadata.create_all(engine) 18 ctx1abcdefghijklmnopqr

12 

13 with Session(engine) as session: 18 ctx1abcdefghijklmnopqr

14 

15 def get_session_override(): 18 ctx1abcdefghijklmnopqr

16 return session 18 ctx1abcdefghijklmnopqr

17 

18 app.dependency_overrides[get_session] = get_session_override 18 ctx1abcdefghijklmnopqr

19 

20 client = TestClient(app) # (2)! 18 ctx1abcdefghijklmnopqr

21 

22 response = client.post( # (3)! 18 ctx1abcdefghijklmnopqr

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

24 ) 

25 app.dependency_overrides.clear() 18 ctx1abcdefghijklmnopqr

26 data = response.json() # (4)! 18 ctx1abcdefghijklmnopqr

27 

28 assert response.status_code == 200 # (5)! 18 ctx1abcdefghijklmnopqr

29 assert data["name"] == "Deadpond" # (6)! 18 ctx1abcdefghijklmnopqr

30 assert data["secret_name"] == "Dive Wilson" # (7)! 18 ctx1abcdefghijklmnopqr

31 assert data["age"] is None # (8)! 18 ctx1abcdefghijklmnopqr

32 assert data["id"] is not None # (9)! 18 ctx1abcdefghijklmnopqr