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())