"""payment_entities module tests."""

from unittest.mock import patch

import pytest

from payee.utils.payment_entities import is_awal_account


@pytest.mark.parametrize(
    'payment_entity_name,expected',
    (
        ('AWAL-UK', True),
        ('KNR-UK', False),
        (None, False),
    ),
)
@patch('payee.utils.payment_entities.get_payment_entity_name_by_account_id')
def test_is_awal_account(mock_get_payment_entity_name, payment_entity_name, expected):
    mock_get_payment_entity_name.return_value = payment_entity_name

    assert is_awal_account(123) is expected
    mock_get_payment_entity_name.assert_called_once_with(123)