Add styleName to document matcher

1 files changed, 2 insertions, 0 deletions

diff --git a/lib/document-matchers.js b/lib/document-matchers.js
index 9f138ea..5075172 100644
--- a/lib/document-matchers.js
+++ b/lib/document-matchers.js
@@ -16,6 +16,7 @@ function Matcher(elementType, options) {
     options = options || {};
     this._elementType = elementType;
     this._styleId = options.styleId;
+    this._styleName = options.styleName;
     if (options.list) {
         this._listIndex = options.list.levelIndex;
         this._listIsOrdered = options.list.isOrdered;
@@ -25,6 +26,7 @@ function Matcher(elementType, options) {
 Matcher.prototype.matches = function(element) {
     return element.type === this._elementType &&
         (this._styleId === undefined || element.styleId === this._styleId) &&
+        (this._styleName === undefined || element.styleName === this._styleName) &&
         (this._listIndex === undefined || isList(element, this._listIndex, this._listIsOrdered));
 };