Python ingest_multiple_dates_main примеры использования

Пример #1

def _test_ingest_multiple_dates_main(params):
    with mock.patch(
        'sherlock.batch.ingest_multiple_dates.ETLStep',
        return_value=EtlTest(),
        autospec=True,
    ):
        validate_execution_results(ingest_multiple_dates_main(params))

Пример #2

def _test_ingest_multiple_dates_main_exceptions(params):
    excToStatus = {
        KeyboardInterruptError: (
            'cancelled' if params.load_only or params.serial_stepper else 'error'
        ),
        KeyboardInterrupt: 'cancelled',
        ProcessError: 'error',
        SystemExit: ('unknown' if params.load_only or params.serial_stepper else 'error')
    }
    for exc_type in excToStatus.keys():
        with mock.patch(
            'sherlock.batch.ingest_multiple_dates.ETLStep',
            return_value=EtlErrorTest(exc_type),
            autospec=True,
        ):
            expected_status = excToStatus[exc_type]
            validate_execution_results(
                ingest_multiple_dates_main(params), expected_status
            )