"""Test for backfill vat summary."""
from unittest.mock import patch

import pytest
from scripts.backfill_vat_summary import _verify_data


@patch('scripts.backfill_vat_summary.logging')
@patch('scripts.backfill_vat_summary.LedgerAccountContract')
@patch('scripts.backfill_vat_summary.StatementPeriod')
def test_backfill_vat_summary_verification(
    mock_statement_period, mock_ledger, mock_logging
):
    """Test verification method."""
    row = {
        'statement_period_id': 290,
        'contract_id': 1,
        'account_id': 8,
        'payee_currency_code': 'GBP'
    }
    mock_statement_period.get_by_id.return_value = []
    mock_ledger.get_events_for_account_and_contract.return_value = None

    with pytest.raises(SystemExit) as error:   # noqa: F841
        _verify_data(row, 290)
        mock_logging.error.assert_called_once_with('Check that contract: 1 belongs to account: 8.')