chore(ui): include test and custom types (#5940)

vitest-dev · Jun 20, 2024 · a17635b · a17635b
1 parent 7ae160b
commit a17635b
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/packages/ui/client/composables/explorer/types.ts b/packages/ui/client/composables/explorer/types.ts
@@ -37,6 +37,14 @@ export interface UITaskTreeNode extends TaskTreeNode {
  duration?: number
 }
 
+export interface TestTreeNode extends UITaskTreeNode {
+ type: 'test'
+}
+
+export interface CustomTestTreeNode extends UITaskTreeNode {
+ type: 'custom'
+}
+
 export interface ParentTreeNode extends UITaskTreeNode {
  children: Set<string>
  tasks: UITaskTreeNode[]

diff --git a/packages/ui/client/composables/explorer/utils.ts b/packages/ui/client/composables/explorer/utils.ts
@@ -1,12 +1,19 @@
 import type { File, Task } from '@vitest/runner'
 import { isAtomTest } from '@vitest/runner/utils'
-import type { FileTreeNode, ParentTreeNode, SuiteTreeNode, UITaskTreeNode } from '~/composables/explorer/types'
+import type {
+ CustomTestTreeNode,
+ FileTreeNode,
+ ParentTreeNode,
+ SuiteTreeNode,
+ TestTreeNode,
+ UITaskTreeNode,
+} from '~/composables/explorer/types'
 import { client } from '~/composables/client'
 import { getProjectNameColor, isSuite as isTaskSuite } from '~/utils/task'
 import { explorerTree } from '~/composables/explorer/index'
 import { openedTreeItemsSet } from '~/composables/explorer/state'
 
-export function isTestNode(node: UITaskTreeNode): node is FileTreeNode {
+export function isTestNode(node: UITaskTreeNode): node is TestTreeNode | CustomTestTreeNode {
  return node.type === 'test' || node.type === 'custom'
 }