"""Test for neo4j identity model."""

from unittest import mock
from unittest.mock import patch

from account.models import neo4j_identity
from account.models.types import ProfileInfo
from tests.unit.conftest import get_transactional_session_mock


@patch('account.models.neo4j_identity.get_session', autospec=None)
def test_get_identity_by_auth0_user_id(mock_session):
    """Test get_identity_by_auth0_user_id."""
    expected = [{'identity_id': 'test-identity-id'}]
    mock_session.return_value = get_transactional_session_mock(expected)
    response = neo4j_identity.get_identity_by_auth0_user_id('test-auth0-user-id')
    assert response
    assert response == 'test-identity-id'


@patch('account.models.neo4j_identity.get_session', autospec=None)
def test_get_settings_profile(mock_session):
    """Test get_identity_settings_profile."""
    identity_uuid = 'uuid'
    profile_obj = {
        'profileType': 'SettingsProfile',
        'profileId': 182392521,
        'roles': [],
        'uuid': '778f7c28-146b-402f-91ea-7532cd0f0181',
    }

    result_node = mock.MagicMock()
    result_node.__getitem__.side_effect = lambda key, default=None: profile_obj.get(key, default)

    mock_session.return_value = get_transactional_session_mock([result_node])
    actual = neo4j_identity.get_identity_settings_profile(identity_uuid)
    assert actual == ProfileInfo(
        profile_type='SettingsProfile',
        profile_id=182392521,
        roles=[],
        uuid='778f7c28-146b-402f-91ea-7532cd0f0181',
    )