test: clean up temp files

Author: ymc9

packages/testtools/src/schema.ts

@@ -30,6 +30,7 @@ import prismaPlugin from 'zenstack/plugins/prisma';
 */
 export const FILE_SPLITTER = '#FILE_SPLITTER#';
 
+const tempDirs: string[] = [];
 tmp.setGracefulCleanup();
 
 export type FullDbClientContract = CrudContract & {
@@ -254,6 +255,7 @@ export function createProjectAndCompile(schema: string, options: SchemaLoadOptio
     let projectDir = opt.projectDir;
     if (!projectDir) {
         const r = tmp.dirSync({ unsafeCleanup: true });
+        tempDirs.push(r.name);
         projectDir = r.name;
     }
 
@@ -423,3 +425,17 @@ function loadModule(module: string, basePath: string): any {
     const modulePath = require.resolve(module, { paths: [basePath] });
     return require(modulePath);
 }
+
+export function cleanUpTemps() {
+    tempDirs.forEach((d) => {
+        console.log('Cleaning up temp dir:', d);
+        if (fs.existsSync(d)) {
+            try {
+                fs.rmSync(d, { recursive: true, force: true });
+            } catch {
+                // ignore
+            }
+        }
+    });
+    tempDirs.splice(0, tempDirs.length);
+}

tests/integration/test-setup.ts

@@ -1,3 +1,4 @@
+import { cleanUpTemps } from '@zenstackhq/testtools';
 import {
     toBeRejectedByPolicy,
     toBeNotFound,
@@ -15,3 +16,7 @@ expect.extend({
     toResolveNull,
     toBeRejectedWithCode,
 });
+
+afterAll(() => {
+    cleanUpTemps();
+});

tests/regression/test-setup.ts

@@ -1,3 +1,4 @@
+import { cleanUpTemps } from '@zenstackhq/testtools';
 import {
     toBeNotFound,
     toBeRejectedByPolicy,
@@ -15,3 +16,7 @@ expect.extend({
     toResolveNull,
     toBeRejectedWithCode,
 });
+
+afterAll(() => {
+    cleanUpTemps();
+});