Function test_upload_file_background

tests/units/test_app.py:1446–1481

Test that an error is thrown handler is a background task. Args: state: The state class. tmp_path: Temporary path. token: a Token.

(
    state: FileUploadState | ChildFileUploadState | GrandChildFileUploadState,
    tmp_path: Path,
    token: str,
)

    [FileUploadState, ChildFileUploadState, GrandChildFileUploadState],
)
async def test_upload_file_background(
    state: FileUploadState | ChildFileUploadState | GrandChildFileUploadState,
    tmp_path: Path,
    token: str,
):
    """Test that an error is thrown handler is a background task.

    Args:
        state: The state class.
        tmp_path: Temporary path.
        token: a Token.
    """
    app = App(_state=State)

    request_mock = unittest.mock.Mock()
    request_mock.headers = {
        "reflex-client-token": token,
        "reflex-event-handler": f"{state.get_full_name()}.bg_upload",
    }

    file1 = UploadFile(filename="image1.jpg", file=io.BytesIO(b"data"))

    async def form():  # noqa: RUF029
        return FormData([("files", file1)])

    request_mock.form = form

    fn = upload(app)
    with pytest.raises(TypeError) as err:
        await fn(request_mock)
    assert (
        err.value.args[0]
        == f"@rx.event(background=True) is not supported for upload handler `{state.get_full_name()}.bg_upload`."
    )

    await app.state_manager.close()


def _build_chunk_upload_multipart_body(