Added caption support to code card renderer

core/server/lib/mobiledoc/cards/code.js

@@ -21,6 +21,18 @@ module.exports = createCard({
         code.appendChild(dom.createTextNode(payload.code));
         pre.appendChild(code);
 
-        return pre;
+        if (payload.caption) {
+            let figure = dom.createElement('figure');
+            figure.setAttribute('class', 'kg-card kg-code-card');
+            figure.appendChild(pre);
+
+            let figcaption = dom.createElement('figcaption');
+            figcaption.appendChild(dom.createRawHTMLSection(payload.caption));
+            figure.appendChild(figcaption);
+
+            return figure;
+        } else {
+            return pre;
+        }
     }
 });

core/test/unit/lib/mobiledoc/cards/code_spec.js

@@ -43,4 +43,19 @@ describe('Code card', function () {
 
         serializer.serialize(card.render(opts)).should.match('');
     });
+
+    it('Renders a figure if a caption is provided', function () {
+        let opts = {
+            env: {
+                dom: new SimpleDom.Document()
+            },
+            payload: {
+                code: '<p>Test</p>',
+                language: 'html',
+                caption: 'Some <strong>HTML</strong>'
+            }
+        };
+
+        serializer.serialize(card.render(opts)).should.match('<!--kg-card-begin: code--><figure class="kg-card kg-code-card"><pre><code class="language-html">&lt;p&gt;Test&lt;/p&gt;</code></pre><figcaption>Some <strong>HTML</strong></figcaption></figure><!--kg-card-end: code-->');
+    });
 });