Run js tests cleanup using onTestFinished

e2b-dev · Dec 3, 2024 · 666b3d4 · 666b3d4
1 parent 84e1307
commit 666b3d4
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/packages/js-sdk/tests/sandbox/files/watch.test.ts b/packages/js-sdk/tests/sandbox/files/watch.test.ts
@@ -1,4 +1,4 @@
-import { expect } from 'vitest'
+import { expect, onTestFinished } from 'vitest'
 
 import { sandboxTest } from '../../setup.js'
 import { FilesystemEventType, NotFoundError, SandboxError } from '../../../src'
@@ -38,8 +38,9 @@ sandboxTest('watch recursive directory changes', async ({ sandbox }) => {
   const content = 'This file will be watched.'
   const newContent = 'This file has been modified.'
 
-  await sandbox.files.remove(dirname)
   await sandbox.files.makeDir(`${dirname}/${nestedDirname}`)
+  onTestFinished(() => sandbox.files.remove(dirname))
+
   await sandbox.files.write(`${dirname}/${nestedDirname}/${filename}`, content)
 
   let trigger: () => void
@@ -70,8 +71,8 @@ sandboxTest('watch recursive directory after nested folder addition', async ({ s
   const filename = 'test_watch.txt'
   const content = 'This file will be watched.'
 
-  await sandbox.files.remove(dirname)
   await sandbox.files.makeDir(dirname)
+  onTestFinished(() => sandbox.files.remove(dirname))
 
   let triggerFile: () => void
   let triggerFolder: () => void