Coverage for docs_src/sql_databases/sql_app/tests/test_sql_app.py: 100%

1from fastapi.testclient import TestClient 5 ctx1abcde

2from sqlalchemy import create_engine 5 ctx1abcde

3from sqlalchemy.orm import sessionmaker 5 ctx1abcde

4from sqlalchemy.pool import StaticPool 5 ctx1abcde

5 

6from ..database import Base 5 ctx1abcde

7from ..main import app, get_db 5 ctx1abcde

8 

9SQLALCHEMY_DATABASE_URL = "sqlite://" 5 ctx1abcde

10 

11engine = create_engine( 5 ctx1abcde

12 SQLALCHEMY_DATABASE_URL, 

13 connect_args={"check_same_thread": False}, 

14 poolclass=StaticPool, 

15) 

16TestingSessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) 5 ctx1abcde

17 

18 

19Base.metadata.create_all(bind=engine) 5 ctx1abcde

20 

21 

22def override_get_db(): 5 ctx1abcde

23 try: 5 ctx1abcde

24 db = TestingSessionLocal() 5 ctx1abcde

25 yield db 5 ctx1abcde

26 finally: 

27 db.close() 5 ctx1abcde

28 

29 

30app.dependency_overrides[get_db] = override_get_db 5 ctx1abcde

31 

32client = TestClient(app) 5 ctx1abcde

33 

34 

35def test_create_user(): 5 ctx1abcde

36 response = client.post( 5 ctx1abcde

37 "/users/", 

38 json={"email": "deadpool@example.com", "password": "chimichangas4life"}, 

39 ) 

40 assert response.status_code == 200, response.text 5 ctx1abcde

41 data = response.json() 5 ctx1abcde

42 assert data["email"] == "deadpool@example.com" 5 ctx1abcde

43 assert "id" in data 5 ctx1abcde

44 user_id = data["id"] 5 ctx1abcde

45 

46 response = client.get(f"/users/{user_id}") 5 ctx1abcde

47 assert response.status_code == 200, response.text 5 ctx1abcde

48 data = response.json() 5 ctx1abcde

49 assert data["email"] == "deadpool@example.com" 5 ctx1abcde

50 assert data["id"] == user_id 5 ctx1abcde