From 159866f0321aac0e7ebf5635de5528b3a0b32ce2 Mon Sep 17 00:00:00 2001
From: mAAdhaTTah <jamesorodig@gmail.com>
Date: Tue, 21 Nov 2017 16:40:35 -0500
Subject: [PATCH] Test case showing `code` element is blocks & spans

---
 src/__snapshots__/index.test.js.snap |  8 ++++++++
 src/index.test.js                    | 18 ++++++++++++++++++
 2 files changed, 26 insertions(+)

diff --git a/src/__snapshots__/index.test.js.snap b/src/__snapshots__/index.test.js.snap
index a46d0a0..c951318 100644
--- a/src/__snapshots__/index.test.js.snap
+++ b/src/__snapshots__/index.test.js.snap
@@ -29,6 +29,14 @@ exports[`should allow overriding code element with components version 1`] = `
 </div>
 `;
 
+exports[`should allow overriding codeblock element with components version 1`] = `
+<div>
+  <div>
+    code
+  </div>
+</div>
+`;
+
 exports[`should be able to combine in compilation 1`] = `
 <div>
   <h1
diff --git a/src/index.test.js b/src/index.test.js
index 835711f..25670e2 100644
--- a/src/index.test.js
+++ b/src/index.test.js
@@ -398,6 +398,24 @@ it('should allow overriding code element with components version', () => {
   expect(tree).toMatchSnapshot();
 });
 
+it('should allow overriding codeblock element with components version', () => {
+    const compile = marksyComponents({
+        createElement,
+        elements: {
+            code() {
+                return <div>code</div>
+            }
+        }
+    });
+    const compiled = compile('```js\nconst foo = "bar"\n```');
+
+    const tree = renderer.create(
+        <TestComponent>{compiled.tree}</TestComponent>
+    ).toJSON();
+
+    expect(tree).toMatchSnapshot();
+});
+
 it('should highlight code with highlight.js', () => {
   const compile = marksy({
     createElement,