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

1import pytest 7 ctx1gabcdef

2from fastapi.testclient import TestClient 7 ctx1gabcdef

3from sqlmodel import Session, SQLModel, create_engine 7 ctx1gabcdef

4from sqlmodel.pool import StaticPool 7 ctx1gabcdef

5 

6from .main import app, get_session 7 ctx1gabcdef

7 

8 

9@pytest.fixture(name="session") 7 ctx1gabcdef

10def session_fixture(): 6 ctx1abcdef

11 engine = create_engine( 7 ctx1gabcdef

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

13 ) 

14 SQLModel.metadata.create_all(engine) 7 ctx1gabcdef

15 with Session(engine) as session: 7 ctx1gabcdef

16 yield session 7 ctx1gabcdef

17 

18 

19@pytest.fixture(name="client") # (1)! 7 ctx1gabcdef

20def client_fixture(session: Session): # (2)! 7 ctx1gabcdef

21 def get_session_override(): # (3)! 7 ctx1gabcdef

22 return session 14 ctx1hijklmnopqrstu

23 

24 app.dependency_overrides[get_session] = get_session_override # (4)! 7 ctx1gabcdef

25 

26 client = TestClient(app) # (5)! 7 ctx1gabcdef

27 yield client # (6)! 7 ctx1gabcdef

28 app.dependency_overrides.clear() # (7)! 7 ctx1gabcdef

29 

30 

31def test_create_hero(client: TestClient): # (8)! 7 ctx1gabcdef

32 response = client.post( 14 ctx1hijklmnopqrstu

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

34 ) 

35 data = response.json() 14 ctx1hijklmnopqrstu

36 

37 assert response.status_code == 200 14 ctx1hijklmnopqrstu

38 assert data["name"] == "Deadpond" 14 ctx1hijklmnopqrstu

39 assert data["secret_name"] == "Dive Wilson" 14 ctx1hijklmnopqrstu

40 assert data["age"] is None 14 ctx1hijklmnopqrstu

41 assert data["id"] is not None 14 ctx1hijklmnopqrstu