refactor: update test setup and configuration

Author: zhensherlock

package-lock.json

@@ -22,7 +22,7 @@
     "dev:stdio": "cross-env NODE_ENV=local concurrently \"tsc --noEmit --watch\" \"node scripts/dev.js\"",
     "dev:web": "cross-env NODE_ENV=local TRANSPORT=web concurrently \"tsc --noEmit --watch\" \"node scripts/dev.js\"",
     "test": "vitest run",
-    "coverage": "vitest run --coverage"
+    "coverage": "rimraf coverage && npm run test && c8 report --reporter=html"
   },
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.15.0",
@@ -48,6 +48,7 @@
     "@typescript-eslint/eslint-plugin": "^8.36.0",
     "@typescript-eslint/parser": "^8.36.0",
     "@vitest/coverage-v8": "^3.2.4",
+    "c8": "^10.1.3",
     "concurrently": "^9.2.0",
     "cross-env": "^7.0.3",
     "esbuild": "^0.25.6",
@@ -59,6 +60,7 @@
     "lint-staged": "^16.1.2",
     "prettier": "^3.6.2",
     "rimraf": "^6.0.1",
+    "tsx": "^4.20.3",
     "typescript": "^5.8.3",
     "vitest": "^3.2.4"
   },

package.json

@@ -2,7 +2,6 @@ import { McpServer, ResourceTemplate } from '@modelcontextprotocol/sdk/server/mc
 import type { OptionsType } from '@/types'
 
 export const registerResources = (server: McpServer, options: OptionsType) => {
-  // @ts-ignore
   server.registerResource(
     'search',
     new ResourceTemplate('search://{keyword}', {
@@ -17,7 +16,7 @@ export const registerResources = (server: McpServer, options: OptionsType) => {
         contents: [
           {
             uri: uri.href,
-            text: `hello ${keyword}`,
+            text: `search ${keyword}`,
           },
         ],
       }

src/resources/index.ts

@@ -0,0 +1,24 @@
+import { describe, expect, test } from 'vitest'
+
+describe('echoPrompt', () => {
+  test('should return correct prompt content for a valid argument', async () => {
+    expect(
+      await global.client.getPrompt({
+        name: 'echo',
+        arguments: {
+          message: 'hello',
+        },
+      }),
+    ).toStrictEqual({
+      messages: [
+        {
+          role: 'user',
+          content: {
+            type: 'text',
+            text: 'Please process this message: hello',
+          },
+        },
+      ],
+    })
+  })
+})

tests/prompts/index.test.ts

@@ -0,0 +1,18 @@
+import { describe, expect, test } from 'vitest'
+
+describe('searchResource', () => {
+  test('should return correct resource content for a valid URI', async () => {
+    expect(
+      await global.client.readResource({
+        uri: 'search://hello',
+      }),
+    ).toStrictEqual({
+      contents: [
+        {
+          uri: 'search://hello',
+          text: 'search hello',
+        },
+      ],
+    })
+  })
+})

tests/resources/index.test.ts

@@ -0,0 +1,24 @@
+import { describe, expect, test } from 'vitest'
+
+describe('watermarkJsPlusDocTool', () => {
+  test('returns data for a valid input', async () => {
+    const res = (await global.client.callTool({
+      name: 'query-watermark-js-plus',
+      arguments: {
+        keyword: 'On Demand',
+      },
+    })) as { content: unknown[] }
+
+    expect(res.content.length).toBeGreaterThan(0)
+  }, 20000)
+
+  test('returns a "not found" response for an unrecognized input', async () => {
+    const res = (await global.client.callTool({
+      name: 'query-watermark-js-plus',
+      arguments: {
+        keyword: 'ESLint',
+      },
+    })) as { content: unknown[] }
+    expect(res.content.length).toEqual(0)
+  }, 20000)
+})