85 lines
3.3 KiB
Python
85 lines
3.3 KiB
Python
|
|
|
|
from fastapi import security
|
|
import jwt
|
|
|
|
|
|
def test_users_list(test_client,login_user):
|
|
|
|
response = test_client.get("/api/v1/users", headers={"Authorization": "Bearer " + login_user})
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert len(data["data"]) == 2
|
|
|
|
def test_users_list_for_admin(test_client,login_admin):
|
|
|
|
response = test_client.get("/api/v1/users", headers={"Authorization": "Bearer " + login_admin})
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert len(data["data"]) == 3
|
|
|
|
def test_user_create(test_client,login_user):
|
|
user_data = {
|
|
"email": "newuser@example.com",
|
|
"password": "password123",
|
|
"first_name": "New",
|
|
"last_name": "User",
|
|
"is_active": True,
|
|
"is_superuser": False
|
|
}
|
|
response = test_client.post("/api/v1/users", json=user_data, headers={"Authorization": "Bearer " + login_user})
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert "data" in data
|
|
assert data["data"] is not None
|
|
assert data["data"]["email"] == user_data["email"]
|
|
assert data["data"]["first_name"] == user_data["first_name"]
|
|
assert data["data"]["last_name"] == user_data["last_name"]
|
|
assert data["data"]["is_active"] == user_data["is_active"]
|
|
assert data["data"]["is_superuser"] == user_data["is_superuser"]
|
|
|
|
def test_admin_create(test_client,login_user):
|
|
user_data = {
|
|
"email": "newuser@example.com",
|
|
"password": "password123",
|
|
"first_name": "New",
|
|
"last_name": "User",
|
|
"is_active": True,
|
|
"is_superuser": True
|
|
}
|
|
response = test_client.post("/api/v1/users", json=user_data, headers={"Authorization": "Bearer " + login_user})
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert "data" not in data
|
|
|
|
def test_admin_create_for_admin(test_client,login_admin):
|
|
user_data = {
|
|
"email": "admin@example.com",
|
|
"password": "password123",
|
|
"first_name": "New",
|
|
"last_name": "User",
|
|
"is_active": True,
|
|
"is_superuser": True
|
|
}
|
|
response = test_client.post("/api/v1/users", json=user_data, headers={"Authorization": "Bearer " + login_admin})
|
|
assert response.status_code == 200
|
|
data = response.json()
|
|
assert "data" in data
|
|
assert data["data"] is not None
|
|
assert data["data"]["email"] == user_data["email"]
|
|
assert data["data"]["first_name"] == user_data["first_name"]
|
|
assert data["data"]["last_name"] == user_data["last_name"]
|
|
assert data["data"]["is_active"] == user_data["is_active"]
|
|
assert data["data"]["is_superuser"] == user_data["is_superuser"]
|
|
|
|
def test_user_details(test_client, login_user,user):
|
|
payload = jwt.decode(login_user, security.SECRET_KEY, algorithms=[security.ALGORITHM])
|
|
id = payload.get("sub")
|
|
response = test_client.get("/api/v1/users/"+ id)
|
|
assert response.status_code == 200
|
|
assert response.json()["data"]["email"] ==user["email"]
|
|
assert data["data"]["first_name"] == user["first_name"]
|
|
assert data["data"]["last_name"] == user["last_name"]
|
|
assert data["data"]["is_active"] == user["is_active"]
|
|
assert data["data"]["is_superuser"] == user["is_superuser"]
|
|
assert response.json()["data"]["id"] == id |