diff --git a/components/tree-select/__tests__/__snapshots__/demo.test.js.snap b/components/tree-select/__tests__/__snapshots__/demo.test.js.snap
index 95ba4c2bc9..30f45a755a 100644
--- a/components/tree-select/__tests__/__snapshots__/demo.test.js.snap
+++ b/components/tree-select/__tests__/__snapshots__/demo.test.js.snap
@@ -24,7 +24,23 @@ exports[`renders ./components/tree-select/demo/basic.md correctly 1`] = `
class="ant-select-arrow"
style="outline:none"
>
-
+
+
+
@@ -53,7 +69,25 @@ exports[`renders ./components/tree-select/demo/checkable.md correctly 1`] = `
>
+ >
+
+
+
+
@@ -162,7 +196,23 @@ exports[`renders ./components/tree-select/demo/treeData.md correctly 1`] = `
class="ant-select-arrow"
style="outline:none"
>
-
+
+
+
diff --git a/components/tree/__tests__/__snapshots__/demo.test.js.snap b/components/tree/__tests__/__snapshots__/demo.test.js.snap
index a0a168bc64..5e4422ed20 100644
--- a/components/tree/__tests__/__snapshots__/demo.test.js.snap
+++ b/components/tree/__tests__/__snapshots__/demo.test.js.snap
@@ -14,19 +14,19 @@ exports[`renders ./components/tree/demo/basic.md correctly 1`] = `
class="ant-tree-switcher ant-tree-switcher_open"
>
@@ -61,19 +61,19 @@ exports[`renders ./components/tree/demo/basic.md correctly 1`] = `
class="ant-tree-switcher ant-tree-switcher_close"
>
@@ -104,19 +104,19 @@ exports[`renders ./components/tree/demo/basic.md correctly 1`] = `
class="ant-tree-switcher ant-tree-switcher_open"
>
@@ -193,19 +193,19 @@ exports[`renders ./components/tree/demo/basic-controlled.md correctly 1`] = `
class="ant-tree-switcher ant-tree-switcher_open"
>
@@ -240,19 +240,19 @@ exports[`renders ./components/tree/demo/basic-controlled.md correctly 1`] = `
class="ant-tree-switcher ant-tree-switcher_open"
>
@@ -364,19 +364,19 @@ exports[`renders ./components/tree/demo/basic-controlled.md correctly 1`] = `
class="ant-tree-switcher ant-tree-switcher_open"
>
@@ -515,19 +515,19 @@ exports[`renders ./components/tree/demo/basic-controlled.md correctly 1`] = `
class="ant-tree-switcher ant-tree-switcher_close"
>
@@ -592,19 +592,19 @@ exports[`renders ./components/tree/demo/customized-icon.md correctly 1`] = `
class="ant-tree-switcher ant-tree-switcher_open"
>
@@ -751,19 +751,19 @@ exports[`renders ./components/tree/demo/directory.md correctly 1`] = `
class="ant-tree-switcher ant-tree-switcher_open"
>
@@ -892,19 +892,19 @@ exports[`renders ./components/tree/demo/directory.md correctly 1`] = `
class="ant-tree-switcher ant-tree-switcher_open"
>
@@ -1042,19 +1042,19 @@ exports[`renders ./components/tree/demo/draggable.md correctly 1`] = `
class="ant-tree-switcher ant-tree-switcher_open"
>
@@ -1084,19 +1084,19 @@ exports[`renders ./components/tree/demo/draggable.md correctly 1`] = `
class="ant-tree-switcher ant-tree-switcher_open"
>
@@ -1188,19 +1188,19 @@ exports[`renders ./components/tree/demo/draggable.md correctly 1`] = `
class="ant-tree-switcher ant-tree-switcher_close"
>
@@ -1248,19 +1248,19 @@ exports[`renders ./components/tree/demo/draggable.md correctly 1`] = `
class="ant-tree-switcher ant-tree-switcher_close"
>
@@ -1315,19 +1315,19 @@ exports[`renders ./components/tree/demo/dynamic.md correctly 1`] = `
class="ant-tree-switcher ant-tree-switcher_close"
>
@@ -1351,19 +1351,19 @@ exports[`renders ./components/tree/demo/dynamic.md correctly 1`] = `
class="ant-tree-switcher ant-tree-switcher_close"
>
@@ -1731,19 +1731,19 @@ exports[`renders ./components/tree/demo/search.md correctly 1`] = `
class="ant-tree-switcher ant-tree-switcher_close"
>
@@ -1772,19 +1772,19 @@ exports[`renders ./components/tree/demo/search.md correctly 1`] = `
class="ant-tree-switcher ant-tree-switcher_close"
>
diff --git a/components/tree/__tests__/__snapshots__/directory.test.js.snap b/components/tree/__tests__/__snapshots__/directory.test.js.snap
index 1dccf09912..3b3617e660 100644
--- a/components/tree/__tests__/__snapshots__/directory.test.js.snap
+++ b/components/tree/__tests__/__snapshots__/directory.test.js.snap
@@ -14,19 +14,19 @@ exports[`Directory Tree defaultExpandAll 1`] = `
class="ant-tree-switcher ant-tree-switcher_open"
>
@@ -155,19 +155,19 @@ exports[`Directory Tree defaultExpandAll 1`] = `
class="ant-tree-switcher ant-tree-switcher_open"
>
@@ -305,19 +305,19 @@ exports[`Directory Tree defaultExpandParent 1`] = `
class="ant-tree-switcher ant-tree-switcher_close"
>
@@ -362,19 +362,19 @@ exports[`Directory Tree defaultExpandParent 1`] = `
class="ant-tree-switcher ant-tree-switcher_close"
>
@@ -428,19 +428,19 @@ exports[`Directory Tree expand click 1`] = `
class="ant-tree-switcher ant-tree-switcher_open"
>
@@ -570,19 +570,19 @@ exports[`Directory Tree expand click 1`] = `
class="ant-tree-switcher ant-tree-switcher_close"
>
@@ -636,19 +636,19 @@ exports[`Directory Tree expand click 2`] = `
class="ant-tree-switcher ant-tree-switcher_close"
>
@@ -778,19 +778,19 @@ exports[`Directory Tree expand click 2`] = `
class="ant-tree-switcher ant-tree-switcher_close"
>
@@ -844,19 +844,19 @@ exports[`Directory Tree expand double click 1`] = `
class="ant-tree-switcher ant-tree-switcher_open"
>
@@ -986,19 +986,19 @@ exports[`Directory Tree expand double click 1`] = `
class="ant-tree-switcher ant-tree-switcher_close"
>
@@ -1052,19 +1052,19 @@ exports[`Directory Tree expand double click 2`] = `
class="ant-tree-switcher ant-tree-switcher_close"
>
@@ -1194,19 +1194,19 @@ exports[`Directory Tree expand double click 2`] = `
class="ant-tree-switcher ant-tree-switcher_close"
>
@@ -1260,19 +1260,19 @@ exports[`Directory Tree expand with state control click 1`] = `
class="ant-tree-switcher ant-tree-switcher_open"
>
@@ -1484,19 +1484,19 @@ exports[`Directory Tree expandedKeys update 1`] = `
class="ant-tree-switcher ant-tree-switcher_close"
>
@@ -1541,19 +1541,19 @@ exports[`Directory Tree expandedKeys update 1`] = `
class="ant-tree-switcher ant-tree-switcher_open"
>
@@ -1692,19 +1692,19 @@ exports[`Directory Tree group select 1`] = `
class="ant-tree-switcher ant-tree-switcher_open"
>
@@ -1833,19 +1833,19 @@ exports[`Directory Tree group select 1`] = `
class="ant-tree-switcher ant-tree-switcher_open"
>
@@ -1983,19 +1983,19 @@ exports[`Directory Tree group select 2`] = `
class="ant-tree-switcher ant-tree-switcher_open"
>
@@ -2124,19 +2124,19 @@ exports[`Directory Tree group select 2`] = `
class="ant-tree-switcher ant-tree-switcher_open"
>
@@ -2274,19 +2274,19 @@ exports[`Directory Tree selectedKeys update 1`] = `
class="ant-tree-switcher ant-tree-switcher_open"
>
@@ -2415,19 +2415,19 @@ exports[`Directory Tree selectedKeys update 1`] = `
class="ant-tree-switcher ant-tree-switcher_open"
>