"""Test for a vendor model."""

from unittest.mock import patch

from product_digital.models import neo4j_vendor
from tests.conftest import get_transactional_session_mock


@patch('product_digital.models.neo4j_vendor.get_session', autospec=None)
def test_get_recent_awal_artist(mock_session):
    """Test neo4j awal query."""
    mock_session.return_value = get_transactional_session_mock([{'lp': {'name': 'awal'}}])
    vendor_id = 1234
    response = neo4j_vendor.get_recent_awal_artist(vendor_id)
    assert response
    assert response == 'awal'