Merge pull request #18 from testdouble/2-prototype

Test prototype-based create()

Author: searls

lib/create.coffee

@@ -3,11 +3,26 @@ store = require('./store')
 calls = require('./store/calls')
 stubbings = require('./store/stubbings')
 
-module.exports = (name) ->
+module.exports = (nameOrType) ->
+  if nameOrType?.prototype?
+    createTestDoublesForEntireType(nameOrType)
+  else
+    createTestDouble(nameOrType)
+
+createTestDouble = (name) ->
   _.tap createTestDoubleFunction(), (testDouble) ->
     if name? then store.for(testDouble).name = name
 
 createTestDoubleFunction = ->
   testDouble = (args...) ->
     calls.log(testDouble, args, this)
     stubbings.get(testDouble, args)
+
+createTestDoublesForEntireType = (type) ->
+  _.reduce Object.getOwnPropertyNames(type.prototype), (memo, name) ->
+    memo[name] = if _.isFunction(type.prototype[name])
+      createTestDouble("#{type.name}##{name}")
+    else
+      type.prototype[name]
+    memo
+  , {}

test/lib/verify-test.coffee

@@ -34,6 +34,16 @@ describe '.verify', ->
     When -> @result = (shouldThrow => @verify(@testDouble()))
     Then -> expect(@result).to.contain("verification on test double `#footime`.")
 
+  context 'with a prototype-modeling double', ->
+    Given -> @SomeType = `function Foo() {}`
+    Given -> @SomeType::bar = ->
+    Given -> @SomeType::baz = ->
+    Given -> @SomeType::biz = "not a function!"
+    Given -> @testDoubleObj = @create(@SomeType)
+    When -> @result = (shouldThrow => @verify(@testDoubleObj.baz()))
+    Then -> expect(@result).to.contain("verification on test double `Foo#baz`.")
+    Then -> @testDoubleObj.biz == "not a function!"
+
   context 'a double-free verification error', ->
     Then -> shouldThrow (=> @verify()), """
       No test double invocation detected for `verify()`.