React snippet rendering: Add MDX test case

storybookjs · May 28, 2021 · d951eec · d951eec
1 parent b55d051
commit d951eec
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/addons/docs/src/frameworks/react/jsxDecorator.test.tsx b/addons/docs/src/frameworks/react/jsxDecorator.test.tsx
@@ -252,4 +252,24 @@ describe('jsxDecorator', () => {
     jsxDecorator(storyFn, context);
     expect(transformSource).toHaveBeenCalledWith('<div>\n  args story\n</div>', context);
   });
+
+  it('renders MDX properly', () => {
+    // FIXME: generate this from actual MDX
+    const mdxElement = {
+      type: { displayName: 'MDXCreateElement' },
+      props: {
+        mdxType: 'div',
+        originalType: 'div',
+        className: 'foo',
+      },
+    };
+
+    jsxDecorator(() => mdxElement, makeContext('mdx-args', { __isArgsStory: true }, {}));
+
+    expect(mockChannel.emit).toHaveBeenCalledWith(
+      SNIPPET_RENDERED,
+      'jsx-test--mdx-args',
+      '<div className="foo" />'
+    );
+  });
 });