fix: invoke onCollect when ws connected

vitest-dev · Jun 19, 2022 · 40bdd0a · 40bdd0a
1 parent 4b180f0
commit 40bdd0a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/pure/watch/client.ts b/src/pure/watch/client.ts
@@ -36,7 +36,10 @@ export function buildWatchClient(
       log.info('WS Opened')
       status.value = 'OPEN'
       client.state.filesMap.clear()
-      client.rpc.getFiles().then(files => client.state.collectFiles(files))
+      client.rpc.getFiles().then((files) => {
+        client.state.collectFiles(files)
+        handlers?.onCollected?.(files)
+      })
       client.rpc.getConfig().then(_config => config.value = _config)
     })