"""Unit tests for Vendor model."""

from unittest import mock
from unittest.mock import MagicMock

from permissions.models import vendor


def test_get_by_uuid() -> None:
    """Test get_by_uuid queries correctly."""
    mock_session = MagicMock()
    test_uuid = 'test-vendor-uuid'
    mock_vendor = MagicMock()
    mock_session.query.return_value.filter_by.return_value.one.return_value = mock_vendor

    result = vendor.Vendor.get_by_uuid(mock_session, test_uuid)

    assert result == mock_vendor
    mock_session.query.assert_called_once_with(vendor.Vendor)
    mock_session.query.return_value.filter_by.assert_called_once_with(vendor_uuid=test_uuid)
    mock_session.query.return_value.filter_by.return_value.one.assert_called_once()


@mock.patch('permissions.models.vendor.Vendor.get_by_uuid')
def test_get_vendor_id_by_uuid(mock_get_by_uuid: MagicMock) -> None:
    """Test get_vendor_id_by_uuid returns vendor id."""
    mock_session = MagicMock()
    test_uuid = 'test-vendor-uuid'
    test_vendor_id = 12345

    mock_vendor = MagicMock(vendor_id=test_vendor_id)
    mock_get_by_uuid.return_value = mock_vendor

    result = vendor.get_vendor_id_by_uuid(mock_session, test_uuid)

    assert result == test_vendor_id
    mock_get_by_uuid.assert_called_once_with(mock_session, test_uuid)