Merge pull request #54 from SevenOutman/feat/class-regex

Add regex support for `.class` assertion

README.md

@@ -29,6 +29,13 @@ document.getElementsByName('bar').should.have.class('foo')
 expect(document.querySelector('main article')).to.have.class('foo')
 ```
 
+Also accepts regex as argument.
+
+```js
+document.getElementsByName('bar').should.have.class(/foo/)
+expect(document.querySelector('main article')).to.have.class(/foo/)
+```
+
 ### `id(id)`
 Assert that the [HTMLElement][] has the given id.
 

chai-dom.js

@@ -79,6 +79,16 @@
 
   chai.Assertion.addMethod('class', function(className) {
     var el = flag(this, 'object')
+
+    if (className instanceof RegExp) {
+      return this.assert(
+        Array.from(el.classList).some(function(cls) { return className.test(cls) })
+        , 'expected ' + elToString(el) + ' to have class matching #{exp}'
+        , 'expected ' + elToString(el) + ' not to have class matching #{exp}'
+        , className
+      )
+    } 
+
     this.assert(
       el.classList.contains(className)
       , 'expected ' + elToString(el) + ' to have class #{exp}'

test/tests.js

@@ -132,21 +132,41 @@ describe('DOM assertions', function() {
       subject.should.have.class('baz');
     })
 
+    it('passes when the element has the class matching given regex', function () {
+      subject.should.have.class(/foo/)
+    })
+
     it('passes negated when the element does not have the class', function() {
       subject.should.not.have.class('bar')
     })
 
+    it('passes negated when the element does not have the class matching given regex', function () {
+      subject.should.not.have.class(/bar/)
+    })
+
     it('fails when the element does not have the class', function() {
       (function() {
         subject.should.have.class('bar')
       }).should.fail('expected div.foo.shazam.baz to have class \'bar\'')
     })
 
+    it('fails when the element does not have the class matching given regex', function() {
+      (function() {
+        subject.should.have.class(/bar/)
+      }).should.fail('expected div.foo.shazam.baz to have class matching /bar/')
+    })
+
     it('fails negated when the element has the class', function() {
       (function() {
         subject.should.not.have.class('foo')
       }).should.fail('expected div.foo.shazam.baz not to have class \'foo\'')
     })
+
+    it('fails negated when the element has the class matching given regex', function() {
+      (function() {
+        subject.should.not.have.class(/foo/)
+      }).should.fail('expected div.foo.shazam.baz not to have class matching /foo/')
+    })
   })
 
   describe('id', function() {