From 87f614b7033ec598d5356265c59459f4c7d9ba0e Mon Sep 17 00:00:00 2001
From: Amour1688 <lcz_1996@foxmail.com>
Date: Sat, 12 Dec 2020 00:36:24 +0800
Subject: [PATCH] chore: add test for object slots

---
 .../babel-plugin-jsx/src/transform-vue-jsx.ts |  4 +-
 packages/babel-plugin-jsx/test/index.test.tsx | 38 ++++++++++++++++---
 2 files changed, 34 insertions(+), 8 deletions(-)

diff --git a/packages/babel-plugin-jsx/src/transform-vue-jsx.ts b/packages/babel-plugin-jsx/src/transform-vue-jsx.ts
index 59fca37..dda6796 100644
--- a/packages/babel-plugin-jsx/src/transform-vue-jsx.ts
+++ b/packages/babel-plugin-jsx/src/transform-vue-jsx.ts
@@ -121,8 +121,8 @@ const transformJSXElement = (
     } else if (
       t.isCallExpression(child) && child.loc && isComponent
     ) { // the element was generated and doesn't have location information
-      const slotId = path.scope.generateUidIdentifier('slot');
-      const scope = path.scope;
+      const { scope } = path;
+      const slotId = scope.generateUidIdentifier('slot');
       if (scope) {
         scope.push({
           id: slotId,
diff --git a/packages/babel-plugin-jsx/test/index.test.tsx b/packages/babel-plugin-jsx/test/index.test.tsx
index 4f6e5f3..df0b6d4 100644
--- a/packages/babel-plugin-jsx/test/index.test.tsx
+++ b/packages/babel-plugin-jsx/test/index.test.tsx
@@ -571,23 +571,49 @@ describe('should support passing object slots via JSX children', () => {
     expect(wrapper.html()).toBe('<span>foo<!----></span>');
   });
 
-  test('single expression, array map expression', () => {
-    const Test = defineComponent({
-      setup(_, { slots }) {
-        return () => <span>{slots.default?.()}</span>;
+  test('single expression, function expression variable', () => {
+    const foo = () => 'foo';
+
+    const wrapper = mount({
+      render() {
+        return (
+          <A>{foo}</A>
+        );
       },
     });
 
+    expect(wrapper.html()).toBe('<span>foo<!----></span>');
+  });
+
+  test('single expression, array map expression', () => {
     const data = ['A', 'B', 'C'];
 
     const wrapper = mount({
       render() {
         return (
-          <div>{data.map((item: string) => <Test><span>{item}</span></Test>)}</div>
+          <>
+            {data.map((item) => <A><span>{item}</span></A>)}
+          </>
         );
       },
     });
 
-    expect(wrapper.html()).toBe('<div><span><span>A</span></span><span><span>B</span></span><span><span>C</span></span></div>');
+    expect(wrapper.html()).toBe('<span><span>A</span><!----></span><span><span>B</span><!----></span><span><span>C</span><!----></span>');
+  });
+
+  test('xx', () => {
+    const data = ['A', 'B', 'C'];
+
+    const wrapper = mount({
+      render() {
+        return (
+          <>
+            {data.map((item) => <A>{() => <span>{item}</span>}</A>)}
+          </>
+        );
+      },
+    });
+
+    expect(wrapper.html()).toBe('<span><span>A</span><!----></span><span><span>B</span><!----></span><span><span>C</span><!----></span>');
   });
 });