When <center> tag is processed, add the class 'center' instead of ove…

…rwriting the whole class attribute
foundation · Mar 7, 2016 · 6a68c1b · 6a68c1b
1 parent dc79e9b
commit 6a68c1b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/lib/componentFactory.js b/lib/componentFactory.js
@@ -80,9 +80,9 @@ module.exports = function(element) {
     // <center>
     case this.components.center:
       if (element.children().length > 0) {
-        element.children().attr({
-          align: 'center', class: 'center'
-        });
+        element.children()
+          .attr('align', 'center')
+          .addClass('center');
       }
 
       element.attr('data-parsed', '');

diff --git a/test/components.js b/test/components.js
@@ -159,7 +159,7 @@ describe('Callout', () => {
     var expected = `
       <table>
         <tr>
-          <th class="callout"></th>
+          <td class="callout"></td>
         </tr>
       </table>
     `;
@@ -172,7 +172,7 @@ describe('Callout', () => {
     var expected = `
       <table>
         <tr>
-          <th class="callout primary"></th>
+          <td class="callout primary"></td>
         </tr>
       </table>
     `;