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))
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 )