"""Test model ProfileUser."""

from grass.consts import headers, user
from grass.models import profile_user


def test_get_user_by_identity():
    """Test get_user_by_identity."""
    identity = 'abc1232243'
    uuid = 'uuid'
    actual = profile_user.get_user_by_identity(identity, uuid)

    assert actual.orchard_identity_id == identity
    assert actual.orchard_identity_uuid == uuid
    assert actual.user_group == user.UserGroups.PROFILE
    assert not actual.has_active_profile


def test_set_roles():
    """Test"""

    user = profile_user.ProfileUser()

    token = {
        headers.JWT_PROFILES: [
            {
                'profile_type': 'otherprofile',
                'profile_id': '1234',
                'roles': ['analytics', 'accounting'],
            },
            {'profile_type': 'dummy', 'profile_id': '456', 'roles': ['catalog']},
        ]
    }

    assert user.set_roles(token, 1234, 'otherprofile')
    assert user.roles == ['analytics', 'accounting']

    user.set_roles(token, 123, 'otherprofile') is False
    assert user.roles is None

    assert user.set_roles(token, 1234, 'dummy') is False
    assert user.roles is None