diff --git a/packages/components/tree-v2/__tests__/tree.test.ts b/packages/components/tree-v2/__tests__/tree.test.ts index bbc6546961752f..95cb7541753e5f 100644 --- a/packages/components/tree-v2/__tests__/tree.test.ts +++ b/packages/components/tree-v2/__tests__/tree.test.ts @@ -183,9 +183,11 @@ const createTree = ( describe('Virtual Tree', () => { test('create', async () => { - const { treeVm } = createTree() + const { wrapper, treeVm } = createTree() await nextTick() expect(treeVm.flattenTree.length).toEqual(NODE_NUMBER) + const iconWrapper = wrapper.find(TREE_NODE_EXPAND_ICON_CLASS_NAME) + expect(iconWrapper.element.innerHTML).contains('svg') }) test('click node', async () => {