"""Test PayeeLedgerProcessor."""

from unittest.mock import MagicMock
from unittest.mock import patch

from commit_royalty.ledger_entry_queue import LedgerEntryQueue
from commit_royalty.payee_ledger_processor import PayeeLedgerProcessor


def test_payee_ledger_processor_process_row(
        mock_event, mock_sales, mock_results):
    """Test processing a row."""
    mock_sf = MagicMock()
    mock_sf.fetchall.return_value = mock_sales
    mock_queue = LedgerEntryQueue()
    with patch.object(mock_queue, 'flush_entries', mock_flush):
        processor = PayeeLedgerProcessor(1, mock_event, mock_sf, mock_queue)
        for sales in mock_sales:
            processor.process_row(sales)
        assert processor._entries._entries == mock_results[0:4]


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


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