"""Unit tests for Parent Company model."""

from unittest.mock import MagicMock

from permissions.models import parent_company


def test_get_by_uuid() -> None:
    """Test get_parent_company_by_uuid queries correctly."""
    mock_session = MagicMock()
    test_uuid = 'test-parent-company-uuid'
    mock_pc = MagicMock()
    mock_session.query.return_value.filter_by.return_value.one_or_none.return_value = mock_pc

    result = parent_company.ParentCompany.get_by_uuid(mock_session, test_uuid)

    assert result == mock_pc
    mock_session.query.assert_called_once_with(parent_company.ParentCompany)
    mock_session.query.return_value.filter_by.assert_called_once_with(uuid=test_uuid)
    mock_session.query.return_value.filter_by.return_value.one_or_none.assert_called_once()