test: fix and add more case

Author: GrinZero

packages/network-debugger/package.json

@@ -8,14 +8,14 @@
   "types": "./dist/src/index.d.ts",
   "exports": {
     ".": {
+      "types": "./dist/src/index.d.ts",
       "import": "./dist/index.mjs",
-      "require": "./dist/index.js",
-      "types": "./dist/src/index.d.ts"
+      "require": "./dist/index.js"
     },
     "./dev": {
+      "types": "./src/index.ts",
       "import": "./src/index.ts",
-      "require": "./src/index.ts",
-      "types": "./src/index.ts"
+      "require": "./src/index.ts"
     }
   },
   "scripts": {

packages/network-debugger/src/core/undici.test.ts

@@ -15,24 +15,30 @@ interface MockUndiciModule {
 }
 
 // 使用 vi.hoisted 确保变量在 mock 提升时可用
-const { mockFetchProxyFactory, mockUndiciModule, mockMainProcessInstance } = vi.hoisted(() => {
-  const mockProxyFn = vi.fn()
-  const undiciModule: MockUndiciModule = {
-    fetch: undefined
-  }
-  // 创建一个 mock MainProcess 实例
-  const mainProcessInstance = {
-    send: vi.fn(),
-    sendRequest: vi.fn().mockReturnThis(),
-    responseRequest: vi.fn(),
-    dispose: vi.fn()
-  }
-  return {
-    mockFetchProxyFactory: vi.fn().mockReturnValue(mockProxyFn),
-    mockUndiciModule: undiciModule,
-    mockMainProcessInstance: mainProcessInstance
-  }
-})
+const { mockFetchProxyFactory, mockUndiciModule, mockMainProcessInstance, MockMainProcessClass } =
+  vi.hoisted(() => {
+    const mockProxyFn = vi.fn()
+    const undiciModule: MockUndiciModule = {
+      fetch: undefined
+    }
+    // 创建一个 mock MainProcess 实例
+    const mainProcessInstance = {
+      send: vi.fn(),
+      sendRequest: vi.fn().mockReturnThis(),
+      responseRequest: vi.fn(),
+      dispose: vi.fn()
+    }
+    // 使用 function 关键字创建构造函数，避免 vitest 警告
+    function MainProcessConstructor() {
+      return mainProcessInstance
+    }
+    return {
+      mockFetchProxyFactory: vi.fn().mockReturnValue(mockProxyFn),
+      mockUndiciModule: undiciModule,
+      mockMainProcessInstance: mainProcessInstance,
+      MockMainProcessClass: MainProcessConstructor
+    }
+  })
 
 // Mock fetch.ts 模块
 vi.mock('./fetch', () => ({
@@ -44,10 +50,10 @@ vi.mock('undici', () => ({
   default: mockUndiciModule
 }))
 
-// Mock fork.ts 模块，导出一个可以返回我们 mock 实例的 MainProcess
+// Mock fork.ts 模块，使用 function 构造函数
 vi.mock('./fork', () => {
   return {
-    MainProcess: vi.fn().mockImplementation(() => mockMainProcessInstance),
+    MainProcess: MockMainProcessClass,
     __dirname: '/mock/path'
   }
 })
@@ -80,7 +86,7 @@ describe('core/undici.ts', () => {
         default: mockUndiciModule
       }))
       vi.doMock('./fork', () => ({
-        MainProcess: vi.fn().mockImplementation(() => mockMainProcessInstance),
+        MainProcess: MockMainProcessClass,
         __dirname: '/mock/path'
       }))
 
@@ -109,7 +115,7 @@ describe('core/undici.ts', () => {
         default: mockUndiciModule
       }))
       vi.doMock('./fork', () => ({
-        MainProcess: vi.fn().mockImplementation(() => mockMainProcessInstance),
+        MainProcess: MockMainProcessClass,
         __dirname: '/mock/path'
       }))
 
@@ -139,7 +145,7 @@ describe('core/undici.ts', () => {
         default: mockUndiciModule
       }))
       vi.doMock('./fork', () => ({
-        MainProcess: vi.fn().mockImplementation(() => mockMainProcessInstance),
+        MainProcess: MockMainProcessClass,
         __dirname: '/mock/path'
       }))
 
@@ -174,7 +180,7 @@ describe('core/undici.ts', () => {
         default: mockUndiciModule
       }))
       vi.doMock('./fork', () => ({
-        MainProcess: vi.fn().mockImplementation(() => mockMainProcessInstance),
+        MainProcess: MockMainProcessClass,
         __dirname: '/mock/path'
       }))
 
