frontend/lib/src/components/widgets/FileUploader/UploadedFile.test.tsx - develop - vshmidt/streamlit

frontend/lib/src/components/widgets/FileUploader/UploadedFile.test.tsx

77 строк2 KB

Ken McGrady

Update copyright header to 2026 (#13491)

02 янв 2026, 16:21

02 янв 2026, 16:21374540a

/**
 * Copyright (c) Streamlit Inc. (2018-2022) Snowflake Inc. (2022-2026)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { screen } from "@testing-library/react"
import { userEvent } from "@testing-library/user-event"

import { render } from "~lib/test_util"

import UploadedFile, { Props, UploadedFileStatus } from "./UploadedFile"
import { FileStatus, UploadFileInfo } from "./UploadFileInfo"

const getProps = (fileStatus: FileStatus): Props => ({
  fileInfo: new UploadFileInfo("filename.txt", 15, 1, fileStatus),
  onDelete: vi.fn(),
  disabled: false,
})

describe("FileStatus widget", () => {
  it("shows progress bar when uploading", () => {
    const props = getProps({
      type: "uploading",
      abortController: new AbortController(),
      progress: 40,
    })
    render(<UploadedFileStatus {...props} />)
    expect(screen.getByRole("progressbar")).toBeInTheDocument()
  })

  it("shows error status", () => {
    const props = getProps({
      type: "error",
      errorMessage: "Everything is terrible",
    })
    render(<UploadedFileStatus {...props} />)
    const errorMessage = screen.getByTestId("stFileUploaderFileErrorMessage")
    expect(errorMessage).toHaveTextContent("Everything is terrible")
  })

  it("show file size when uploaded", () => {
    const props = getProps({
      type: "uploaded",
      fileId: "fileId",
      fileUrls: {},
    })
    render(<UploadedFileStatus {...props} />)
    expect(screen.getByText("15.0B")).toBeInTheDocument()
  })
})

describe("UploadedFile widget", () => {
  it("renders without crashing", async () => {
    const user = userEvent.setup()
    const props = getProps({
      type: "uploaded",
      fileId: "fileId",
      fileUrls: {},
    })
    render(<UploadedFile {...props} />)
    expect(screen.getByTestId("stFileUploaderFile")).toBeInTheDocument()
    const deleteBtn = screen.getByRole("button")
    await user.click(deleteBtn)
    expect(props.onDelete).toHaveBeenCalledWith(1)
  })
})