Function test_upload_file_without_annotation

tests/units/test_app.py:1403–1438

Test that an error is thrown when there's no param annotated with rx.UploadFile or list[UploadFile]. 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_without_annotation(
    state: FileUploadState | ChildFileUploadState | GrandChildFileUploadState,
    tmp_path: Path,
    token: str,
):
    """Test that an error is thrown when there's no param annotated with rx.UploadFile or list[UploadFile].

    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()}.handle_upload2",
    }

    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(ValueError) as err:
        await fn(request_mock)
    assert (
        err.value.args[0]
        == f"`{state.get_full_name()}.handle_upload2` handler should have a parameter annotated as list[rx.UploadFile]"
    )

    await app.state_manager.close()


@pytest.mark.asyncio