[New] Add hasClass method

Author: eddyerburgh

docs/SUMMARY.md

@@ -6,6 +6,7 @@
     * [contains](/api/mount/contains.md) 
     * [find](/api/mount/find.md)
     * [hasAttribute](/api/mount/hasAttribute.md)
+    * [hasClass](/api/mount/hasClass.md)
     * [update](/api/mount/update.md)
     * [setData](/api/mount/setData.md)
     * [setProps](/api/mount/setProps.md)

docs/api/mount/hasClass.md

@@ -0,0 +1,21 @@
+# hasClass(className)
+
+Check if wrapper DOM node has a class name. Returns a boolean.
+
+### Arguments
+
+`className` (`String`): class name to assert element contains.
+
+### Returns
+
+(`Boolean`): `true` if element contains class. `false` otherwise.
+
+## Example
+
+```js
+import { mount } from 'avoriaz';
+import Foo from './Foo.vue';
+
+const wrapper = mount(Foo);
+expect(wrapper.hasClass('bar')).to.equal(true);
+```

src/Wrapper.js

@@ -50,6 +50,20 @@ export default class Wrapper {
     return this.element.getAttribute(attribute) === value
   }
 
+  /**
+   * Asserts wrapper has a class name
+   *
+   * @param {String} className - class name to assert
+   * @returns {Boolean}
+   */
+  hasClass (className) {
+    if (typeof className !== 'string') {
+      throw new Error('wrapper.hasClass() must be passed a string')
+    }
+
+    return this.element.className.split(' ').indexOf(className) !== -1
+  }
+
   /**
    * Finds every node in the mount tree of the current wrapper that matches the provided selector.
    *

src/WrapperArray.js

@@ -16,6 +16,10 @@ export default class WrapperArray {
     return this.wrappers.every(wrapper => wrapper.hasAttribute(attribute, value))
   }
 
+  hasClass (className) {
+    return this.wrappers.every(wrapper => wrapper.hasClass(className))
+  }
+
   find (selector) {
     if (this.wrappers.length === 0) {
       throw new Error('find cannot be called on 0 items')

test/unit/specs/WrapperArray.spec.js

@@ -58,6 +58,19 @@ describe('WrapperArray', () => {
     expect(wrapperArray.hasAttribute('attribute', 'value')).to.equal(false)
   })
 
+  it('hasClass returns true if every wrapper.hasClass() returns true', () => {
+    const className = 'class'
+    const hasClass = sinon.stub()
+    hasClass.withArgs(className).returns(true)
+    const wrapperArray = new WrapperArray([{ hasClass }, { hasClass }])
+    expect(wrapperArray.hasClass(className)).to.equal(true)
+  })
+
+  it('hasClass returns false if not every wrapper.hasClass() returns true', () => {
+    const wrapperArray = new WrapperArray([{ hasClass: () => true }, { hasClass: () => false }])
+    expect(wrapperArray.hasClass('class')).to.equal(false)
+  })
+
   it('setData calls setData on each wrapper', () => {
     const setData = sinon.stub()
     const data = {}

test/unit/specs/mount/hasClass.spec.js

@@ -0,0 +1,28 @@
+import { compileToFunctions } from 'vue-template-compiler'
+import mount from '../../../../src/mount'
+
+describe('hasClass', () => {
+  it('returns true if wrapper has class name', () => {
+    const compiled = compileToFunctions('<div class="a-class" />')
+    const wrapper = mount(compiled)
+    expect(wrapper.hasClass('a-class')).to.equal(true)
+  })
+
+  it('returns false if wrapper does not have class name', () => {
+    const compiled = compileToFunctions('<div />')
+    const wrapper = mount(compiled)
+    expect(wrapper.hasClass('not-class-name')).to.equal(false)
+  })
+
+  it('throws an error if selector is not a string', () => {
+    const compiled = compileToFunctions('<div />')
+    const wrapper = mount(compiled)
+    const invalidSelectors = [
+      undefined, null, NaN, 0, 2, true, false, () => {}, {}, []
+    ]
+    invalidSelectors.forEach((invalidSelector) => {
+      const message = 'wrapper.hasClass() must be passed a string'
+      expect(() => wrapper.hasClass(invalidSelector)).to.throw(Error, message)
+    })
+  })
+})