Added regression test and functionality to not collapse whitespace fo…

…r heading tags
jserme · Sep 7, 2016 · c43001c · c43001c
1 parent 8b50386
commit c43001c
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/index.js b/index.js
@@ -213,12 +213,15 @@ function htmlmin (str, options) {
       }
 
       if (options.collapseWhitespace) {
+
+        var nonCollapsedWhiteSpaceElements = ['p', 'h1', 'h2', 'h3', 'h4', 'h5', 'h6']
+
         // left trim
         if (lastTag) {
           text = text.replace(/^[\n\s\t\r]+/im, '')
         }
 
-        if (lastTag !== 'p' && lastTag) {
+        if (nonCollapsedWhiteSpaceElements.indexOf(lastTag) == -1 && lastTag) {
           text = text.replace(/[\n\s\t\r]+$/im, '')
         }
 

diff --git a/test/allSpec.js b/test/allSpec.js
@@ -103,6 +103,9 @@ describe('htmlmin', function () {
       'u'
     ]
     list.forEach(function (el) {
+      assert.equal(htmlmin('<h2>foo <' + el + '>baz</' + el + '> bar</h2>', {
+        collapseWhitespace: true
+      }), '<h2>foo <' + el + '>baz</' + el + '> bar</h2>')
       assert.equal(htmlmin('<p>foo <' + el + '>baz</' + el + '> bar</p>', {
         collapseWhitespace: true
       }), '<p>foo <' + el + '>baz</' + el + '> bar</p>')