48 lines
1.7 KiB
Python
48 lines
1.7 KiB
Python
import json
|
|
from models import User, db
|
|
from flask_jwt_extended import create_access_token
|
|
|
|
def test_create_user(test_client):
|
|
"""New user registration test"""
|
|
response = test_client.post(
|
|
"/users",
|
|
data=json.dumps({"username": "testuser", "email": "test@example.com", "password": "testpass", "role": "User"}),
|
|
content_type="application/json",
|
|
)
|
|
assert response.status_code == 201 # User should be created successfully
|
|
data = response.get_json()
|
|
assert data["username"] == "testuser"
|
|
|
|
def test_login(test_client):
|
|
"""User login test with wrong password"""
|
|
user = User(username="testuser", email="test@example.com", password="hashed_pass", role="User")
|
|
db.session.add(user)
|
|
db.session.commit()
|
|
|
|
response = test_client.post(
|
|
"/login",
|
|
data=json.dumps({"username": "testuser", "password": "testpass"}),
|
|
content_type="application/json",
|
|
)
|
|
assert response.status_code == 401 # User should not be logged - wrong password
|
|
|
|
def test_get_users(test_client):
|
|
"""Get all users test - JWT required"""
|
|
response = test_client.get("/users")
|
|
assert response.status_code == 401
|
|
|
|
def test_get_user_with_token(test_client):
|
|
"""Test to get user data after auth using JWT token"""
|
|
user = User(username="admin", email="admin@example.com", password="hashed_pass", role="Administrator")
|
|
print(user.id)
|
|
db.session.add(user)
|
|
db.session.commit()
|
|
|
|
access_token = create_access_token(identity=str(user.id))
|
|
headers = {"Authorization": f"Bearer {access_token}"}
|
|
|
|
response = test_client.get(f"/users/{user.id}", headers=headers)
|
|
assert response.status_code == 200
|
|
data = response.get_json()
|
|
assert data["username"] == "admin"
|