minor enhancement of the code and increase code coverage to 100%

YahooArchive · Jun 24, 2015 · 61aa76f · 61aa76f
1 parent 02b2cee
commit 61aa76f
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 25 deletions.
diff --git a/src/html-purify.js b/src/html-purify.js
@@ -29,6 +29,7 @@ var CssParser = require('css-js');
         this.cssParser = new CssParser({"ver": "strict", "throwError": false});
     }
 
+    // TODO: introduce polyfill for Array.indexOf
     function contains(arr, element) {
         for (var i = 0, len = arr.length; i < len; i++) {
             if (arr[i] === element) {
@@ -55,14 +56,11 @@ var CssParser = require('css-js');
         case 1:
 	    this.output += ch;
 	    break;
+
         case 2:
-            if (prevState === 35) {
-                this.attrVals[attributeName] = attributeValue;
-            }
-            if (prevState === 36) {
-                this.attrVals[attributeName] = attributeValue;
-            }
-            if (prevState === 40) {
+            if (prevState === 35 ||
+                prevState === 36 || 
+                prevState === 40) {
                 this.attrVals[attributeName] = attributeValue;
             }
 
@@ -121,7 +119,6 @@ var CssParser = require('css-js');
         }
     }
 
-
     Purifier.prototype.purify = function (data) {
         this.output = '';
         this.attrVals = {};

diff --git a/src/tag-attr-list.js b/src/tag-attr-list.js
@@ -121,7 +121,6 @@ Whitelist.Tags = [
     "wbr"
 ];
 
-
 Whitelist.Attributes = [
     "ept",
     "acceptCharset",
@@ -222,22 +221,22 @@ Whitelist.Attributes = [
 ];
 
 Whitelist.HrefAttributes = [
-"action",
-"background",
-"codebase",
-"cite",
-"classid",
-"formaction",
-"folder",
-"href",
-"icon",
-"longdesc",
-"manifest",
-"profile",
-"poster",
-"src",
-"usemap",
-"xlink:href"
+    "action",
+    "background",
+    "codebase",
+    "cite",
+    "classid",
+    "formaction",
+    "folder",
+    "href",
+    "icon",
+    "longdesc",
+    "manifest",
+    "profile",
+    "poster",
+    "src",
+    "usemap",
+    "xlink:href"
 ];
 
 module.exports = Whitelist;