fix: allow saving test files in UI

vitest-dev · Aug 15, 2023 · 7ff15fb · 7ff15fb
1 parent d729128
commit 7ff15fb
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/packages/ui/client/components/views/ViewEditor.vue b/packages/ui/client/components/views/ViewEditor.vue
@@ -116,7 +116,7 @@ watch([cm, failed], ([cmValue]) => {
 
 async function onSave(content: string) {
   hasBeenEdited.value = true
-  await client.rpc.writeFile(props.file!.filepath, content)
+  await client.rpc.saveTestFile(props.file!.filepath, content)
   serverCode.value = content
   draft.value = false
 }

diff --git a/packages/vitest/src/api/setup.ts b/packages/vitest/src/api/setup.ts
@@ -79,6 +79,12 @@ export function setup(vitestOrWorkspace: Vitest | WorkspaceProject, server?: Vit
             return null
           return fs.readFile(id, 'utf-8')
         },
+        async saveTestFile(id, content) {
+          // can save only already existing test file
+          if (!ctx.state.filesMap.has(id) || !existsSync(id))
+            return
+          return fs.writeFile(id, content, 'utf-8')
+        },
         async saveSnapshotFile(id, content) {
           if (!ctx.snapshot.resolvedPaths.has(id))
             return

diff --git a/packages/vitest/src/api/types.ts b/packages/vitest/src/api/types.ts
@@ -23,6 +23,7 @@ export interface WebSocketHandlers {
   getTransformResult(id: string): Promise<TransformResultWithSource | undefined>
   readSnapshotFile(id: string): Promise<string | null>
   readTestFile(id: string): Promise<string | null>
+  saveTestFile(id: string, content: string): Promise<void>
   saveSnapshotFile(id: string, content: string): Promise<void>
   removeSnapshotFile(id: string): Promise<void>
   snapshotSaved(snapshot: SnapshotResult): void