Fix #71 Do + DoAndReturn signature change error msg

Update the error handling for Call.Do and Call.DoAndReturn in the case where
the argument passed does not match expectations.

* panic if the argument is not a function
* panic if the number of input arguments do not match those expected by Call
* panic if the types of the input arguments do not match those expected
by Call

Call.DoAndReturn has additional validations on the return signature

* panic if the number of return arguments do not match those expected by
Call
* panic if the types of return arguments do not match those expected by
Call

gomock/call.go

@@ -19,6 +19,8 @@ import (
  "reflect"
  "strconv"
  "strings"
+
+ "github.com/golang/mock/gomock/internal/validate"
 )
 
 // Call represents an expected call to a mock.
@@ -106,9 +108,20 @@ func (c *Call) MaxTimes(n int) *Call {
 // The return values from this function are returned by the mocked function.
 // It takes an interface{} argument to support n-arity functions.
 func (c *Call) DoAndReturn(f interface{}) *Call {
- // TODO: Check arity and types here, rather than dying badly elsewhere.
  v := reflect.ValueOf(f)
 
+ switch v.Kind() {
+ case reflect.Func:
+ mt := c.methodType
+
+ ft := v.Type()
+ if err := validate.InputAndOutputSig(ft, mt); err != nil {
+ panic(fmt.Sprintf("DoAndReturn: %s", err))
+ }
+ default:
+ panic("DoAndReturn: argument must be a function")
+ }
+
  c.addAction(func(args []interface{}) []interface{} {
  vargs := make([]reflect.Value, len(args))
  ft := v.Type()
@@ -135,9 +148,20 @@ func (c *Call) DoAndReturn(f interface{}) *Call {
 // return values call DoAndReturn.
 // It takes an interface{} argument to support n-arity functions.
 func (c *Call) Do(f interface{}) *Call {
- // TODO: Check arity and types here, rather than dying badly elsewhere.
  v := reflect.ValueOf(f)
 
+ switch v.Kind() {
+ case reflect.Func:
+ mt := c.methodType
+
+ ft := v.Type()
+ if err := validate.InputSig(ft, mt); err != nil {
+ panic(fmt.Sprintf("Do: %s", err))
+ }
+ default:
+ panic("Do: argument must be a function")
+ }
+
  c.addAction(func(args []interface{}) []interface{} {
  vargs := make([]reflect.Value, len(args))
  ft := v.Type()