137 lines
3.6 KiB
Python
137 lines
3.6 KiB
Python
import pytest
|
|
from bitmask import Bitmask
|
|
from sachet.server.models import Permissions, UserSchema
|
|
from datetime import datetime
|
|
|
|
user_schema = UserSchema()
|
|
|
|
def test_get(client, tokens, validate_info):
|
|
"""Test accessing the user information endpoint as a normal user."""
|
|
|
|
# access user info endpoint
|
|
resp = client.get(
|
|
"/users/jeff",
|
|
headers={
|
|
"Authorization": f"bearer {tokens['jeff']}"
|
|
}
|
|
)
|
|
assert resp.status_code == 200
|
|
validate_info("jeff", resp.get_json())
|
|
|
|
|
|
# access other user's info endpoint
|
|
resp = client.get(
|
|
"/users/administrator",
|
|
headers={
|
|
"Authorization": f"bearer {tokens['jeff']}"
|
|
}
|
|
)
|
|
assert resp.status_code == 403
|
|
|
|
def test_userinfo_admin(client, tokens, validate_info):
|
|
"""Test accessing other user's information as an admin."""
|
|
|
|
# first test that admin can access its own info
|
|
resp = client.get(
|
|
"/users/administrator",
|
|
headers={
|
|
"Authorization": f"bearer {tokens['administrator']}"
|
|
}
|
|
)
|
|
assert resp.status_code == 200
|
|
validate_info("administrator", resp.get_json())
|
|
|
|
# now test accessing other user's info
|
|
resp = client.get(
|
|
"/users/jeff",
|
|
headers={
|
|
"Authorization": f"bearer {tokens['administrator']}"
|
|
}
|
|
)
|
|
assert resp.status_code == 200
|
|
validate_info("jeff", resp.get_json())
|
|
|
|
def test_patch(client, users, tokens, validate_info):
|
|
"""Test modifying user information as an administrator."""
|
|
|
|
# try with regular user to make sure it doesn't work
|
|
resp = client.patch(
|
|
"/users/jeff",
|
|
json = { "permissions": ["ADMIN"] },
|
|
headers={
|
|
"Authorization": f"bearer {tokens['jeff']}"
|
|
}
|
|
)
|
|
assert resp.status_code == 403
|
|
|
|
# test malformed patch
|
|
resp = client.patch(
|
|
"/users/jeff",
|
|
json = "hurr durr",
|
|
headers={
|
|
"Authorization": f"bearer {tokens['administrator']}"
|
|
}
|
|
)
|
|
assert resp.status_code == 400
|
|
|
|
resp = client.patch(
|
|
"/users/jeff",
|
|
json = { "permissions": ["ADMIN"] },
|
|
headers={
|
|
"Authorization": f"bearer {tokens['administrator']}"
|
|
}
|
|
)
|
|
assert resp.status_code == 200
|
|
|
|
# modify the expected values
|
|
users["jeff"]["permissions"] = Bitmask(Permissions.ADMIN)
|
|
|
|
# request new info
|
|
resp = client.get(
|
|
"/users/jeff",
|
|
headers={
|
|
"Authorization": f"bearer {tokens['jeff']}"
|
|
}
|
|
)
|
|
assert resp.status_code == 200
|
|
validate_info("jeff", resp.get_json())
|
|
|
|
def test_put(client, users, tokens, validate_info):
|
|
"""Test replacing user information as an administrator."""
|
|
|
|
# try with regular user to make sure it doesn't work
|
|
resp = client.patch(
|
|
"/users/jeff",
|
|
json = dict(),
|
|
headers={
|
|
"Authorization": f"bearer {tokens['jeff']}"
|
|
}
|
|
)
|
|
assert resp.status_code == 403
|
|
|
|
new_data = {k:v for k, v in users["jeff"].items()}
|
|
new_data["permissions"] = Bitmask(Permissions.ADMIN)
|
|
new_data["register_date"] = datetime(2022,2,2,0,0,0)
|
|
|
|
resp = client.put(
|
|
"/users/jeff",
|
|
json = user_schema.dump(new_data),
|
|
headers={
|
|
"Authorization": f"bearer {tokens['administrator']}"
|
|
}
|
|
)
|
|
assert resp.status_code == 200
|
|
|
|
# modify the expected values
|
|
users["jeff"]["permissions"] = Bitmask(Permissions.ADMIN)
|
|
|
|
# request new info
|
|
resp = client.get(
|
|
"/users/jeff",
|
|
headers={
|
|
"Authorization": f"bearer {tokens['jeff']}"
|
|
}
|
|
)
|
|
assert resp.status_code == 200
|
|
validate_info("jeff", resp.get_json())
|