Initial open-source release
This commit is contained in:
Will Chen
77
src/__tests__/README.md
Normal file
77
src/__tests__/README.md
Normal file
@@ -0,0 +1,77 @@
|
||||
# Test Documentation
|
||||
|
||||
This directory contains unit tests for the Dyad application.
|
||||
|
||||
## Testing Setup
|
||||
|
||||
We use [Vitest](https://vitest.dev/) as our testing framework, which is designed to work well with Vite and modern JavaScript.
|
||||
|
||||
### Test Commands
|
||||
|
||||
Add these commands to your `package.json`:
|
||||
|
||||
```json
|
||||
"test": "vitest run",
|
||||
"test:watch": "vitest",
|
||||
"test:ui": "vitest --ui"
|
||||
```
|
||||
|
||||
- `npm run test` - Run tests once
|
||||
- `npm run test:watch` - Run tests in watch mode (rerun when files change)
|
||||
- `npm run test:ui` - Run tests with UI reporter
|
||||
|
||||
## Mocking Guidelines
|
||||
|
||||
### Mocking fs module
|
||||
|
||||
When mocking the `node:fs` module, use a default export in the mock:
|
||||
|
||||
```typescript
|
||||
vi.mock('node:fs', async () => {
|
||||
return {
|
||||
default: {
|
||||
mkdirSync: vi.fn(),
|
||||
writeFileSync: vi.fn(),
|
||||
// Add other fs methods as needed
|
||||
}
|
||||
};
|
||||
});
|
||||
```
|
||||
|
||||
### Mocking isomorphic-git
|
||||
|
||||
When mocking isomorphic-git, provide a default export:
|
||||
|
||||
```typescript
|
||||
vi.mock('isomorphic-git', () => ({
|
||||
default: {
|
||||
add: vi.fn().mockResolvedValue(undefined),
|
||||
commit: vi.fn().mockResolvedValue(undefined),
|
||||
// Add other git methods as needed
|
||||
}
|
||||
}));
|
||||
```
|
||||
|
||||
### Testing IPC Handlers
|
||||
|
||||
When testing IPC handlers, mock the Electron IPC system:
|
||||
|
||||
```typescript
|
||||
vi.mock('electron', () => ({
|
||||
ipcMain: {
|
||||
handle: vi.fn(),
|
||||
on: vi.fn(),
|
||||
},
|
||||
}));
|
||||
```
|
||||
|
||||
## Adding New Tests
|
||||
|
||||
1. Create a new file with the `.test.ts` or `.spec.ts` extension
|
||||
2. Import the functions you want to test
|
||||
3. Mock any dependencies using `vi.mock()`
|
||||
4. Write your test cases using `describe()` and `it()`
|
||||
|
||||
## Example
|
||||
|
||||
See `chat_stream_handlers.test.ts` for an example of testing IPC handlers with proper mocking.
|
||||
724
src/__tests__/chat_stream_handlers.test.ts
Normal file
724
src/__tests__/chat_stream_handlers.test.ts
Normal file
@@ -0,0 +1,724 @@
|
||||
import { describe, it, expect, vi, beforeEach } from "vitest";
|
||||
import {
|
||||
getDyadWriteTags,
|
||||
getDyadRenameTags,
|
||||
getDyadDeleteTags,
|
||||
processFullResponseActions,
|
||||
} from "../ipc/processors/response_processor";
|
||||
import fs from "node:fs";
|
||||
import git from "isomorphic-git";
|
||||
import { db } from "../db";
|
||||
|
||||
// Mock fs with default export
|
||||
vi.mock("node:fs", async () => {
|
||||
return {
|
||||
default: {
|
||||
mkdirSync: vi.fn(),
|
||||
writeFileSync: vi.fn(),
|
||||
existsSync: vi.fn(),
|
||||
renameSync: vi.fn(),
|
||||
unlinkSync: vi.fn(),
|
||||
},
|
||||
};
|
||||
});
|
||||
|
||||
// Mock isomorphic-git
|
||||
vi.mock("isomorphic-git", () => ({
|
||||
default: {
|
||||
add: vi.fn().mockResolvedValue(undefined),
|
||||
remove: vi.fn().mockResolvedValue(undefined),
|
||||
commit: vi.fn().mockResolvedValue(undefined),
|
||||
},
|
||||
}));
|
||||
|
||||
// Mock paths module to control getDyadAppPath
|
||||
vi.mock("../paths/paths", () => ({
|
||||
getDyadAppPath: vi.fn().mockImplementation((appPath) => {
|
||||
return `/mock/user/data/path/${appPath}`;
|
||||
}),
|
||||
}));
|
||||
|
||||
// Mock db
|
||||
vi.mock("../db", () => ({
|
||||
db: {
|
||||
query: {
|
||||
chats: {
|
||||
findFirst: vi.fn(),
|
||||
},
|
||||
},
|
||||
},
|
||||
}));
|
||||
|
||||
describe("getDyadWriteTags", () => {
|
||||
it("should return an empty array when no dyad-write tags are found", () => {
|
||||
const result = getDyadWriteTags("No dyad-write tags here");
|
||||
expect(result).toEqual([]);
|
||||
});
|
||||
|
||||
it("should return an array of dyad-write tags", () => {
|
||||
const result = getDyadWriteTags(
|
||||
`I'll create a simple todo list app using React, TypeScript, and shadcn/ui components. Let's get started!
|
||||
|
||||
First, I'll create the necessary files for our todo list application:
|
||||
|
||||
<dyad-write path="src/types/todo.ts" description="Creating types for our todo items">
|
||||
export interface Todo {
|
||||
id: string;
|
||||
text: string;
|
||||
completed: boolean;
|
||||
createdAt: Date;
|
||||
}
|
||||
</dyad-write>
|
||||
|
||||
<dyad-write path="src/components/TodoItem.tsx" description="Creating a component for individual todo items">
|
||||
import React from "react";
|
||||
import { Trash2, Check, X } from "lucide-react";
|
||||
import { Todo } from "../types/todo";
|
||||
import { Button } from "@/components/ui/button";
|
||||
import { cn } from "@/lib/utils";
|
||||
|
||||
interface TodoItemProps {
|
||||
todo: Todo;
|
||||
onToggle: (id: string) => void;
|
||||
onDelete: (id: string) => void;
|
||||
}
|
||||
|
||||
const TodoItem: React.FC<TodoItemProps> = ({ todo, onToggle, onDelete }) => {
|
||||
return (
|
||||
<div className="flex items-center justify-between p-4 border rounded-lg mb-2 bg-white shadow-sm">
|
||||
<div className="flex items-center gap-3 flex-1">
|
||||
<Button
|
||||
variant="outline"
|
||||
size="icon"
|
||||
className={cn(
|
||||
"h-6 w-6 rounded-full",
|
||||
todo.completed && "bg-green-100 text-green-600"
|
||||
)}
|
||||
onClick={() => onToggle(todo.id)}
|
||||
>
|
||||
{todo.completed ? (
|
||||
<Check className="h-4 w-4" />
|
||||
) : (
|
||||
<X className="h-4 w-4 opacity-0" />
|
||||
)}
|
||||
</Button>
|
||||
<span
|
||||
className={cn(
|
||||
"text-sm",
|
||||
todo.completed && "line-through text-gray-500"
|
||||
)}
|
||||
>
|
||||
{todo.text}
|
||||
</span>
|
||||
</div>
|
||||
<Button
|
||||
variant="ghost"
|
||||
size="icon"
|
||||
className="h-8 w-8 text-red-500 hover:text-red-700 hover:bg-red-50"
|
||||
onClick={() => onDelete(todo.id)}
|
||||
>
|
||||
<Trash2 className="h-4 w-4" />
|
||||
</Button>
|
||||
</div>
|
||||
);
|
||||
};
|
||||
|
||||
export default TodoItem;
|
||||
</dyad-write>
|
||||
|
||||
<dyad-write path="src/components/TodoForm.tsx" description="Creating a form component for adding new todos">
|
||||
import React, { useState } from "react";
|
||||
import { Button } from "@/components/ui/button";
|
||||
import { Input } from "@/components/ui/input";
|
||||
import { PlusCircle } from "lucide-react";
|
||||
|
||||
interface TodoFormProps {
|
||||
onAddTodo: (text: string) => void;
|
||||
}
|
||||
|
||||
const TodoForm: React.FC<TodoFormProps> = ({ onAddTodo }) => {
|
||||
const [text, setText] = useState("");
|
||||
|
||||
const handleSubmit = (e: React.FormEvent) => {
|
||||
e.preventDefault();
|
||||
if (text.trim()) {
|
||||
onAddTodo(text.trim());
|
||||
setText("");
|
||||
}
|
||||
};
|
||||
|
||||
return (
|
||||
<form onSubmit={handleSubmit} className="flex gap-2 mb-6">
|
||||
<Input
|
||||
type="text"
|
||||
value={text}
|
||||
onChange={(e) => setText(e.target.value)}
|
||||
placeholder="Add a new task..."
|
||||
className="flex-1"
|
||||
/>
|
||||
<Button type="submit" disabled={!text.trim()}>
|
||||
<PlusCircle className="h-4 w-4 mr-2" />
|
||||
Add
|
||||
</Button>
|
||||
</form>
|
||||
);
|
||||
};
|
||||
|
||||
export default TodoForm;
|
||||
</dyad-write>
|
||||
|
||||
<dyad-write path="src/components/TodoList.tsx" description="Creating the main todo list component">
|
||||
import React from "react";
|
||||
import { Todo } from "../types/todo";
|
||||
import TodoItem from "./TodoItem";
|
||||
import { ScrollArea } from "@/components/ui/scroll-area";
|
||||
|
||||
interface TodoListProps {
|
||||
todos: Todo[];
|
||||
onToggle: (id: string) => void;
|
||||
onDelete: (id: string) => void;
|
||||
}
|
||||
|
||||
const TodoList: React.FC<TodoListProps> = ({ todos, onToggle, onDelete }) => {
|
||||
if (todos.length === 0) {
|
||||
return (
|
||||
<div className="text-center py-8 text-gray-500">
|
||||
No tasks yet. Add one above!
|
||||
</div>
|
||||
);
|
||||
}
|
||||
|
||||
return (
|
||||
<ScrollArea className="h-[400px] rounded-md border p-4">
|
||||
<div className="space-y-2">
|
||||
{todos.map((todo) => (
|
||||
<TodoItem
|
||||
key={todo.id}
|
||||
todo={todo}
|
||||
onToggle={onToggle}
|
||||
onDelete={onDelete}
|
||||
/>
|
||||
))}
|
||||
</div>
|
||||
</ScrollArea>
|
||||
);
|
||||
};
|
||||
|
||||
export default TodoList;
|
||||
</dyad-write>
|
||||
|
||||
<dyad-write path="src/components/TodoStats.tsx" description="Creating a component to show todo statistics">
|
||||
import React from "react";
|
||||
import { Todo } from "../types/todo";
|
||||
import { Card, CardContent } from "@/components/ui/card";
|
||||
|
||||
interface TodoStatsProps {
|
||||
todos: Todo[];
|
||||
}
|
||||
|
||||
const TodoStats: React.FC<TodoStatsProps> = ({ todos }) => {
|
||||
const total = todos.length;
|
||||
const completed = todos.filter((todo) => todo.completed).length;
|
||||
const pending = total - completed;
|
||||
const percentComplete = total > 0 ? Math.round((completed / total) * 100) : 0;
|
||||
|
||||
return (
|
||||
<div className="grid grid-cols-1 md:grid-cols-3 gap-4 mb-6">
|
||||
<Card>
|
||||
<CardContent className="pt-6">
|
||||
<div className="text-center">
|
||||
<p className="text-sm font-medium text-gray-500">Total Tasks</p>
|
||||
<p className="text-3xl font-bold">{total}</p>
|
||||
</div>
|
||||
</CardContent>
|
||||
</Card>
|
||||
|
||||
<Card>
|
||||
<CardContent className="pt-6">
|
||||
<div className="text-center">
|
||||
<p className="text-sm font-medium text-gray-500">Completed</p>
|
||||
<p className="text-3xl font-bold text-green-600">{completed}</p>
|
||||
</div>
|
||||
</CardContent>
|
||||
</Card>
|
||||
|
||||
<Card>
|
||||
<CardContent className="pt-6">
|
||||
<div className="text-center">
|
||||
<p className="text-sm font-medium text-gray-500">Pending</p>
|
||||
<p className="text-3xl font-bold text-amber-600">{pending}</p>
|
||||
</div>
|
||||
</CardContent>
|
||||
</Card>
|
||||
|
||||
<Card className="md:col-span-3">
|
||||
<CardContent className="pt-6">
|
||||
<div className="space-y-2">
|
||||
<div className="flex justify-between text-sm">
|
||||
<span>Progress</span>
|
||||
<span>{percentComplete}%</span>
|
||||
</div>
|
||||
<div className="h-2 bg-gray-200 rounded-full overflow-hidden">
|
||||
<div
|
||||
className="h-full bg-green-500 transition-all duration-300"
|
||||
/>
|
||||
</div>
|
||||
</div>
|
||||
</CardContent>
|
||||
</Card>
|
||||
</div>
|
||||
);
|
||||
};
|
||||
|
||||
export default TodoStats;
|
||||
</dyad-write>
|
||||
|
||||
<dyad-write path="src/pages/Index.tsx" description="Creating the main page for our todo app">
|
||||
import React, { useState, useEffect } from "react";
|
||||
import { v4 as uuidv4 } from "uuid";
|
||||
import { Todo } from "../types/todo";
|
||||
import TodoForm from "../components/TodoForm";
|
||||
import TodoList from "../components/TodoList";
|
||||
import TodoStats from "../components/TodoStats";
|
||||
import { Toaster } from "@/components/ui/toaster";
|
||||
import { useToast } from "@/components/ui/use-toast";
|
||||
|
||||
const Index: React.FC = () => {
|
||||
const [todos, setTodos] = useState<Todo[]>(() => {
|
||||
const savedTodos = localStorage.getItem("todos");
|
||||
if (savedTodos) {
|
||||
try {
|
||||
// Parse the saved todos and convert string dates back to Date objects
|
||||
return JSON.parse(savedTodos).map((todo: any) => ({
|
||||
...todo,
|
||||
createdAt: new Date(todo.createdAt)
|
||||
}));
|
||||
} catch (e) {
|
||||
console.error("Failed to parse saved todos", e);
|
||||
return [];
|
||||
}
|
||||
}
|
||||
return [];
|
||||
});
|
||||
|
||||
const { toast } = useToast();
|
||||
|
||||
// Save todos to localStorage whenever they change
|
||||
useEffect(() => {
|
||||
localStorage.setItem("todos", JSON.stringify(todos));
|
||||
}, [todos]);
|
||||
|
||||
const handleAddTodo = (text: string) => {
|
||||
const newTodo: Todo = {
|
||||
id: uuidv4(),
|
||||
text,
|
||||
completed: false,
|
||||
createdAt: new Date()
|
||||
};
|
||||
|
||||
setTodos([...todos, newTodo]);
|
||||
toast({
|
||||
title: "Task added",
|
||||
description: "Your new task has been added successfully.",
|
||||
});
|
||||
};
|
||||
|
||||
const handleToggleTodo = (id: string) => {
|
||||
setTodos(
|
||||
todos.map((todo) =>
|
||||
todo.id === id ? { ...todo, completed: !todo.completed } : todo
|
||||
)
|
||||
);
|
||||
|
||||
const todo = todos.find(t => t.id === id);
|
||||
if (todo) {
|
||||
toast({
|
||||
title: todo.completed ? "Task marked as incomplete" : "Task completed",
|
||||
description: todo.text,
|
||||
});
|
||||
}
|
||||
};
|
||||
|
||||
const handleDeleteTodo = (id: string) => {
|
||||
const todoToDelete = todos.find(todo => todo.id === id);
|
||||
setTodos(todos.filter((todo) => todo.id !== id));
|
||||
|
||||
if (todoToDelete) {
|
||||
toast({
|
||||
title: "Task deleted",
|
||||
description: todoToDelete.text,
|
||||
variant: "destructive",
|
||||
});
|
||||
}
|
||||
};
|
||||
|
||||
return (
|
||||
<div className="container mx-auto px-4 py-8 max-w-3xl">
|
||||
<h1 className="text-3xl font-bold mb-2 text-center">Todo List</h1>
|
||||
<p className="text-gray-500 mb-8 text-center">
|
||||
Keep track of your tasks and stay organized
|
||||
</p>
|
||||
|
||||
<TodoStats todos={todos} />
|
||||
<TodoForm onAddTodo={handleAddTodo} />
|
||||
<TodoList
|
||||
todos={todos}
|
||||
onToggle={handleToggleTodo}
|
||||
onDelete={handleDeleteTodo}
|
||||
/>
|
||||
<Toaster />
|
||||
</div>
|
||||
);
|
||||
};
|
||||
|
||||
export default Index;
|
||||
</dyad-write>
|
||||
|
||||
<dyad-add-dependency package="uuid"></dyad-add-dependency>
|
||||
|
||||
<dyad-write path="src/types/uuid.d.ts" description="Adding type definitions for uuid">
|
||||
declare module 'uuid' {
|
||||
export function v4(): string;
|
||||
}
|
||||
</dyad-write>
|
||||
|
||||
I've created a complete todo list application with the ability to add, complete, and delete tasks. The app includes statistics and uses local storage to persist data.`
|
||||
);
|
||||
expect(result.length).toEqual(7);
|
||||
});
|
||||
});
|
||||
|
||||
describe("getDyadRenameTags", () => {
|
||||
it("should return an empty array when no dyad-rename tags are found", () => {
|
||||
const result = getDyadRenameTags("No dyad-rename tags here");
|
||||
expect(result).toEqual([]);
|
||||
});
|
||||
|
||||
it("should return an array of dyad-rename tags", () => {
|
||||
const result = getDyadRenameTags(
|
||||
`<dyad-rename from="src/components/UserProfile.jsx" to="src/components/ProfileCard.jsx" />
|
||||
<dyad-rename from="src/utils/helpers.js" to="src/utils/utils.js" />`
|
||||
);
|
||||
expect(result).toEqual([
|
||||
{
|
||||
from: "src/components/UserProfile.jsx",
|
||||
to: "src/components/ProfileCard.jsx",
|
||||
},
|
||||
{ from: "src/utils/helpers.js", to: "src/utils/utils.js" },
|
||||
]);
|
||||
});
|
||||
});
|
||||
|
||||
describe("getDyadDeleteTags", () => {
|
||||
it("should return an empty array when no dyad-delete tags are found", () => {
|
||||
const result = getDyadDeleteTags("No dyad-delete tags here");
|
||||
expect(result).toEqual([]);
|
||||
});
|
||||
|
||||
it("should return an array of dyad-delete paths", () => {
|
||||
const result = getDyadDeleteTags(
|
||||
`<dyad-delete path="src/components/Analytics.jsx" />
|
||||
<dyad-delete path="src/utils/unused.js" />`
|
||||
);
|
||||
expect(result).toEqual([
|
||||
"src/components/Analytics.jsx",
|
||||
"src/utils/unused.js",
|
||||
]);
|
||||
});
|
||||
});
|
||||
|
||||
describe("processFullResponse", () => {
|
||||
beforeEach(() => {
|
||||
vi.clearAllMocks();
|
||||
|
||||
// Mock db query response
|
||||
vi.mocked(db.query.chats.findFirst).mockResolvedValue({
|
||||
id: 1,
|
||||
appId: 1,
|
||||
title: "Test Chat",
|
||||
createdAt: new Date(),
|
||||
app: {
|
||||
id: 1,
|
||||
name: "Mock App",
|
||||
path: "mock-app-path",
|
||||
createdAt: new Date(),
|
||||
updatedAt: new Date(),
|
||||
},
|
||||
messages: [],
|
||||
} as any);
|
||||
|
||||
// Default mock for existsSync to return true
|
||||
vi.mocked(fs.existsSync).mockReturnValue(true);
|
||||
});
|
||||
|
||||
it("should return empty object when no dyad-write tags are found", async () => {
|
||||
const result = await processFullResponseActions(
|
||||
"No dyad-write tags here",
|
||||
1,
|
||||
{
|
||||
chatSummary: undefined,
|
||||
}
|
||||
);
|
||||
expect(result).toEqual({});
|
||||
expect(fs.mkdirSync).not.toHaveBeenCalled();
|
||||
expect(fs.writeFileSync).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it("should process dyad-write tags and create files", async () => {
|
||||
// Set up fs mocks to succeed
|
||||
vi.mocked(fs.mkdirSync).mockImplementation(() => undefined);
|
||||
vi.mocked(fs.writeFileSync).mockImplementation(() => undefined);
|
||||
|
||||
const response = `<dyad-write path="src/file1.js">console.log('Hello');</dyad-write>`;
|
||||
|
||||
const result = await processFullResponseActions(response, 1, {
|
||||
chatSummary: undefined,
|
||||
});
|
||||
|
||||
expect(fs.mkdirSync).toHaveBeenCalledWith(
|
||||
"/mock/user/data/path/mock-app-path/src",
|
||||
{ recursive: true }
|
||||
);
|
||||
expect(fs.writeFileSync).toHaveBeenCalledWith(
|
||||
"/mock/user/data/path/mock-app-path/src/file1.js",
|
||||
"console.log('Hello');"
|
||||
);
|
||||
expect(git.add).toHaveBeenCalledWith(
|
||||
expect.objectContaining({
|
||||
filepath: "src/file1.js",
|
||||
})
|
||||
);
|
||||
expect(git.commit).toHaveBeenCalled();
|
||||
expect(result).toEqual({ updatedFiles: true });
|
||||
});
|
||||
|
||||
it("should handle file system errors gracefully", async () => {
|
||||
// Set up the mock to throw an error on mkdirSync
|
||||
vi.mocked(fs.mkdirSync).mockImplementationOnce(() => {
|
||||
throw new Error("Mock filesystem error");
|
||||
});
|
||||
|
||||
const response = `<dyad-write path="src/error-file.js">This will fail</dyad-write>`;
|
||||
|
||||
const result = await processFullResponseActions(response, 1, {
|
||||
chatSummary: undefined,
|
||||
});
|
||||
|
||||
expect(result).toHaveProperty("error");
|
||||
expect(result.error).toContain("Mock filesystem error");
|
||||
});
|
||||
|
||||
it("should process multiple dyad-write tags and commit all files", async () => {
|
||||
// Clear previous mock calls
|
||||
vi.clearAllMocks();
|
||||
|
||||
// Set up fs mocks to succeed
|
||||
vi.mocked(fs.mkdirSync).mockImplementation(() => undefined);
|
||||
vi.mocked(fs.writeFileSync).mockImplementation(() => undefined);
|
||||
|
||||
const response = `
|
||||
<dyad-write path="src/file1.js">console.log('First file');</dyad-write>
|
||||
<dyad-write path="src/utils/file2.js">export const add = (a, b) => a + b;</dyad-write>
|
||||
<dyad-write path="src/components/Button.tsx">
|
||||
import React from 'react';
|
||||
export const Button = ({ children }) => <button>{children}</button>;
|
||||
</dyad-write>
|
||||
`;
|
||||
|
||||
const result = await processFullResponseActions(response, 1, {
|
||||
chatSummary: undefined,
|
||||
});
|
||||
|
||||
// Check that directories were created for each file path
|
||||
expect(fs.mkdirSync).toHaveBeenCalledWith(
|
||||
"/mock/user/data/path/mock-app-path/src",
|
||||
{ recursive: true }
|
||||
);
|
||||
expect(fs.mkdirSync).toHaveBeenCalledWith(
|
||||
"/mock/user/data/path/mock-app-path/src/utils",
|
||||
{ recursive: true }
|
||||
);
|
||||
expect(fs.mkdirSync).toHaveBeenCalledWith(
|
||||
"/mock/user/data/path/mock-app-path/src/components",
|
||||
{ recursive: true }
|
||||
);
|
||||
|
||||
// Using toHaveBeenNthCalledWith to check each specific call
|
||||
expect(fs.writeFileSync).toHaveBeenNthCalledWith(
|
||||
1,
|
||||
"/mock/user/data/path/mock-app-path/src/file1.js",
|
||||
"console.log('First file');"
|
||||
);
|
||||
expect(fs.writeFileSync).toHaveBeenNthCalledWith(
|
||||
2,
|
||||
"/mock/user/data/path/mock-app-path/src/utils/file2.js",
|
||||
"export const add = (a, b) => a + b;"
|
||||
);
|
||||
expect(fs.writeFileSync).toHaveBeenNthCalledWith(
|
||||
3,
|
||||
"/mock/user/data/path/mock-app-path/src/components/Button.tsx",
|
||||
"\n import React from 'react';\n export const Button = ({ children }) => <button>{children}</button>;\n "
|
||||
);
|
||||
|
||||
// Verify git operations were called for each file
|
||||
expect(git.add).toHaveBeenCalledWith(
|
||||
expect.objectContaining({
|
||||
filepath: "src/file1.js",
|
||||
})
|
||||
);
|
||||
expect(git.add).toHaveBeenCalledWith(
|
||||
expect.objectContaining({
|
||||
filepath: "src/utils/file2.js",
|
||||
})
|
||||
);
|
||||
expect(git.add).toHaveBeenCalledWith(
|
||||
expect.objectContaining({
|
||||
filepath: "src/components/Button.tsx",
|
||||
})
|
||||
);
|
||||
|
||||
// Verify commit was called once after all files were added
|
||||
expect(git.commit).toHaveBeenCalledTimes(1);
|
||||
expect(result).toEqual({ updatedFiles: true });
|
||||
});
|
||||
|
||||
it("should process dyad-rename tags and rename files", async () => {
|
||||
// Set up fs mocks to succeed
|
||||
vi.mocked(fs.existsSync).mockReturnValue(true);
|
||||
vi.mocked(fs.mkdirSync).mockImplementation(() => undefined);
|
||||
vi.mocked(fs.renameSync).mockImplementation(() => undefined);
|
||||
|
||||
const response = `<dyad-rename from="src/components/OldComponent.jsx" to="src/components/NewComponent.jsx" />`;
|
||||
|
||||
const result = await processFullResponseActions(response, 1, {
|
||||
chatSummary: undefined,
|
||||
});
|
||||
|
||||
expect(fs.mkdirSync).toHaveBeenCalledWith(
|
||||
"/mock/user/data/path/mock-app-path/src/components",
|
||||
{ recursive: true }
|
||||
);
|
||||
expect(fs.renameSync).toHaveBeenCalledWith(
|
||||
"/mock/user/data/path/mock-app-path/src/components/OldComponent.jsx",
|
||||
"/mock/user/data/path/mock-app-path/src/components/NewComponent.jsx"
|
||||
);
|
||||
expect(git.add).toHaveBeenCalledWith(
|
||||
expect.objectContaining({
|
||||
filepath: "src/components/NewComponent.jsx",
|
||||
})
|
||||
);
|
||||
expect(git.remove).toHaveBeenCalledWith(
|
||||
expect.objectContaining({
|
||||
filepath: "src/components/OldComponent.jsx",
|
||||
})
|
||||
);
|
||||
expect(git.commit).toHaveBeenCalled();
|
||||
expect(result).toEqual({ updatedFiles: true });
|
||||
});
|
||||
|
||||
it("should handle non-existent files during rename gracefully", async () => {
|
||||
// Set up the mock to return false for existsSync
|
||||
vi.mocked(fs.existsSync).mockReturnValue(false);
|
||||
|
||||
const response = `<dyad-rename from="src/components/NonExistent.jsx" to="src/components/NewFile.jsx" />`;
|
||||
|
||||
const result = await processFullResponseActions(response, 1, {
|
||||
chatSummary: undefined,
|
||||
});
|
||||
|
||||
expect(fs.mkdirSync).toHaveBeenCalled();
|
||||
expect(fs.renameSync).not.toHaveBeenCalled();
|
||||
expect(git.commit).not.toHaveBeenCalled();
|
||||
expect(result).toEqual({});
|
||||
});
|
||||
|
||||
it("should process dyad-delete tags and delete files", async () => {
|
||||
// Set up fs mocks to succeed
|
||||
vi.mocked(fs.existsSync).mockReturnValue(true);
|
||||
vi.mocked(fs.unlinkSync).mockImplementation(() => undefined);
|
||||
|
||||
const response = `<dyad-delete path="src/components/Unused.jsx" />`;
|
||||
|
||||
const result = await processFullResponseActions(response, 1, {
|
||||
chatSummary: undefined,
|
||||
});
|
||||
|
||||
expect(fs.unlinkSync).toHaveBeenCalledWith(
|
||||
"/mock/user/data/path/mock-app-path/src/components/Unused.jsx"
|
||||
);
|
||||
expect(git.remove).toHaveBeenCalledWith(
|
||||
expect.objectContaining({
|
||||
filepath: "src/components/Unused.jsx",
|
||||
})
|
||||
);
|
||||
expect(git.commit).toHaveBeenCalled();
|
||||
expect(result).toEqual({ updatedFiles: true });
|
||||
});
|
||||
|
||||
it("should handle non-existent files during delete gracefully", async () => {
|
||||
// Set up the mock to return false for existsSync
|
||||
vi.mocked(fs.existsSync).mockReturnValue(false);
|
||||
|
||||
const response = `<dyad-delete path="src/components/NonExistent.jsx" />`;
|
||||
|
||||
const result = await processFullResponseActions(response, 1, {
|
||||
chatSummary: undefined,
|
||||
});
|
||||
|
||||
expect(fs.unlinkSync).not.toHaveBeenCalled();
|
||||
expect(git.remove).not.toHaveBeenCalled();
|
||||
expect(git.commit).not.toHaveBeenCalled();
|
||||
expect(result).toEqual({});
|
||||
});
|
||||
|
||||
it("should process mixed operations (write, rename, delete) in one response", async () => {
|
||||
// Set up fs mocks to succeed
|
||||
vi.mocked(fs.existsSync).mockReturnValue(true);
|
||||
vi.mocked(fs.mkdirSync).mockImplementation(() => undefined);
|
||||
vi.mocked(fs.writeFileSync).mockImplementation(() => undefined);
|
||||
vi.mocked(fs.renameSync).mockImplementation(() => undefined);
|
||||
vi.mocked(fs.unlinkSync).mockImplementation(() => undefined);
|
||||
|
||||
const response = `
|
||||
<dyad-write path="src/components/NewComponent.jsx">import React from 'react'; export default () => <div>New</div>;</dyad-write>
|
||||
<dyad-rename from="src/components/OldComponent.jsx" to="src/components/RenamedComponent.jsx" />
|
||||
<dyad-delete path="src/components/Unused.jsx" />
|
||||
`;
|
||||
|
||||
const result = await processFullResponseActions(response, 1, {
|
||||
chatSummary: undefined,
|
||||
});
|
||||
|
||||
// Check write operation happened
|
||||
expect(fs.writeFileSync).toHaveBeenCalledWith(
|
||||
"/mock/user/data/path/mock-app-path/src/components/NewComponent.jsx",
|
||||
"import React from 'react'; export default () => <div>New</div>;"
|
||||
);
|
||||
|
||||
// Check rename operation happened
|
||||
expect(fs.renameSync).toHaveBeenCalledWith(
|
||||
"/mock/user/data/path/mock-app-path/src/components/OldComponent.jsx",
|
||||
"/mock/user/data/path/mock-app-path/src/components/RenamedComponent.jsx"
|
||||
);
|
||||
|
||||
// Check delete operation happened
|
||||
expect(fs.unlinkSync).toHaveBeenCalledWith(
|
||||
"/mock/user/data/path/mock-app-path/src/components/Unused.jsx"
|
||||
);
|
||||
|
||||
// Check git operations
|
||||
expect(git.add).toHaveBeenCalledTimes(2); // For the write and rename
|
||||
expect(git.remove).toHaveBeenCalledTimes(2); // For the rename and delete
|
||||
|
||||
// Check the commit message includes all operations
|
||||
expect(git.commit).toHaveBeenCalledWith(
|
||||
expect.objectContaining({
|
||||
message: expect.stringContaining(
|
||||
"wrote 1 file(s), renamed 1 file(s), deleted 1 file(s)"
|
||||
),
|
||||
})
|
||||
);
|
||||
|
||||
expect(result).toEqual({ updatedFiles: true });
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user