feat(ws-client): allow change reactive for state, filesMap and idMap (#…

…5906)
vitest-dev · Jun 17, 2024 · e6020b9 · e6020b9
1 parent 0dda80e
commit e6020b9
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/packages/ws-client/src/index.ts b/packages/ws-client/src/index.ts
@@ -14,7 +14,7 @@ export interface VitestClientOptions {
   reconnectInterval?: number
   reconnectTries?: number
   connectTimeout?: number
-  reactive?: <T>(v: T) => T
+  reactive?: <T>(v: T, forKey: 'state' | 'idMap' | 'filesMap') => T
   ref?: <T>(v: T) => { value: T }
   WebSocketConstructor?: typeof WebSocket
 }
@@ -44,10 +44,10 @@ export function createClient(url: string, options: VitestClientOptions = {}) {
     state: new StateManager(),
     waitForConnection,
     reconnect,
-  }) as VitestClient
+  }, 'state') as VitestClient
 
-  ctx.state.filesMap = reactive(ctx.state.filesMap)
-  ctx.state.idMap = reactive(ctx.state.idMap)
+  ctx.state.filesMap = reactive(ctx.state.filesMap, 'filesMap')
+  ctx.state.idMap = reactive(ctx.state.idMap, 'idMap')
 
   let onMessage: Function
   const functions: WebSocketEvents = {