test: add vitest configuration and test setup

Author: zhensherlock

package.json

@@ -16,9 +16,8 @@
     "lint": "npx eslint \"src/**/*.{ts,js}\"",
     "build": "tsc --noEmit && cross-env NODE_ENV=production node scripts/build.js",
     "dev": "cross-env NODE_ENV=local concurrently \"tsc --noEmit --watch\" \"node scripts/dev.js\"",
-    "test": "c8 --reporter=lcov --reporter=text tsx ./src/index.ts",
-    "report": "c8 report --reporter=lcov --reporter=html",
-    "coverage": "rimraf coverage && npm run test",
+    "test": "vitest run",
+    "coverage": "rimraf coverage && vitest run --coverage",
     "changelog": "conventional-changelog -p angular -i CHANGELOG.md -s -r 0 -n changelog-option.js"
   },
   "dependencies": {
@@ -31,7 +30,6 @@
     "@typescript-eslint/eslint-plugin": "^8.39.1",
     "@typescript-eslint/parser": "^8.39.1",
     "@vitest/coverage-v8": "^3.2.4",
-    "c8": "^10.1.3",
     "compare-func": "^2.0.0",
     "concurrently": "^9.2.0",
     "conventional-changelog-angular": "^8.0.0",

src/index.ts

@@ -5,7 +5,7 @@ import { StdioClientTransport } from '@modelcontextprotocol/sdk/client/stdio.js'
 import { StreamableHTTPClientTransport } from '@modelcontextprotocol/sdk/client/streamableHttp.js'
 import { SSEClientTransport } from '@modelcontextprotocol/sdk/client/sse.js'
 
-async function testStdioTransport() {
+export async function testStdioTransport() {
   console.log('start stdio client transport')
   const client = new Client({
     name: 'mcp-stdio-client',
@@ -47,7 +47,7 @@ async function testStdioTransport() {
   await sleep(1000)
 }
 
-async function testStreamableHttpTransport() {
+export async function testStreamableHttpTransport() {
   console.log('start streamable http client transport')
   const client = new Client({
     name: 'mcp-http-client',
@@ -84,7 +84,7 @@ async function testStreamableHttpTransport() {
   await sleep(1000)
 }
 
-async function testSSETransport() {
+export async function testSSETransport() {
   console.log('start sse client transport')
   const client = new Client({
     name: 'mcp-sse-client',

tests/index.test.ts

@@ -0,0 +1,16 @@
+import { describe, test } from 'vitest'
+import { testStdioTransport, testStreamableHttpTransport, testSSETransport } from '@/index'
+
+describe('MCP client', () => {
+  test('stdio transport', async () => {
+    await testStdioTransport()
+  })
+
+  test('streamable http transport', async () => {
+    await testStreamableHttpTransport()
+  })
+
+  test('sse transport', async () => {
+    await testSSETransport()
+  })
+})

tests/utils.ts

@@ -0,0 +1,20 @@
+export function waitForValue<T>(
+  getterFn: () => T | undefined | null,
+  checkInterval = 100,
+  timeout = 10000,
+): Promise<T> {
+  return new Promise((resolve, reject) => {
+    const start = Date.now()
+
+    const intervalId = setInterval(() => {
+      const value = getterFn()
+      if (value) {
+        clearInterval(intervalId)
+        resolve(value)
+      } else if (Date.now() - start > timeout) {
+        clearInterval(intervalId)
+        reject(new Error('Timeout waiting for value'))
+      }
+    }, checkInterval)
+  })
+}

vitest.config.ts

@@ -0,0 +1,23 @@
+import { defineConfig } from 'vitest/config'
+
+export default defineConfig({
+  resolve: {
+    alias: {
+      '@': '/src',
+    },
+  },
+  test: {
+    globalSetup: ['./vitest.global.ts'],
+    coverage: {
+      include: ['src/**/*.ts'],
+      reporter: ['text', 'lcov', 'html'],
+    },
+    pool: 'threads',
+    poolOptions: {
+      threads: {
+        maxThreads: 1,
+        minThreads: 1,
+      }
+    },
+  },
+})

vitest.global.ts

@@ -0,0 +1,19 @@
+import { spawn } from 'child_process'
+import { waitForValue } from './tests/utils'
+
+export default async function setup() {
+  const webProcess = spawn('npx', ['-y', '@my-mcp-hub/node-mcp-server', 'web'], {
+    stdio: 'pipe',
+  })
+  let webStarted = false
+  webProcess.stdout?.on('data', async (data) => {
+    const output = data.toString()
+    if (output.includes('MCP server started')) {
+      webStarted = true
+    }
+  });
+  await waitForValue(() => webStarted)
+  return () => {
+    webProcess.kill('SIGINT')
+  }
+}