Handle drives for checkpoints in ContentsManagerMock (#15974)

jupyterlab · Mar 13, 2024 · 946f18c · 946f18c
1 parent 80dd3d6
commit 946f18c
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/packages/services/src/testutils.ts b/packages/services/src/testutils.ts
@@ -347,8 +347,12 @@ export const ContentsManagerMock = jest.fn<Contents.IManager, []>(() => {
       const lastModified = new Date().toISOString();
       const data = { id: UUID.uuid4(), last_modified: lastModified };
       checkpoints.set(path, data);
-      // TODO: handle drives
-      checkPointContent.set(path, files.get('')!.get(path)?.content);
+      const driveName = dummy.driveName(path);
+      const localPath = dummy.localPath(path);
+      checkPointContent.set(
+        path,
+        files.get(driveName)!.get(localPath)?.content
+      );
       return Promise.resolve(data);
     }),
     listCheckpoints: jest.fn(path => {
@@ -369,8 +373,10 @@ export const ContentsManagerMock = jest.fn<Contents.IManager, []>(() => {
       if (!checkpoints.has(path)) {
         return Private.makeResponseError(404);
       }
-      // TODO: handle drives
-      (files.get('')!.get(path) as any).content = checkPointContent.get(path);
+      const driveName = dummy.driveName(path);
+      const localPath = dummy.localPath(path);
+      (files.get(driveName)!.get(localPath) as any).content =
+        checkPointContent.get(path);
       return Promise.resolve();
     }),
     getSharedModelFactory: jest.fn(() => {