Modified test to check right and wrong password during test login

This commit is contained in:
Marcin-Ramotowski 2025-03-27 19:41:48 +00:00
parent d870bcc613
commit c63d087510

View File

@ -1,6 +1,7 @@
import json import json
from models import User, db from models import User, db
from flask_jwt_extended import create_access_token from flask_jwt_extended import create_access_token
from werkzeug.security import generate_password_hash
def test_create_user(test_client): def test_create_user(test_client):
"""New user registration test""" """New user registration test"""
@ -14,17 +15,24 @@ def test_create_user(test_client):
assert data["username"] == "testuser" assert data["username"] == "testuser"
def test_login(test_client): def test_login(test_client):
"""User login test with wrong password""" """User login test"""
user = User(username="testuser", email="test@example.com", password="hashed_pass", role="User") hashed_pass = generate_password_hash("testpass")
user = User(username="testuser", email="test@example.com", password=hashed_pass, role="User")
db.session.add(user) db.session.add(user)
db.session.commit() db.session.commit()
response = test_client.post( response = test_client.post(
"/login", "/login",
data=json.dumps({"username": "testuser", "password": "testpass"}), data=json.dumps({"username": "testuser", "password": "wrongpass"}),
content_type="application/json", content_type="application/json",
) )
assert response.status_code == 401 # User should not be logged - wrong password assert response.status_code == 401 # User should not be logged - wrong password
response = test_client.post(
"/login",
data=json.dumps({"username": "testuser", "password": "testpass"}),
content_type="application/json",
)
assert response.status_code == 200 # User should be logged - right password
def test_get_users(test_client): def test_get_users(test_client):
"""Get all users test - JWT required""" """Get all users test - JWT required"""