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

1from fastapi.testclient import TestClient 3 ctx1abc

2from sqlalchemy import create_engine 3 ctx1abc

3from sqlalchemy.orm import sessionmaker 3 ctx1abc

4 

5from ..database import Base 3 ctx1abc

6from ..main import app, get_db 3 ctx1abc

7 

8SQLALCHEMY_DATABASE_URL = "sqlite:///./test.db" 3 ctx1abc

9 

10engine = create_engine( 3 ctx1abc

11 SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False} 

12) 

13TestingSessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) 3 ctx1abc

14 

15 

16Base.metadata.create_all(bind=engine) 3 ctx1abc

17 

18 

19def override_get_db(): 3 ctx1abc

20 try: 3 ctx1abc

21 db = TestingSessionLocal() 3 ctx1abc

22 yield db 3 ctx1abc

23 finally: 

24 db.close() 3 ctx1abc

25 

26 

27app.dependency_overrides[get_db] = override_get_db 3 ctx1abc

28 

29client = TestClient(app) 3 ctx1abc

30 

31 

32def test_create_user(): 3 ctx1abc

33 response = client.post( 3 ctx1abc

34 "/users/", 

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

36 ) 

37 assert response.status_code == 200, response.text 3 ctx1abc

38 data = response.json() 3 ctx1abc

39 assert data["email"] == "deadpool@example.com" 3 ctx1abc

40 assert "id" in data 3 ctx1abc

41 user_id = data["id"] 3 ctx1abc

42 

43 response = client.get(f"/users/{user_id}") 3 ctx1abc

44 assert response.status_code == 200, response.text 3 ctx1abc

45 data = response.json() 3 ctx1abc

46 assert data["email"] == "deadpool@example.com" 3 ctx1abc

47 assert data["id"] == user_id 3 ctx1abc