Pull out function for comparing style parts with HTML elements

1 files changed, 5 insertions, 1 deletions

diff --git a/lib/html-generator.js b/lib/html-generator.js
index 5f2befa..8a67e53 100644
--- a/lib/html-generator.js
+++ b/lib/html-generator.js
@@ -24,7 +24,7 @@ function HtmlGenerator() {
         });
         
         var partIndex = 0;
-        while (partIndex < firstFreshPartIndex && style[partIndex].tagName === stack[partIndex]) {
+        while (partIndex < firstFreshPartIndex && stylePartMatchesHtmlElement(style[partIndex], stack[partIndex])) {
             partIndex++;
         }
         while (stack.length > partIndex) {
@@ -35,6 +35,10 @@ function HtmlGenerator() {
         }
     }
     
+    function stylePartMatchesHtmlElement(stylePart, htmlElement) {
+        return stylePart.tagName === htmlElement;
+    }
+    
     function push(element) {
         stack.push(element);
         fragments.push(util.format("<%s>", element));