feat(options.stub): add option to pass stubs as an array

eddyerburgh · eddyerburgh · commit 6576d6e9414b · 2017-07-16T15:17:15.000+01:00
diff --git a/src/lib/stub-components.js b/src/lib/stub-components.js
@@ -61,41 +61,50 @@ function createBlankStub (originalComponent: Component) {
 }
 
 export function stubComponents (component: Component, stubs: Object): void {
-  Object.keys(stubs).forEach(stub => {
-    if (!isValidStub(stubs[stub])) {
-      throwError('options.stub values must be passed a string or component')
-    }
-
-    if (!component.components) {
-      component.components = {}
-    }
+  if (!component.components) {
+    component.components = {}
+  }
 
-    if (component.components[stub]) {
-        // Remove cached constructor
-      delete component.components[stub]._Ctor
-      if (typeof stubs[stub] === 'string') {
-        component.components[stub] = createStubFromString(stubs[stub], component.components[stub])
-        stubLifeCycleEvents(component.components[stub])
-      } else {
-        component.components[stub] = {
-          ...stubs[stub],
-          name: component.components[stub].name
-        }
+  if (Array.isArray(stubs)) {
+    stubs.forEach(stub => {
+      if (typeof stub !== 'string') {
+        throwError('each item in options.stub must be a string')
+      }
+      component.components[stub] = createBlankStub({})
+    })
+  } else {
+    Object.keys(stubs).forEach(stub => {
+      if (!isValidStub(stubs[stub])) {
+        throwError('options.stub values must be passed a string or component')
       }
-    } else {
-      if (typeof stubs[stub] === 'string') {
-        component.components[stub] = {
-          ...compileToFunctions(stubs[stub])
+
+      if (component.components[stub]) {
+              // Remove cached constructor
+        delete component.components[stub]._Ctor
+        if (typeof stubs[stub] === 'string') {
+          component.components[stub] = createStubFromString(stubs[stub], component.components[stub])
+          stubLifeCycleEvents(component.components[stub])
+        } else {
+          component.components[stub] = {
+            ...stubs[stub],
+            name: component.components[stub].name
+          }
         }
-        stubLifeCycleEvents(component.components[stub])
       } else {
-        component.components[stub] = {
-          ...stubs[stub]
+        if (typeof stubs[stub] === 'string') {
+          component.components[stub] = {
+            ...compileToFunctions(stubs[stub])
+          }
+          stubLifeCycleEvents(component.components[stub])
+        } else {
+          component.components[stub] = {
+            ...stubs[stub]
+          }
         }
       }
-    }
-    Vue.config.ignoredElements.push(stub)
-  })
+      Vue.config.ignoredElements.push(stub)
+    })
+  }
 }
 
 export function stubAllComponents (component: Component): void {
diff --git a/test/integration/specs/mount/options/stub.spec.js b/test/integration/specs/mount/options/stub.spec.js
@@ -62,10 +62,36 @@ describe('mount.stub', () => {
     expect(wrapper.findAll(Component).length).to.equal(1)
   })
 
-  const invalidValues = [1, null, [], {}, NaN]
-  invalidValues.forEach(invalidValue => {
-    it('throws an error when passed an invalid value as stub', () => {
-      const error = '[vue-test-utils]: options.stub values must be passed a string or component'
+  it('stubs components with dummy when passed as an array', () => {
+    const warn = sinon.stub(console, 'error')
+    const ComponentWithGlobalComponent = {
+      render: h => h('registered-component')
+    }
+    mount(ComponentWithGlobalComponent, {
+      stub: ['registered-component']
+    })
+
+    expect(warn.called).to.equal(false)
+    warn.restore()
+  })
+  it('stubs components with dummy when passed as an array', () => {
+    const ComponentWithGlobalComponent = {
+      render: h => h('registered-component')
+    }
+    const invalidValues = [{}, [], 3]
+    const error = '[vue-test-utils]: each item in options.stub must be a string'
+    invalidValues.forEach(invalidValue => {
+      const fn = () => mount(ComponentWithGlobalComponent, {
+        stub: [invalidValue]
+      })
+      expect(fn).to.throw().with.property('message', error)
+    })
+  })
+
+  it('throws an error when passed an invalid value as stub', () => {
+    const error = '[vue-test-utils]: options.stub values must be passed a string or component'
+    const invalidValues = [1, null, [], {}, NaN]
+    invalidValues.forEach(invalidValue => {
       const fn = () => mount(ComponentWithChildComponent, {
         stub: {
           ChildComponent: invalidValue
