"""Vendor model tests."""

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

_MOCK_DATA = {
    'vendor': {
        'name': 'Good Soldier Records Limited - AWAL',
        'migrated_to_abacus': True,
        'sap_vendor_id': None,
        'vendor_id': 49780,
        'is_owned': 'No',
        'is_distributor': 'N',
        'first_statement_period': 1
    }
}


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

    vendor = Vendor.get_by_account_id(account_id)

    actual = vendor.to_dict()

    assert actual == {
        'name': 'Good Soldier Records Limited - AWAL',
        'migrated_to_abacus': True,
        'sap_vendor_id': None,
        'vendor_id': 49780,
        'is_owned': 'No',
        'is_distributor': 'N',
        'first_statement_period': 1
    }