[Fix] extractProp: support JSXFragment

ljharb · ljharb · commit 8798c586da1c · 2023-07-25T15:23:20.000-07:00
Fixes #132
diff --git a/__tests__/src/elementType-test.js b/__tests__/src/elementType-test.js
@@ -92,5 +92,27 @@ describe('elementType tests', () => {
 
       assert.equal(actual, expected);
     });
+
+    it('works with nested fragments', () => {
+      const code = `
+        <Hello
+          role="checkbox"
+          frag={
+            <>
+              <div>Hello</div>
+              <>
+                <div>There</div>
+              </>
+            </>
+          }
+        />
+      `;
+      const node = getOpeningElement(code);
+
+      const expected = 'Hello';
+      const actual = elementType(node);
+
+      assert.equal(actual, expected);
+    });
   });
 });
diff --git a/__tests__/src/getPropValue-babelparser-test.js b/__tests__/src/getPropValue-babelparser-test.js
@@ -150,6 +150,28 @@ describe('getPropValue', () => {
 
       assert.equal(actual, expected);
     });
+
+    it('supports a prop value containing nested fragments', () => {
+      const propCode = `
+        <>
+          <div>Hello</div>
+          <>
+            <div>There</div>
+          </>
+        </>
+      `;
+      const code = `
+        <Hello
+          role="checkbox"
+          frag={${propCode}}
+        />
+      `;
+
+      const prop = extractProp(code, 'frag');
+      const actual = getPropValue(prop);
+
+      assert.deepEqual(actual, propCode.trim());
+    });
   });
 
   describe('Identifier', () => {
diff --git a/src/values/index.js b/src/values/index.js
@@ -1,6 +1,7 @@
 import Literal from './Literal';
 import JSXElement from './JSXElement';
 import JSXText from './JSXText';
+import JSXFragment from './JSXFragment';
 import JSXExpressionContainer, { extractLiteral } from './expressions';
 
 // Composition map of types to their extractor functions.
@@ -9,6 +10,7 @@ const TYPES = {
   JSXElement,
   JSXExpressionContainer,
   JSXText,
+  JSXFragment,
 };
 
 // Composition map of types to their extractor functions to handle literals.
@@ -28,6 +30,7 @@ const LITERAL_TYPES = {
  * @param value - AST Value object on a JSX Attribute.
  */
 export default function getValue(value) {
+  if (!TYPES[value.type]) console.log(value.type);
   return TYPES[value.type](value);
 }
 
