Add regex support for class assertion

nathanboktae · Sep 18, 2021 · 0ad45ed · 0ad45ed
1 parent 7a83d08
commit 0ad45ed
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 0 deletions.
diff --git a/chai-dom.js b/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}'

diff --git a/test/tests.js b/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() {