Migrate from Jest to Vitest (#94)

grafana · Sep 11, 2024 · 44ade6f · 44ade6f
1 parent 2fcf8f1
commit 44ade6f
Show file tree

Hide file tree

Showing 8 changed files with 761 additions and 2,605 deletions.
diff --git a/config/fileTransformer.js b/config/fileTransformer.js
diff --git a/config/webpack.config.js b/config/webpack.config.js
diff --git a/jest.config.js b/jest.config.js
diff --git a/package.json b/package.json
@@ -22,8 +22,8 @@
     "format:fix": "prettier --write .",
     "lint": "eslint --cache --ignore-path ./.gitignore --ext .js,.jsx,.ts,.tsx .",
     "lint:fix": "yarn lint --fix",
-    "test": "jest --no-cache --runInBand --forceExit --detectOpenHandles",
-    "test:cov": "jest --coverage --no-cache --runInBand --forceExit --detectOpenHandles",
+    "test": "vitest",
+    "test:cov": "vitest run --coverage",
     "addscope": "node tools/packagejson name @pyroscope"
   },
   "publishConfig": {
@@ -55,27 +55,23 @@
     "@types/busboy": "^1.5.4",
     "@types/debug": "^4.1.7",
     "@types/express": "^4.17.13",
-    "@types/jest": "^27.0.1",
     "@types/source-map": "^0.5.7",
     "@types/supertest": "^2.0.12",
     "@typescript-eslint/eslint-plugin": "^7.10.0",
     "@typescript-eslint/parser": "^7.10.0",
+    "@vitest/coverage-v8": "^2.0.5",
     "busboy": "^1.6.0",
     "eslint": "^8.5.6",
     "eslint-config-prettier": "^8.3.0",
     "eslint-plugin-prettier": "^4.0.0",
     "express": "^4.18.1",
     "husky": "^7.0.2",
-    "jest": "^27.2.0",
     "pinst": "^2.1.6",
     "pprof-format": "^2.1.0",
     "prettier": "^2.4.0",
     "supertest": "^6.2.3",
-    "ts-jest": "^27.0.5",
-    "ts-loader": "^9.2.5",
     "typescript": "^4.4.3",
-    "webpack": "^5.52.1",
-    "webpack-cli": "^4.8.0"
+    "vitest": "^2.0.5"
   },
   "engines": {
     "node": ">=v18"

diff --git a/test/index.test.ts b/test/index.test.ts
@@ -1,3 +1,5 @@
+import { describe, it, expect } from 'vitest';
+
 import fs from 'node:fs';
 import { Profile } from 'pprof-format';
 import Pyroscope from '../src';

diff --git a/test/middleware.test.ts b/test/middleware.test.ts
@@ -1,3 +1,5 @@
+import { describe, it, expect } from 'vitest';
+
 import Pyroscope from '../src';
 import request from 'supertest';
 import express from 'express';
@@ -6,9 +8,6 @@ import express from 'express';
 Pyroscope.init();
 
 describe('express middleware', () => {
-  afterAll(async () => {
-    await new Promise<void>((resolve) => setTimeout(() => resolve(), 500)); // avoid jest open handle error
-  });
   it('should be a function', () => {
     expect(typeof Pyroscope.expressMiddleware).toBe('function');
   });

diff --git a/test/profiler.test.ts b/test/profiler.test.ts
@@ -1,3 +1,5 @@
+import { describe, it, expect } from 'vitest';
+
 import Pyroscope from '../src';
 import express from 'express';
 import busboy from 'busboy';
@@ -114,7 +116,7 @@ describe('common behaviour of profilers', () => {
     });
   });
 
-  it('should allow to call start profiling twice', (done) => {
+  it('should allow to call start profiling twice', async () => {
     // Simulate memory usage
     const timer: NodeJS.Timeout = setInterval(() => {
       // Use some memory
@@ -139,16 +141,13 @@ describe('common behaviour of profilers', () => {
     });
     Pyroscope.startHeapProfiling();
     Pyroscope.startHeapProfiling();
-    (async () => {
-      await Pyroscope.stopHeapProfiling();
-      await Pyroscope.stopHeapProfiling();
-      // And stop it without starting CPU
-      await Pyroscope.stop();
 
-      clearInterval(timer);
+    await Pyroscope.stopHeapProfiling();
+    await Pyroscope.stopHeapProfiling();
+    // And stop it without starting CPU
+    await Pyroscope.stop();
 
-      done();
-    })();
+    clearInterval(timer);
   });
 
   it('should have dynamic labels on wall profile', (done) => {