"""Test add_resource_to_profile."""

from permissions.validations.schemas.resource_to_profile import ResourceToProfileSchema


def test_load():
    """Test load for ResourceToProfileSchema."""
    expected = {
        'profile_id': 24234,
        'profile_type': 'ArtistProfile',
        'resource_id': '123',
        'resource_type': 'ArtistInfo',
        'roles': ['analytics'],
    }

    schema = ResourceToProfileSchema().load(
        {
            'resource_type': 'ArtistInfo',
            'resource_id': '123',
            'profile_type': 'ArtistProfile',
            'profile_id': '24234',
            'roles': ['analytics'],
        }
    )
    assert schema == expected


def test_validate():
    """Test validate for ResourceToProfileSchema."""
    expected = {
        'resource_type': [
            'Must be one of: ArtistInfo, Vendor, Subaccount, Label, SubAccount, LabelParticipant, Collaborator.'
        ]
    }  # noqa

    schema = ResourceToProfileSchema().validate(
        {
            'resource_type': 'dummy',
            'resource_id': '123',
            'profile_type': 'ArtistProfile',
            'profile_id': '24234',
            'roles': ['analytics'],
        }
    )
    assert schema == expected