"""Test PayeeLedgerProcessor."""

from decimal import Decimal
from unittest.mock import MagicMock
from unittest.mock import patch

from payments_posted.ledger_entry_queue import LedgerEntryQueue
from payments_posted.payee_ledger_processor import PayeeLedgerProcessor


def test_payee_ledger_processor_process_row(
        mock_event, mock_payment_group_payment_payee):
    """Test processing a row."""
    mock = MagicMock()
    mock.return_value = mock_payment_group_payment_payee
    mock_queue = LedgerEntryQueue()
    with patch.object(mock_queue, 'flush_entries', mock_flush):
        processor = PayeeLedgerProcessor(1, mock_event, mock_queue)
        for data in mock_payment_group_payment_payee:
            processor.process_row(data)
        assert processor._general_ledger_amount == Decimal('1457551.00')
        assert processor._payee_ledger_amount == Decimal('-1457551.00')


def mock_flush(self):
    """Mock flush method."""
    self._entries = []


def test_payee_ledger_processor_process(
        mock_event, mock_payment_group_payment_payee,
        mock_results, request_engine
):
    """Test process()."""
    request_engine['ows-ledger'].add_spec('POST', '/ledger/bulk', status=201)
    mock_queue = MagicMock()
    processor = PayeeLedgerProcessor(1, mock_event, mock_queue)
    processor.process(mock_payment_group_payment_payee)