"""Test timeseries model operations."""

from unittest.mock import call
from unittest.mock import patch

from socials.models import pending


account_id = 569002
identity_id = '6d7d12e7-1158-4c4f-a5ac-96275605d212'

url = 'https://www.instagram.com/alyvjpg'
platform = 'instagram'


@patch('socials.models.pending.get_session')
def test_get_pending_social_accounts_success(
    mock_neo4j_get_session
):
    """Test successful retrieval of pending social media accounts."""
    mock_neo4j_session = mock_neo4j_get_session.return_value
    mock_neo4j_session.run.return_value = [{
        'psa': {
            'url': url,
            'platform': platform
        }
    }]

    result = pending.get_pending_social_accounts(
        account_id,
        identity_id
    )

    assert mock_neo4j_session.run.called
    assert mock_neo4j_session.run.call_args_list == [
        call(
            pending.pending_social_query,
            account_id=account_id,
            identity_id=identity_id
        )
    ]
    assert result == [{
        'url': url,
        'platform': platform
    }]