"""Revenue NR model tests."""
from moneyhub.models import RevenueNr
from tests.unit.conftest import using_mock_snowflake_table

ACCOUNT_ID = 24601
_MOCK_DATA = {
    'revenue_nr_dbt': {
        'account_id': ACCOUNT_ID,
    },
}


@using_mock_snowflake_table(RevenueNr, _MOCK_DATA)
def test_get_by_account_id():
    """Test getting nr revenue by account."""
    actual = RevenueNr.get_by_account_id(ACCOUNT_ID)

    assert actual.account_id == ACCOUNT_ID