"""Test for VendContact model."""

from unittest import mock

from account.models.vend_contact import VendContact


def test_get_master_contact_success():
    """Test get_master_contact when 1 vend_contact is found."""
    vendor_id = 1
    session_mock = mock.Mock()
    filter_mock = mock.Mock()
    session_mock.query.return_value.filter_by = filter_mock
    expected_vend_contact = mock.Mock(id=2)
    filter_mock.return_value.all.return_value = [expected_vend_contact]

    result = VendContact.get_master_contact(tx=session_mock, vendor_id=vendor_id)
    assert result == [expected_vend_contact]
    session_mock.query.assert_called_once_with(VendContact)
    filter_mock.assert_called_once_with(vendor_id=vendor_id, master='Y', active='Y')