"""Unit tests for Payment Group Payment Batch schemas."""

from payment.schemas.payment_group_payment_batch import (
    PaymentGroupPaymentBatchSchema,
    PaymentGroupPaymentBatchStatusSchema,
)


def test_payment_group_payment_batch_schema():
    """Test Payment Group Payment Batch schema."""
    mock_dump = {
        'payment_group_payment_batch_id': 1,
        'payment_group_payment_id': 1,
        'payoneer_program_id': 1001,
        'batch_num': 1,
    }

    response = PaymentGroupPaymentBatchSchema().dump(mock_dump)
    assert response == mock_dump


def test_payment_group_payment_batch_status_schema():
    """Test Payment Group Payment Batch Status schema."""
    mock_dump = {
        'payment_group_payment_batch_id': 1,
        'payment_group_payment_id': 1,
        'payoneer_program_id': 1001,
        'batch_num': 1,
        'abacus_state_id': 1,
        'action_status': 'init',
        'error_message': 'Batch Failed',
    }

    response = PaymentGroupPaymentBatchStatusSchema().dump(mock_dump)
    assert response == mock_dump