"""Unit tests for AccountContract model."""
from abacus_contract.models.account_contract import AccountContract
from tests.utils.factories import ContractFactory


def test_create(create_mock_account):
    """Test creation of account_contract."""
    account_id = 1
    contract = ContractFactory.create()
    account_contract = AccountContract.create(
        account_id=account_id,
        contract_id=contract.contract_id
    )

    results = AccountContract.query.all()

    assert len(results) == 1
    assert all(res.account_id == account_contract.account_id for res in results)
    assert all(res.contract_id == account_contract.contract_id for res in results)
    assert all(
        res.account_contract_id == account_contract.account_contract_id
        for res in results
    )