Function _make_extractor

tests/extraction/test_runner.py:66–98

Build a fake BatchExtractor with prepare/finalize/batch_provider.

(
    prepared_map: dict[str, PreparedFile],
    finalize_results: dict[str, ExtractionResult] | None = None,
    finalize_error: Exception | None = None,
)

def _make_extractor(
    prepared_map: dict[str, PreparedFile],
    finalize_results: dict[str, ExtractionResult] | None = None,
    finalize_error: Exception | None = None,
) -> _FakeExtractor:
    """Build a fake BatchExtractor with prepare/finalize/batch_provider."""
    ext = _FakeExtractor()

    def _prepare(gz_path: str) -> PreparedFile:
        return prepared_map[gz_path]

    ext.prepare.side_effect = _prepare

    if finalize_error:

        def _finalize(gz_path, prepared, responses):
            raise finalize_error

        ext.finalize.side_effect = _finalize
    elif finalize_results:

        def _finalize(gz_path, prepared, responses):
            return finalize_results[gz_path]

        ext.finalize.side_effect = _finalize
    else:

        def _finalize(gz_path, prepared, responses):
            return _make_result(gz_path)

        ext.finalize.side_effect = _finalize

    return ext


def _make_batch_provider(