"""Tests for payments generate functions."""

from types import SimpleNamespace as Obj
from unittest.mock import patch

from payments_generate import notify_failure, notify_success


@patch('payments_generate.get_abacus_state_id')
@patch('payments_generate.update_abacus_state')
def test_notify_failure(
        mock_state_update,
        mock_get_state_id,
        payments_generate_dag_config
):
    """Test to update generate_payments action on failure."""
    payment_group_payment_id = payments_generate_dag_config['target_id']
    abacus_state_id = 111
    mock_get_state_id.return_value = abacus_state_id
    notify_failure(Obj(conf=payments_generate_dag_config))
    mock_get_state_id.assert_called_once_with(
        'payment-group-payment',
        payment_group_payment_id,
        'generate_payments'
    )
    mock_state_update.assert_called_once_with(
        abacus_state_id,
        body={'action_status': 'error'}
    )


@patch('payments_generate.get_abacus_state_id')
@patch('payments_generate.update_abacus_state')
def test_notify_success(
        mock_state_update,
        mock_get_state_id,
        payments_generate_dag_config
):
    """Test to update generate_payments action on success."""
    payment_group_payment_id = payments_generate_dag_config['target_id']
    abacus_state_id = 222
    mock_get_state_id.return_value = abacus_state_id
    notify_success(Obj(conf=payments_generate_dag_config))
    mock_get_state_id.assert_called_once_with(
        'payment-group-payment',
        payment_group_payment_id,
        'generate_payments'
    )
    mock_state_update.assert_called_once_with(
        abacus_state_id,
        body={'action_status': 'complete'}
    )