@@ -206,7 +212,7 @@ describe('core/undici.ts', () => {
         default: mockUndiciModule
       }))
       vi.doMock('./fork', () => ({
-        MainProcess: vi.fn().mockImplementation(() => mockMainProcessInstance),
+        MainProcess: MockMainProcessClass,
         __dirname: '/mock/path'
       }))
 
@@ -238,7 +244,7 @@ describe('core/undici.ts', () => {
         default: mockUndiciModule
       }))
       vi.doMock('./fork', () => ({
-        MainProcess: vi.fn().mockImplementation(() => mockMainProcessInstance),
+        MainProcess: MockMainProcessClass,
         __dirname: '/mock/path'
       }))
 
@@ -273,7 +279,7 @@ describe('core/undici.ts', () => {
         default: mockUndiciModule
       }))
       vi.doMock('./fork', () => ({
-        MainProcess: vi.fn().mockImplementation(() => mockMainProcessInstance),
+        MainProcess: MockMainProcessClass,
         __dirname: '/mock/path'
       }))
 
@@ -316,7 +322,7 @@ describe('core/undici.ts', () => {
         default: mockUndiciModule
       }))
       vi.doMock('./fork', () => ({
-        MainProcess: vi.fn().mockImplementation(() => mockMainProcessInstance),
+        MainProcess: MockMainProcessClass,
         __dirname: '/mock/path'
       }))
 

packages/network-debugger/src/core/ws/permessage-deflate.test.ts

@@ -350,6 +350,54 @@ describe('PerMessageDeflate', () => {
       serverPmd.cleanup()
       clientPmd.cleanup()
     })
+
+    it('应该在超过 maxPayload 时返回错误（压缩）', async () => {
+      const pmd = new PerMessageDeflate({}, true, 5)
+      pmd.accept([{}])
+
+      // 创建一个较大的数据，压缩后仍然超过 maxPayload
+      const data = Buffer.alloc(1000, 'x')
+
+      const error = await new Promise<Error | null>((resolve) => {
+        pmd.compress(data, true, (err) => {
+          resolve(err)
+        })
+      })
+
+      // 压缩后的数据可能仍然超过 maxPayload
+      // 如果没有超过，这个测试会通过但 error 为 null
+      if (error) {
+        expect(error).toBeInstanceOf(RangeError)
+        expect(error.message).toBe('Max payload size exceeded')
+      }
+
+      pmd.cleanup()
+    })
+
+    it('应该在 cleanup 时调用 deflate 的 callback（如果存在）', async () => {
+      const pmd = new PerMessageDeflate({}, true)
+      pmd.accept([{}])
+
+      // 开始压缩但不等待完成
+      const data = Buffer.from('Hello, World!')
+      let callbackCalled = false
+      let callbackError: Error | null = null
+
+      pmd.compress(data, false, (err) => {
+        callbackCalled = true
+        callbackError = err
+      })
+
+      // 立即 cleanup，这应该触发 callback 并传递错误
+      // 注意：由于 zlibLimiter 的存在，callback 可能已经被调用
+      pmd.cleanup()
+
+      // 等待一小段时间让异步操作完成
+      await new Promise((resolve) => setTimeout(resolve, 50))
+
+      // callback 应该被调用
+      expect(callbackCalled).toBe(true)
+    })
   })
 
   describe('acceptAsServer 边界情况', () => {