test(sortedHtml): fix comment support in sortedHtml helper

1 files changed, 3 insertions, 0 deletions

diff --git a/test/testabilityPatch.js b/test/testabilityPatch.js
index cef00993..dd0f4f97 100644
--- a/test/testabilityPatch.js
+++ b/test/testabilityPatch.js
@@ -91,11 +91,14 @@ function dealoc(obj) {
 function sortedHtml(element, showNgClass) {
   var html = "";
   forEach(jqLite(element), function toString(node) {
+
     if (node.nodeName == "#text") {
       html += node.nodeValue.
         replace(/&(\w+[&;\W])?/g, function(match, entity){return entity?match:'&';}).
         replace(/</g, '&lt;').
         replace(/>/g, '&gt;');
+    } else if (node.nodeName == "#comment") {
+      html += '<!--' + node.nodeValue + '-->';
     } else {
       html += '<' + (node.nodeName || '?NOT_A_NODE?').toLowerCase();
       var attributes = node.attributes || [];