"""Test OrchardAsset model operations."""

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

from sound_recordings.cypher import orchard_assets as cypher
from sound_recordings.models import orchard_assets


@patch('sound_recordings.models.orchard_assets.get_session')
def test_get_tuids(mock_neo4j_get_session):
    """Test query for tuids on asset."""
    mock_neo4j_session = mock_neo4j_get_session.return_value
    mock_neo4j_results = mock_neo4j_session.run.return_value
    mock_neo4j_results.single.return_value = {
        'tuids': [123, 456]
    }

    results = orchard_assets.get_track_ids('asset456')

    assert mock_neo4j_session.run.called
    assert mock_neo4j_session.run.call_args_list == [
        call(
            cypher.GET_TRACK_IDS,
            asset_id='asset456'
        )
    ]
    assert results == [123, 456]