Override default code-block element. Fix #41

See https://github.com/facebook/draft-js/blob/16ed6e5a1e7c9d5a0d5718b9a50c8256ee33f938/src/model/immutable/DefaultDraftBlockRenderMap.js. See also facebookarchive/draft-js#406.
springload · Feb 8, 2017 · 431c3fd · 431c3fd
1 parent dcab498
commit 431c3fd
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/lib/api/behavior.js b/lib/api/behavior.js
@@ -19,6 +19,15 @@ export default {
     getBlockRenderMap(blockTypes = []) {
         const renderMap = {};
 
+        // Override default element for code block.
+        // Fix https://github.com/facebook/draft-js/issues/406.
+        if (blockTypes.some(block => block.type === BLOCK_TYPE.CODE)) {
+            renderMap[BLOCK_TYPE.CODE] = {
+                element: 'code',
+                wrapper: DefaultDraftBlockRenderMap.get(BLOCK_TYPE.CODE).wrapper,
+            };
+        }
+
         blockTypes.filter(block => block.element)
             .forEach((block) => {
                 renderMap[block.type] = { element: block.element };

diff --git a/lib/api/behavior.test.js b/lib/api/behavior.test.js
@@ -24,6 +24,20 @@ describe('behavior', () => {
                 { type: 'TEST' },
             ]).get('TEST')).not.toBeDefined();
         });
+
+        describe('code block element', () => {
+            it('default is "code"', () => {
+                expect(behavior.getBlockRenderMap([
+                    { type: BLOCK_TYPE.CODE },
+                ]).get(BLOCK_TYPE.CODE).element).toEqual('code');
+            });
+
+            it('can be overriden', () => {
+                expect(behavior.getBlockRenderMap([
+                    { type: BLOCK_TYPE.CODE, element: 'span' },
+                ]).get(BLOCK_TYPE.CODE).element).toEqual('span');
+            });
+        });
     });
 
     describe('#getBlockStyleFn', () => {