refactor: check for element in hasClass

Author: eddyerburgh

src/wrappers/wrapper.js

@@ -77,7 +77,8 @@ export default class Wrapper implements BaseWrapper {
       throwError('wrapper.hasClass() must be passed a string')
     }
 
-    return this.element.className.split(' ').indexOf(className) !== -1
+    return !!(this.element &&
+    this.element.className.split(' ').indexOf(className) !== -1)
   }
 
   /**

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

@@ -14,6 +14,19 @@ describe('hasClass', () => {
     expect(wrapper.hasClass('not-class-name')).to.equal(false)
   })
 
+  it('returns false if wrapper includes class name in string, but not as a seperate class', () => {
+    const compiled = compileToFunctions('<div class="class-name-together"/>')
+    const wrapper = mount(compiled)
+    expect(wrapper.hasClass('class-name')).to.equal(false)
+  })
+
+  it('returns false if wrapper does not have an element', () => {
+    const compiled = compileToFunctions('<div />')
+    const wrapper = mount(compiled)
+    wrapper.element = null
+    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)