Function parse_multipart_data

tests/test_responses.py:832–870

(data: bytes, boundary: bytes | str)

def parse_multipart_data(data: bytes, boundary: bytes | str) -> list[MultipartPart]:
    parts: list[MultipartPart] = []
    done = False

    current_headers: dict[bytes, bytes] = {}
    current_header_field: bytes = b""

    def on_part_begin() -> None:
        nonlocal current_headers
        current_headers = {}

    def on_part_data(data: bytes, start: int, end: int) -> None:
        parts.append(MultipartPart(current_headers, data[start:end]))

    def on_header_field(data: bytes, start: int, end: int) -> None:
        nonlocal current_header_field
        current_header_field = data[start:end]

    def on_header_value(data: bytes, start: int, end: int) -> None:
        current_headers[current_header_field] = data[start:end]

    def on_end() -> None:
        nonlocal done
        done = True

    parser = MultipartParser(
        boundary,
        dict(
            on_part_begin=on_part_begin,
            on_part_data=on_part_data,
            on_header_field=on_header_field,
            on_header_value=on_header_value,
            on_end=on_end,
        ),
    )
    parser.write(data)
    parser.finalize()
    assert done
    return parts


def test_file_response_insert_ranges(file_response_client: TestClient) -> None: