-
-
-
- );
- }),
+import { DatasetInfo } from "@/actions";
+import { Active, DragEndEvent, Over, UniqueIdentifier } from "@dnd-kit/core";
+import { act, fireEvent, render, screen } from "@testing-library/react"; // Removed waitFor
+import { ReactNode } from "react";
+import {
+ afterEach,
+ beforeEach,
+ describe,
+ expect,
+ Mock,
+ test,
+ vi,
+} from "vitest";
+import { CreateDatasetDialog, CreateDatasetDialogProps } from "./dataset";
+
+vi.mock("@/actions", () => ({}));
+vi.mock("@/urls", () => ({
+ imageUrl: (path: string) => `mock://${path}`,
}));
-const mockOnCreate = vi.fn();
-const mockOnUpdate = vi.fn();
-
-describe("CreateDatasetDialog", () => {
- beforeEach(async () => {
- vi.mock("react-router-dom");
- vi.mocked(useNavigate).mockReturnValue(vi.fn());
- render(
- Dataset Name: {props.datasetName}
-Dataset Description: {props.datasetDescription}
-{children}
;
+ },
+ useSensor: vi.fn(),
+ useSensors: vi.fn(),
+ PointerSensor: vi.fn(),
+ KeyboardSensor: vi.fn(),
+ closestCenter: vi.fn(),
+ };
+});
- const createButton = screen.getByRole("button", { name: "Create" });
- await userEvent.click(createButton);
+vi.mock("@dnd-kit/sortable", async () => {
+ const actual = await vi.importActual("@dnd-kit/sortable");
+ return {
+ ...actual,
+ SortableContext: ({ children }: { children: ReactNode }) => (
+ {children}
+ ),
+ useSortable: ({ id }: { id: string }) => ({
+ attributes: {
+ role: "button",
+ "aria-roledescription": "sortable",
+ "data-sortable-id": id,
+ },
+ listeners: { onMouseDown: vi.fn(), onKeyDown: vi.fn() },
+ setNodeRef: vi.fn(),
+ transform: null,
+ transition: null,
+ isDragging: false,
+ }),
+ arrayMove: vi.fn((arr, from, to) => {
+ const newArray = [...arr];
+ const element = newArray.splice(from, 1)[0];
+ newArray.splice(to, 0, element);
+ return newArray;
+ }),
+ verticalListSortingStrategy: vi.fn(),
+ sortableKeyboardCoordinates: vi.fn(),
+ };
+});
- expect(mockOnCreate).toHaveBeenCalledWith({
- name: "test-list-dataset",
- description: "This is a test list dataset.",
- type: "list",
- options: ["Option 1", "Option 2", "Option 3"],
+const mockFile = (
+ name: string,
+ type: string,
+ content: string = "",
+ size?: number,
+): File => {
+ const blob = new Blob([content], { type });
+ const file = new File([blob], name, { type, lastModified: Date.now() });
+ if (size !== undefined) {
+ Object.defineProperty(file, "size", {
+ value: size,
+ writable: false,
+ configurable: true,
});
- });
+ }
+ return file;
+};
- it("should call onCreate with correct data for csv type dataset", async () => {
- const nameInput = screen.getByLabelText("Name");
- await userEvent.type(nameInput, "test-csv-dataset");
+describe("CreateDatasetDialog Management", () => {
+ let mockOnClose: Mock<() => void>;
+ let mockOnCreate: Mock
+
+ );
+}
+
+type DatasetType = "list" | "csv" | "image";
export function CreateDatasetDialog({
dataset,
@@ -60,7 +152,7 @@ export function CreateDatasetDialog({
const [description, setDescription] = useState("");
const [type, setType] = useState
+
+ {item.thumbnail && datasetType === "image" && (
+ 
+ )}
+
+ {item.name}
+ {item.file
+ ? `(${(item.file.size / 1024).toFixed(2)} KB)`
+ : "(existing)"}
+
+
+
+
+
@@ -332,38 +629,90 @@ export function CreateDatasetDialog({
{filesError && (
{filesError}
)} - {selectedFiles.length > 0 && ( -
- {selectedFiles.map((file) => (
-
)}
+
+ {type === "image" && (
+
-
- {file.name} ({(file.size / 1024).toFixed(2)} KB)
-
-
+ {fileItems.length > 0 && (
+
+ item.id)}
+ strategy={verticalListSortingStrategy}
+ >
+
+
-
+
+
+
)}
+ {fileItems.map((item) => (
+
- ))}
- - {dataset - ? "Select one or more CSV files to REPLACE original data or leave it empty if you don't wan t to change." + {dataset && type === "csv" + ? "Add new CSV files to replace existing ones. Leave empty to keep current files. You can reorder files by dragging." : "Select one or more CSV files."}
+
+
+ )}
+
+ {filesError && (
+
+ item.id)}
+ strategy={verticalListSortingStrategy}
+ >
+
+
+
+
+ )}
+
+ {filesError}
+ )} + {fileItems.length > 0 && ( +
+ {fileItems.map((item) => (
+
+ + {dataset && type === "image" + ? "Add new image files to replace existing ones. Leave empty to keep current files. You can reorder files by dragging." + : "Select one or more image files (PNG, JPG, GIF)."} +
+