"""Revenue Distro model tests."""

from moneyhub.models import RevenueDistro
from tests.unit.conftest import using_mock_snowflake_table

ACCOUNT_ID = 24601
SUBACCOUNT_ID = 1
_MOCK_DATA = {
    'revenue_distro_dbt': {
        'account_id': ACCOUNT_ID,
        'subaccount_id': SUBACCOUNT_ID
    },
}


@using_mock_snowflake_table(RevenueDistro, _MOCK_DATA)
def test_get_by_account_id():
    """Test getting distro revenue by account."""
    actual = RevenueDistro.get_by_account_id(ACCOUNT_ID, None, False)

    assert actual.account_id == ACCOUNT_ID


@using_mock_snowflake_table(RevenueDistro, _MOCK_DATA)
def test_get_by_account_id_with_subaccount():
    """Test getting distro revenue by account and subaccount id."""
    actual = RevenueDistro.get_by_account_id(ACCOUNT_ID, SUBACCOUNT_ID, True)

    assert actual.account_id == ACCOUNT_ID


@using_mock_snowflake_table(RevenueDistro, _MOCK_DATA)
def test_get_by_account_id_with_subaccount_no_data():
    """Test getting distro revenue by account and subaccount id no data."""
    subaccount_id = 2
    actual = RevenueDistro.get_by_account_id(ACCOUNT_ID, subaccount_id, True)

    assert actual is None