Added Testing for User route

app/main.py

@@ -23,4 +23,4 @@ app.include_router(vote.router)
 
 @app.get("/")
 def home():
-    return "FastAPI Project"
+    return {"message": "FastAPI Project"}

tests/database.py

@@ -0,0 +1,34 @@
+from fastapi.testclient import TestClient
+import pytest
+from app.database import get_db, Base
+from app.main import app
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
+from app.config import setting
+
+
+SQLALCHEMY_DATABASE_URL = f"postgresql://{setting.database_username}:{setting.database_password}@{setting.database_hostname}:{setting.database_port}/{setting.database_name}_Test"
+
+engine = create_engine(SQLALCHEMY_DATABASE_URL)
+TestingSessionMaker = sessionmaker(autoflush=False, autocommit=False, bind=engine)
+
+@pytest.fixture
+def session():
+    Base.metadata.drop_all(bind=engine)
+    Base.metadata.create_all(bind=engine)
+    db = TestingSessionMaker()
+    try:
+        yield db
+    finally:
+        db.close()
+
+@pytest.fixture
+def client(session):
+    def override_get_db():
+        try:
+            yield session
+        finally:
+            session.close()
+
+    app.dependency_overrides[get_db] = override_get_db
+    yield TestClient(app)

tests/test_users.py

@@ -0,0 +1,13 @@
+from app.schemas import user
+from tests.database import client, session
+
+def test_root(client):
+    response = client.get("/")
+    assert response.status_code == 200
+    assert response.json().get('message') == "FastAPI Project"
+
+def test_create_user(client):
+    response = client.post("/users/", json={"email": "abdullah@gmail.com", "password": "abdullah1234"})
+    new_user = user.UserResponse(**response.json())
+    assert response.status_code == 201
+    assert new_user.email == "abdullah@gmail.com"