"""Test for model identity."""

from unittest.mock import patch

from notifications.models import label


def test_get_default_brand_for_vendor():
    """Test get_default_brand_for_vendor."""
    with patch('notifications.models.label.get_session') as mock_fn:
        session_mock = mock_fn.return_value
        label.get_default_brand_for_label('Vendor', 7123)

        assert session_mock.run.call_count == 1
        assert session_mock.run.call_args[0] == (
            'MATCH (cb:CompanyBrand)-[:HAS_LABEL]->(v:Vendor) WHERE v.id = 7123 RETURN cb.name',
        )
        assert session_mock.run.call_args[1] == {'feed_id': 7123}


def test_get_default_brand_for_subaccount():
    """Test get_default_brand_for_subaccount."""
    with patch('notifications.models.label.get_session') as mock_fn:
        session_mock = mock_fn.return_value
        label.get_default_brand_for_label('Subaccount', 54219)

        assert session_mock.run.call_count == 1
        assert session_mock.run.call_args[0] == (
            """MATCH (cb:CompanyBrand)-[:HAS_LABEL]->(v:Vendor)-[:OWNS]->(sa:Subaccount) WHERE sa.id = 54219 RETURN cb.name""",  # noqa: E501
        )
        assert session_mock.run.call_args[1] == {'feed_id': 54219}