add tests shallow.hasClass()

enzymejs · Mar 18, 2019 · 27ea407 · 27ea407
1 parent c193b50
commit 27ea407
Showing 1 changed file with 38 additions and 0 deletions.
diff --git a/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx b/packages/enzyme-test-suite/test/ShallowWrapper-spec.jsx
@@ -8739,4 +8739,42 @@ describe('shallow', () => {
       expect(spy).to.have.property('callCount', 1);
     });
   });
+
+  describe('.hasClass()', () => {
+    it('works for standalone classNames', () => {
+      const wrapper = shallow(<div className="foo" />);
+      expect(wrapper.hasClass('foo')).to.equal(true);
+      expect(wrapper.hasClass('bar')).to.equal(false);
+    });
+
+    it('works for multiple classNames', () => {
+      const wrapper = shallow(<div className="foo bar baz" />);
+      expect(wrapper.hasClass('foo')).to.equal(true);
+      expect(wrapper.hasClass('bar')).to.equal(true);
+      expect(wrapper.hasClass('baz')).to.equal(true);
+      expect(wrapper.hasClass('bax')).to.equal(false);
+    });
+
+    it('also allows hyphens', () => {
+      const wrapper = shallow(<div className="foo-bar" />);
+      expect(wrapper.hasClass('foo-bar')).to.equal(true);
+    });
+
+    it('works if className has a function in toString property', () => {
+      function classes() {}
+      classes.toString = () => 'foo-bar';
+      const wrapper = shallow(<div className={classes} />);
+      expect(wrapper.hasClass('foo-bar')).to.equal(true);
+    });
+
+    it('works if searching with a RegExp', () => {
+      const wrapper = shallow(<div className="ComponentName-classname-123" />);
+      expect(wrapper.hasClass(/(ComponentName)-(classname)-(\d+)/)).to.equal(true);
+    });
+
+    it('fails if searching for a missing classname with a RegExp', () => {
+      const wrapper = shallow(<div className="ComponentName-classname-123 ComponentName-otherclassname-23" />);
+      expect(wrapper.hasClass(/(ComponentName)-(other)-(\d+)/)).to.equal(false);
+    });
+  });
 });