From 881045a1d0fee037cbda5359d32bb0a18eec0e86 Mon Sep 17 00:00:00 2001
From: XieZongChen <46394163+amadeus711@users.noreply.github.com>
Date: Mon, 20 Jun 2022 23:37:57 +0800
Subject: [PATCH] test(tree): add test (#3144)

---
 src/tree/tests/Tree.spec.ts | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/src/tree/tests/Tree.spec.ts b/src/tree/tests/Tree.spec.ts
index 02905acef2b..17120908c15 100644
--- a/src/tree/tests/Tree.spec.ts
+++ b/src/tree/tests/Tree.spec.ts
@@ -450,4 +450,28 @@ describe('n-tree', () => {
     await node[0].trigger('click')
     expect(wrapper.findAll('.n-tree-node--selected').length).toBe(2)
   })
+
+  it('should work with `node-props` prop', async () => {
+    const testClass = 'menu-test'
+    const wrapper = mount(NTree, {
+      props: {
+        data: [
+          {
+            label: '1',
+            key: '1'
+          },
+          {
+            label: '2',
+            key: '2'
+          },
+          {
+            label: '3',
+            key: '3'
+          }
+        ],
+        nodeProps: () => ({ class: testClass })
+      }
+    })
+    expect(wrapper.find('.n-tree-node').classes()).toContain(testClass)
+  })
 })