completed user login and existing user fixture

Author: omogbolahan94

test/conftest.py

@@ -9,7 +9,7 @@
 from app.database import get_db
 from app.database import Base
 from app.oauth2 import create_access_token
-from app import models
+from app import models, utils
 from alembic import command
 
 
@@ -41,5 +41,36 @@ def override_get_db():
             yield session
         finally:
             session.close()
+    
     app.dependency_overrides[get_db] = override_get_db
     yield TestClient(app)
+
+
+@pytest.fixture
+def test_user_exist(session):
+    """
+    Create a test user directly in the test database session. 
+    This is used to verify login test
+    """
+    user = models.Users(
+        email="gbolahan@gmail.com",
+        password=utils.hash("password123")  
+    )
+    session.add(user)
+    session.commit()
+    session.refresh(user)
+
+    return {"id": user.id, "email": user.email, "password": "password123"}
+
+
+@pytest.fixture
+def test_user_exist_not(client):
+    user_data = {"email": "ola@gmail.com",
+                 "password": "123"}
+    res = client.post("/users/", json=user_data)
+
+    assert res.status_code == 201
+
+    new_user = res.json()
+    new_user['password'] = user_data['password']
+    return new_user

test/test_users.py

@@ -6,9 +6,21 @@
 
 def test_create_user(client):
     res = client.post(
-        "/users/", json={"email": "hello123@gmail.com", "password": "password123", "id": 1})
+        "/users/", json={"email": "omogbolahan@gmail.com", "password": "password123", "id": 1})
 
     print(res.json())
     new_user = schemas.UserResponse(**res.json())
-    assert new_user.email == "hello123@gmail.com"
-    assert res.status_code == 201
+    assert new_user.email == "omogbolahan@gmail.com"
+    assert res.status_code == 201
+
+
+def test_login_user(test_user_exist, client):
+    res = client.post(
+        "/login", data={"username": test_user_exist['email'], "password": test_user_exist['password']})
+    login_res = schemas.Token(**res.json())
+    payload = jwt.decode(login_res.access_token,
+                         settings.jwt_secret_key, algorithms=[settings.jwt_algorithm])
+    id = payload.get("user_id")
+    assert id == test_user_exist['id']
+    assert login_res.token_type == "bearer"
+    assert res.status_code == 200