|
1 | 1 | /* Copyright 2024 Marimo. All rights reserved. */ |
2 | 2 | import { afterEach, beforeEach, describe, expect, test, vi } from "vitest"; |
3 | 3 | import { toast } from "@/components/ui/use-toast"; |
| 4 | +import type { FilePath } from "@/utils/paths"; |
4 | 5 | import { RequestingTree } from "../requesting-tree"; |
5 | 6 |
|
6 | 7 | const sendListFiles = vi.fn(); |
@@ -261,4 +262,60 @@ describe("RequestingTree", () => { |
261 | 262 | expect(mockOnChange).toHaveBeenCalled(); |
262 | 263 | }); |
263 | 264 | }); |
| 265 | + |
| 266 | + describe("relativeFromRoot", () => { |
| 267 | + test("should return relative path for Unix paths", async () => { |
| 268 | + const tree = new RequestingTree({ |
| 269 | + listFiles: sendListFiles, |
| 270 | + createFileOrFolder: sendCreateFileOrFolder, |
| 271 | + deleteFileOrFolder: sendDeleteFileOrFolder, |
| 272 | + renameFileOrFolder: sendRenameFileOrFolder, |
| 273 | + }); |
| 274 | + |
| 275 | + sendListFiles.mockResolvedValue({ |
| 276 | + files: [], |
| 277 | + root: "/home/user/project", |
| 278 | + }); |
| 279 | + |
| 280 | + await tree.initialize(vi.fn()); |
| 281 | + |
| 282 | + const relativePath = tree.relativeFromRoot( |
| 283 | + "/home/user/project/src/file.py" as FilePath, |
| 284 | + ); |
| 285 | + expect(relativePath).toBe("src/file.py"); |
| 286 | + }); |
| 287 | + |
| 288 | + test("should return relative path for Windows paths", async () => { |
| 289 | + const tree = new RequestingTree({ |
| 290 | + listFiles: sendListFiles, |
| 291 | + createFileOrFolder: sendCreateFileOrFolder, |
| 292 | + deleteFileOrFolder: sendDeleteFileOrFolder, |
| 293 | + renameFileOrFolder: sendRenameFileOrFolder, |
| 294 | + }); |
| 295 | + |
| 296 | + sendListFiles.mockResolvedValue({ |
| 297 | + files: [], |
| 298 | + root: "C:\\Users\\test\\project", |
| 299 | + }); |
| 300 | + |
| 301 | + await tree.initialize(vi.fn()); |
| 302 | + |
| 303 | + const relativePath = tree.relativeFromRoot( |
| 304 | + "C:\\Users\\test\\project\\src\\file.py" as FilePath, |
| 305 | + ); |
| 306 | + expect(relativePath).toBe("src\\file.py"); |
| 307 | + }); |
| 308 | + |
| 309 | + test("should return original path when not under root", async () => { |
| 310 | + const relativePath = requestingTree.relativeFromRoot( |
| 311 | + "/other/path/file.py" as FilePath, |
| 312 | + ); |
| 313 | + expect(relativePath).toBe("/other/path/file.py"); |
| 314 | + }); |
| 315 | + |
| 316 | + test("should handle root path exactly", async () => { |
| 317 | + const relativePath = requestingTree.relativeFromRoot("/root" as FilePath); |
| 318 | + expect(relativePath).toBe("/root"); |
| 319 | + }); |
| 320 | + }); |
264 | 321 | }); |
0 commit comments