Allow HTML path element part to match multiple tag names

3 files changed, 40 insertions, 6 deletions

diff --git a/lib/html-generator.js b/lib/html-generator.js
index 3705767..62c58a8 100644
--- a/lib/html-generator.js
+++ b/lib/html-generator.js
@@ -47,7 +47,7 @@ function HtmlGenerator() {
     }
     
     function stylePartMatchesStackElement(stylePart, stackElement) {
-        return stackElement && stylePart.tagName === stackElement.tagName;
+        return stackElement && stylePart.matchesElement(stackElement);
     }
     
     function open(htmlPathElement) {
diff --git a/lib/html-paths.js b/lib/html-paths.js
index 1d3b1f4..9ecb6dd 100644
--- a/lib/html-paths.js
+++ b/lib/html-paths.js
@@ -20,10 +20,29 @@ function elements(elementStyles) {
 
 function element(tagName, attributes, options) {
     options = options || {};
-    return {
-        tagName: tagName,
-        attributes: attributes || {},
-        fresh: options.fresh
-    };
+    return new Element(tagName, attributes, options);
 }
 
+function Element(tagName, attributes, options) {
+    var tagNames = {};
+    if (_.isArray(tagName)) {
+        tagName.forEach(function(tagName) {
+            tagNames[tagName] = true;
+        });
+        tagName = tagName[0];
+    } else {
+        tagNames[tagName] = true;
+    }
+    
+    this.tagName = tagName;
+    this.tagNames = tagNames;
+    this.attributes = attributes || {};
+    this.fresh = options.fresh;
+}
+    
+Element.prototype.matchesElement = function(element) {
+    return this.tagNames[element.tagName];
+}
+
+
+
diff --git a/tests/html-paths.tests.js b/tests/html-paths.tests.js
new file mode 100644
index 0000000..7e3ed86
--- /dev/null
+++ b/tests/html-paths.tests.js
@@ -0,0 +1,15 @@
+var assert = require("assert");
+
+var test = require("./testing").test;
+var htmlPaths = require("../lib/html-paths");
+
+
+describe("html-paths", function() {
+    test("element can match multiple tag names", function() {
+        var pathPart = htmlPaths.element(["ul", "ol"]);
+        assert.ok(pathPart.matchesElement({tagName: "ul"}));
+        assert.ok(pathPart.matchesElement({tagName: "ol"}));
+        assert.ok(!pathPart.matchesElement({tagName: "p"}));
+    });
+});
+