"""Test resource-management functions"""
import uuid

import pytest

from gn3.auth.authorisation.groups import Group
from gn3.auth.authorisation.resources import (
    Resource, user_resources, create_resource, ResourceCategory,
    public_resources)

from tests.unit.auth import conftest

group = Group(uuid.UUID("9988c21d-f02f-4d45-8966-22c968ac2fbf"), "TheTestGroup")
resource_category = ResourceCategory(
    uuid.UUID("fad071a3-2fc8-40b8-992b-cdefe7dcac79"), "mrna", "mRNA Dataset")
create_resource_failure = {
    "status": "error",
    "message": "Unauthorised: Could not create resource"
}
uuid_fn = lambda : uuid.UUID("d32611e3-07fc-4564-b56c-786c6db6de2b")

@pytest.mark.unit_test
@pytest.mark.parametrize(
    "user,expected",
    tuple(zip(
        conftest.TEST_USERS,
        (Resource(
            group, uuid.UUID("d32611e3-07fc-4564-b56c-786c6db6de2b"),
            "test_resource", resource_category, False),
         create_resource_failure,
         create_resource_failure,
         create_resource_failure,
         create_resource_failure))))
def test_create_resource(mocker, test_app, test_users_in_group, user, expected):
    """Test that resource creation works as expected."""
    mocker.patch("gn3.auth.authorisation.resources.uuid4", uuid_fn)
    conn, _group, _users = test_users_in_group
    with test_app.app_context() as flask_context:
        flask_context.g.user = user
        assert create_resource(conn, "test_resource", resource_category) == expected

SORTKEY = lambda resource: resource.resource_id

@pytest.mark.unit_test
def test_public_resources(test_resources):
    """
    GIVEN: some resources in the database
    WHEN: public resources are requested
    THEN: only list the resources that are public
    """
    conn, _res = test_resources
    assert sorted(public_resources(conn), key=SORTKEY) == sorted(tuple(
        res for res in conftest.TEST_RESOURCES if res.public), key=SORTKEY)

PUBLIC_RESOURCES = sorted(conftest.TEST_RESOURCES, key=SORTKEY)

@pytest.mark.skip # REMOVE THIS LINE!!!
@pytest.mark.unit_test
@pytest.mark.parametrize(
    "user,expected",
    tuple(zip(
        conftest.TEST_USERS,
        (sorted(conftest.TEST_RESOURCES, key=SORTKEY),
         sorted(res for res in conftest.TEST_RESOURCES
                if str(res.resource_id) not in
                ("2130aec0-fefd-434d-92fd-9ca342348b2d",
                 "14496a1c-c234-49a2-978c-8859ea274054")),
         PUBLIC_RESOURCES, PUBLIC_RESOURCES))))
def test_user_resources(fixture_user_resources, user, expected):
    """
    GIVEN: some resources in the database
    WHEN: a particular user's resources are requested
    THEN: list only the resources for which the user can access
    """
    assert user_resources(fixture_user_resources, user) == expected