"""Unit tests for Payment Group Payment Batch Account model."""

from abacus_common_logic.connectors.database import db
from sqlalchemy import select

from payment.models.payment_group_payment_batch_account import (
    PaymentGroupPaymentBatchAccount,
)
from tests.utils.factories import (
    PaymentGroupPaymentAccountFactory,
    PaymentGroupPaymentBatchFactory,
)


def test_create(mock_accounts):
    """Create a PaymentGroupPaymentBatchAccount."""
    payment_group_payment_account = PaymentGroupPaymentAccountFactory.create()
    payment_group_payment_batch = PaymentGroupPaymentBatchFactory.create()
    payment_group_payment_account_id = (
        payment_group_payment_account.payment_group_payment_account_id
    )
    payment_group_payment_batch_id = (
        payment_group_payment_batch.payment_group_payment_batch_id
    )
    PaymentGroupPaymentBatchAccount.create(
        payment_group_payment_account_id=payment_group_payment_account_id,
        payment_group_payment_batch_id=payment_group_payment_batch_id,
    )

    payment_batch_accounts = (
        db.session.execute(select(PaymentGroupPaymentBatchAccount)).scalars().all()
    )

    assert len(payment_batch_accounts) == 1

    assert (
        payment_batch_accounts[0].payment_group_payment_batch_id
        == payment_group_payment_batch_id
    )
    assert (
        payment_batch_accounts[0].payment_group_payment_account_id
        == payment_group_payment_account_id
    )
    assert (
        payment_batch_accounts[0].payment_group_payment_batch_account_id
        == payment_batch_accounts[0].payment_group_payment_batch_account_id
    )