From 20ea7dd9860e5f632e95d6644c81aae25cbc69c8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E4=B8=89=E5=92=B2=E6=99=BA=E5=AD=90?= <sxzz@sxzz.moe>
Date: Tue, 1 Mar 2022 21:52:35 +0800
Subject: [PATCH] test: improve renderFn

---
 .../components/col/__tests__/layout.spec.tsx  | 38 ++++++++-----------
 1 file changed, 15 insertions(+), 23 deletions(-)

diff --git a/packages/components/col/__tests__/layout.spec.tsx b/packages/components/col/__tests__/layout.spec.tsx
index 931e2df72f..692cec9682 100644
--- a/packages/components/col/__tests__/layout.spec.tsx
+++ b/packages/components/col/__tests__/layout.spec.tsx
@@ -31,15 +31,11 @@ describe('Col', () => {
   })
 
   it('gutter', () => {
-    const TestComponent = defineComponent({
-      setup: () => () =>
-        (
-          <Row gutter={20}>
-            <Col span={12} ref="col"></Col>
-          </Row>
-        ),
-    })
-    const wrapper = mount(TestComponent)
+    const wrapper = mount(() => (
+      <Row gutter={20}>
+        <Col span={12} ref="col"></Col>
+      </Row>
+    ))
     const colElm = wrapper.findComponent({ ref: 'col' }).element as HTMLElement
     expect(colElm.style.paddingLeft === '10px').toBe(true)
     expect(colElm.style.paddingRight === '10px').toBe(true)
@@ -73,20 +69,16 @@ describe('Col', () => {
   })
 
   it('responsive', () => {
-    const TestComponent = defineComponent({
-      setup: () => () =>
-        (
-          <Row gutter={20}>
-            <Col
-              ref="col"
-              sm={{ span: 4, offset: 2 }}
-              md={8}
-              lg={{ span: 6, offset: 3 }}
-            />
-          </Row>
-        ),
-    })
-    const wrapper = mount(TestComponent)
+    const wrapper = mount(() => (
+      <Row gutter={20}>
+        <Col
+          ref="col"
+          sm={{ span: 4, offset: 2 }}
+          md={8}
+          lg={{ span: 6, offset: 3 }}
+        />
+      </Row>
+    ))
     const colElmClass = wrapper.findComponent({ ref: 'col' }).classes()
     expect(colElmClass.includes('el-col-sm-4')).toBe(true)
     expect(colElmClass.includes('el-col-sm-4')).toBe(true)