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

1import pytest # (1)! 10 ctx1abcdefghij

2from fastapi.testclient import TestClient 10 ctx1abcdefghij

3from sqlmodel import Session, SQLModel, create_engine 10 ctx1abcdefghij

4from sqlmodel.pool import StaticPool 10 ctx1abcdefghij

5 

6from .main import app, get_session 10 ctx1abcdefghij

7 

8 

9@pytest.fixture(name="session") # (2)! 10 ctx1abcdefghij

10def session_fixture(): # (3)! 10 ctx1abcdefghij

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 # (4)! 10 ctx1abcdefghij

17 

18 

19def test_create_hero(session: Session): # (5)! 10 ctx1abcdefghij

20 def get_session_override(): 20 ctx1klmnopqrstuvwxyzABCD

21 return session # (6)! 20 ctx1klmnopqrstuvwxyzABCD

22 

23 app.dependency_overrides[get_session] = get_session_override 20 ctx1klmnopqrstuvwxyzABCD

24 

25 client = TestClient(app) 20 ctx1klmnopqrstuvwxyzABCD

26 

27 response = client.post( 20 ctx1klmnopqrstuvwxyzABCD

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

29 ) 

30 app.dependency_overrides.clear() 20 ctx1klmnopqrstuvwxyzABCD

31 data = response.json() 20 ctx1klmnopqrstuvwxyzABCD

32 

33 assert response.status_code == 200 20 ctx1klmnopqrstuvwxyzABCD

34 assert data["name"] == "Deadpond" 20 ctx1klmnopqrstuvwxyzABCD

35 assert data["secret_name"] == "Dive Wilson" 20 ctx1klmnopqrstuvwxyzABCD

36 assert data["age"] is None 20 ctx1klmnopqrstuvwxyzABCD

37 assert data["id"] is not None 20 ctx1klmnopqrstuvwxyzABCD