"""Vendor model tests."""
from decimal import Decimal

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

_MOCK_DATA = {
    'dim_subaccount': {
        'subaccountid': 49780,
        'subaccount_split_type': 'Net',
        'commissionoverride': 0.8
    }
}


@using_mock_snowflake_table(DimSubaccount, _MOCK_DATA)
def test_get_by_account_id():
    """Test getting vendor by account id."""
    subaccount_id = 49780

    actual = DimSubaccount.get_by_id(subaccount_id).to_dict()

    assert actual == {
        'subaccount_id': Decimal(49780),
        'subaccount_split_type': 'Net',
        'commission_override': Decimal('0.8')
    }