add csi-test to vendor
This commit is contained in:
428
vendor/github.com/golang/mock/gomock/call.go
generated
vendored
Normal file
428
vendor/github.com/golang/mock/gomock/call.go
generated
vendored
Normal file
@@ -0,0 +1,428 @@
|
||||
// Copyright 2010 Google Inc.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
package gomock
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"reflect"
|
||||
"strconv"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// Call represents an expected call to a mock.
|
||||
type Call struct {
|
||||
t TestReporter // for triggering test failures on invalid call setup
|
||||
|
||||
receiver interface{} // the receiver of the method call
|
||||
method string // the name of the method
|
||||
methodType reflect.Type // the type of the method
|
||||
args []Matcher // the args
|
||||
origin string // file and line number of call setup
|
||||
|
||||
preReqs []*Call // prerequisite calls
|
||||
|
||||
// Expectations
|
||||
minCalls, maxCalls int
|
||||
|
||||
numCalls int // actual number made
|
||||
|
||||
// actions are called when this Call is called. Each action gets the args and
|
||||
// can set the return values by returning a non-nil slice. Actions run in the
|
||||
// order they are created.
|
||||
actions []func([]interface{}) []interface{}
|
||||
}
|
||||
|
||||
// newCall creates a *Call. It requires the method type in order to support
|
||||
// unexported methods.
|
||||
func newCall(t TestReporter, receiver interface{}, method string, methodType reflect.Type, args ...interface{}) *Call {
|
||||
if h, ok := t.(testHelper); ok {
|
||||
h.Helper()
|
||||
}
|
||||
|
||||
// TODO: check arity, types.
|
||||
margs := make([]Matcher, len(args))
|
||||
for i, arg := range args {
|
||||
if m, ok := arg.(Matcher); ok {
|
||||
margs[i] = m
|
||||
} else if arg == nil {
|
||||
// Handle nil specially so that passing a nil interface value
|
||||
// will match the typed nils of concrete args.
|
||||
margs[i] = Nil()
|
||||
} else {
|
||||
margs[i] = Eq(arg)
|
||||
}
|
||||
}
|
||||
|
||||
origin := callerInfo(3)
|
||||
actions := []func([]interface{}) []interface{}{func([]interface{}) []interface{} {
|
||||
// Synthesize the zero value for each of the return args' types.
|
||||
rets := make([]interface{}, methodType.NumOut())
|
||||
for i := 0; i < methodType.NumOut(); i++ {
|
||||
rets[i] = reflect.Zero(methodType.Out(i)).Interface()
|
||||
}
|
||||
return rets
|
||||
}}
|
||||
return &Call{t: t, receiver: receiver, method: method, methodType: methodType,
|
||||
args: margs, origin: origin, minCalls: 1, maxCalls: 1, actions: actions}
|
||||
}
|
||||
|
||||
// AnyTimes allows the expectation to be called 0 or more times
|
||||
func (c *Call) AnyTimes() *Call {
|
||||
c.minCalls, c.maxCalls = 0, 1e8 // close enough to infinity
|
||||
return c
|
||||
}
|
||||
|
||||
// MinTimes requires the call to occur at least n times. If AnyTimes or MaxTimes have not been called, MinTimes also
|
||||
// sets the maximum number of calls to infinity.
|
||||
func (c *Call) MinTimes(n int) *Call {
|
||||
c.minCalls = n
|
||||
if c.maxCalls == 1 {
|
||||
c.maxCalls = 1e8
|
||||
}
|
||||
return c
|
||||
}
|
||||
|
||||
// MaxTimes limits the number of calls to n times. If AnyTimes or MinTimes have not been called, MaxTimes also
|
||||
// sets the minimum number of calls to 0.
|
||||
func (c *Call) MaxTimes(n int) *Call {
|
||||
c.maxCalls = n
|
||||
if c.minCalls == 1 {
|
||||
c.minCalls = 0
|
||||
}
|
||||
return c
|
||||
}
|
||||
|
||||
// DoAndReturn declares the action to run when the call is matched.
|
||||
// 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)
|
||||
|
||||
c.addAction(func(args []interface{}) []interface{} {
|
||||
vargs := make([]reflect.Value, len(args))
|
||||
ft := v.Type()
|
||||
for i := 0; i < len(args); i++ {
|
||||
if args[i] != nil {
|
||||
vargs[i] = reflect.ValueOf(args[i])
|
||||
} else {
|
||||
// Use the zero value for the arg.
|
||||
vargs[i] = reflect.Zero(ft.In(i))
|
||||
}
|
||||
}
|
||||
vrets := v.Call(vargs)
|
||||
rets := make([]interface{}, len(vrets))
|
||||
for i, ret := range vrets {
|
||||
rets[i] = ret.Interface()
|
||||
}
|
||||
return rets
|
||||
})
|
||||
return c
|
||||
}
|
||||
|
||||
// Do declares the action to run when the call is matched. The function's
|
||||
// return values are ignored to retain backward compatibility. To use the
|
||||
// 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)
|
||||
|
||||
c.addAction(func(args []interface{}) []interface{} {
|
||||
vargs := make([]reflect.Value, len(args))
|
||||
ft := v.Type()
|
||||
for i := 0; i < len(args); i++ {
|
||||
if args[i] != nil {
|
||||
vargs[i] = reflect.ValueOf(args[i])
|
||||
} else {
|
||||
// Use the zero value for the arg.
|
||||
vargs[i] = reflect.Zero(ft.In(i))
|
||||
}
|
||||
}
|
||||
v.Call(vargs)
|
||||
return nil
|
||||
})
|
||||
return c
|
||||
}
|
||||
|
||||
// Return declares the values to be returned by the mocked function call.
|
||||
func (c *Call) Return(rets ...interface{}) *Call {
|
||||
if h, ok := c.t.(testHelper); ok {
|
||||
h.Helper()
|
||||
}
|
||||
|
||||
mt := c.methodType
|
||||
if len(rets) != mt.NumOut() {
|
||||
c.t.Fatalf("wrong number of arguments to Return for %T.%v: got %d, want %d [%s]",
|
||||
c.receiver, c.method, len(rets), mt.NumOut(), c.origin)
|
||||
}
|
||||
for i, ret := range rets {
|
||||
if got, want := reflect.TypeOf(ret), mt.Out(i); got == want {
|
||||
// Identical types; nothing to do.
|
||||
} else if got == nil {
|
||||
// Nil needs special handling.
|
||||
switch want.Kind() {
|
||||
case reflect.Chan, reflect.Func, reflect.Interface, reflect.Map, reflect.Ptr, reflect.Slice:
|
||||
// ok
|
||||
default:
|
||||
c.t.Fatalf("argument %d to Return for %T.%v is nil, but %v is not nillable [%s]",
|
||||
i, c.receiver, c.method, want, c.origin)
|
||||
}
|
||||
} else if got.AssignableTo(want) {
|
||||
// Assignable type relation. Make the assignment now so that the generated code
|
||||
// can return the values with a type assertion.
|
||||
v := reflect.New(want).Elem()
|
||||
v.Set(reflect.ValueOf(ret))
|
||||
rets[i] = v.Interface()
|
||||
} else {
|
||||
c.t.Fatalf("wrong type of argument %d to Return for %T.%v: %v is not assignable to %v [%s]",
|
||||
i, c.receiver, c.method, got, want, c.origin)
|
||||
}
|
||||
}
|
||||
|
||||
c.addAction(func([]interface{}) []interface{} {
|
||||
return rets
|
||||
})
|
||||
|
||||
return c
|
||||
}
|
||||
|
||||
// Times declares the exact number of times a function call is expected to be executed.
|
||||
func (c *Call) Times(n int) *Call {
|
||||
c.minCalls, c.maxCalls = n, n
|
||||
return c
|
||||
}
|
||||
|
||||
// SetArg declares an action that will set the nth argument's value,
|
||||
// indirected through a pointer. Or, in the case of a slice, SetArg
|
||||
// will copy value's elements into the nth argument.
|
||||
func (c *Call) SetArg(n int, value interface{}) *Call {
|
||||
if h, ok := c.t.(testHelper); ok {
|
||||
h.Helper()
|
||||
}
|
||||
|
||||
mt := c.methodType
|
||||
// TODO: This will break on variadic methods.
|
||||
// We will need to check those at invocation time.
|
||||
if n < 0 || n >= mt.NumIn() {
|
||||
c.t.Fatalf("SetArg(%d, ...) called for a method with %d args [%s]",
|
||||
n, mt.NumIn(), c.origin)
|
||||
}
|
||||
// Permit setting argument through an interface.
|
||||
// In the interface case, we don't (nay, can't) check the type here.
|
||||
at := mt.In(n)
|
||||
switch at.Kind() {
|
||||
case reflect.Ptr:
|
||||
dt := at.Elem()
|
||||
if vt := reflect.TypeOf(value); !vt.AssignableTo(dt) {
|
||||
c.t.Fatalf("SetArg(%d, ...) argument is a %v, not assignable to %v [%s]",
|
||||
n, vt, dt, c.origin)
|
||||
}
|
||||
case reflect.Interface:
|
||||
// nothing to do
|
||||
case reflect.Slice:
|
||||
// nothing to do
|
||||
default:
|
||||
c.t.Fatalf("SetArg(%d, ...) referring to argument of non-pointer non-interface non-slice type %v [%s]",
|
||||
n, at, c.origin)
|
||||
}
|
||||
|
||||
c.addAction(func(args []interface{}) []interface{} {
|
||||
v := reflect.ValueOf(value)
|
||||
switch reflect.TypeOf(args[n]).Kind() {
|
||||
case reflect.Slice:
|
||||
setSlice(args[n], v)
|
||||
default:
|
||||
reflect.ValueOf(args[n]).Elem().Set(v)
|
||||
}
|
||||
return nil
|
||||
})
|
||||
return c
|
||||
}
|
||||
|
||||
// isPreReq returns true if other is a direct or indirect prerequisite to c.
|
||||
func (c *Call) isPreReq(other *Call) bool {
|
||||
for _, preReq := range c.preReqs {
|
||||
if other == preReq || preReq.isPreReq(other) {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
// After declares that the call may only match after preReq has been exhausted.
|
||||
func (c *Call) After(preReq *Call) *Call {
|
||||
if h, ok := c.t.(testHelper); ok {
|
||||
h.Helper()
|
||||
}
|
||||
|
||||
if c == preReq {
|
||||
c.t.Fatalf("A call isn't allowed to be its own prerequisite")
|
||||
}
|
||||
if preReq.isPreReq(c) {
|
||||
c.t.Fatalf("Loop in call order: %v is a prerequisite to %v (possibly indirectly).", c, preReq)
|
||||
}
|
||||
|
||||
c.preReqs = append(c.preReqs, preReq)
|
||||
return c
|
||||
}
|
||||
|
||||
// Returns true if the minimum number of calls have been made.
|
||||
func (c *Call) satisfied() bool {
|
||||
return c.numCalls >= c.minCalls
|
||||
}
|
||||
|
||||
// Returns true iff the maximum number of calls have been made.
|
||||
func (c *Call) exhausted() bool {
|
||||
return c.numCalls >= c.maxCalls
|
||||
}
|
||||
|
||||
func (c *Call) String() string {
|
||||
args := make([]string, len(c.args))
|
||||
for i, arg := range c.args {
|
||||
args[i] = arg.String()
|
||||
}
|
||||
arguments := strings.Join(args, ", ")
|
||||
return fmt.Sprintf("%T.%v(%s) %s", c.receiver, c.method, arguments, c.origin)
|
||||
}
|
||||
|
||||
// Tests if the given call matches the expected call.
|
||||
// If yes, returns nil. If no, returns error with message explaining why it does not match.
|
||||
func (c *Call) matches(args []interface{}) error {
|
||||
if !c.methodType.IsVariadic() {
|
||||
if len(args) != len(c.args) {
|
||||
return fmt.Errorf("Expected call at %s has the wrong number of arguments. Got: %d, want: %d",
|
||||
c.origin, len(args), len(c.args))
|
||||
}
|
||||
|
||||
for i, m := range c.args {
|
||||
if !m.Matches(args[i]) {
|
||||
return fmt.Errorf("Expected call at %s doesn't match the argument at index %s.\nGot: %v\nWant: %v",
|
||||
c.origin, strconv.Itoa(i), args[i], m)
|
||||
}
|
||||
}
|
||||
} else {
|
||||
if len(c.args) < c.methodType.NumIn()-1 {
|
||||
return fmt.Errorf("Expected call at %s has the wrong number of matchers. Got: %d, want: %d",
|
||||
c.origin, len(c.args), c.methodType.NumIn()-1)
|
||||
}
|
||||
if len(c.args) != c.methodType.NumIn() && len(args) != len(c.args) {
|
||||
return fmt.Errorf("Expected call at %s has the wrong number of arguments. Got: %d, want: %d",
|
||||
c.origin, len(args), len(c.args))
|
||||
}
|
||||
if len(args) < len(c.args)-1 {
|
||||
return fmt.Errorf("Expected call at %s has the wrong number of arguments. Got: %d, want: greater than or equal to %d",
|
||||
c.origin, len(args), len(c.args)-1)
|
||||
}
|
||||
|
||||
for i, m := range c.args {
|
||||
if i < c.methodType.NumIn()-1 {
|
||||
// Non-variadic args
|
||||
if !m.Matches(args[i]) {
|
||||
return fmt.Errorf("Expected call at %s doesn't match the argument at index %s.\nGot: %v\nWant: %v",
|
||||
c.origin, strconv.Itoa(i), args[i], m)
|
||||
}
|
||||
continue
|
||||
}
|
||||
// The last arg has a possibility of a variadic argument, so let it branch
|
||||
|
||||
// sample: Foo(a int, b int, c ...int)
|
||||
if i < len(c.args) && i < len(args) {
|
||||
if m.Matches(args[i]) {
|
||||
// Got Foo(a, b, c) want Foo(matcherA, matcherB, gomock.Any())
|
||||
// Got Foo(a, b, c) want Foo(matcherA, matcherB, someSliceMatcher)
|
||||
// Got Foo(a, b, c) want Foo(matcherA, matcherB, matcherC)
|
||||
// Got Foo(a, b) want Foo(matcherA, matcherB)
|
||||
// Got Foo(a, b, c, d) want Foo(matcherA, matcherB, matcherC, matcherD)
|
||||
continue
|
||||
}
|
||||
}
|
||||
|
||||
// The number of actual args don't match the number of matchers,
|
||||
// or the last matcher is a slice and the last arg is not.
|
||||
// If this function still matches it is because the last matcher
|
||||
// matches all the remaining arguments or the lack of any.
|
||||
// Convert the remaining arguments, if any, into a slice of the
|
||||
// expected type.
|
||||
vargsType := c.methodType.In(c.methodType.NumIn() - 1)
|
||||
vargs := reflect.MakeSlice(vargsType, 0, len(args)-i)
|
||||
for _, arg := range args[i:] {
|
||||
vargs = reflect.Append(vargs, reflect.ValueOf(arg))
|
||||
}
|
||||
if m.Matches(vargs.Interface()) {
|
||||
// Got Foo(a, b, c, d, e) want Foo(matcherA, matcherB, gomock.Any())
|
||||
// Got Foo(a, b, c, d, e) want Foo(matcherA, matcherB, someSliceMatcher)
|
||||
// Got Foo(a, b) want Foo(matcherA, matcherB, gomock.Any())
|
||||
// Got Foo(a, b) want Foo(matcherA, matcherB, someEmptySliceMatcher)
|
||||
break
|
||||
}
|
||||
// Wrong number of matchers or not match. Fail.
|
||||
// Got Foo(a, b) want Foo(matcherA, matcherB, matcherC, matcherD)
|
||||
// Got Foo(a, b, c) want Foo(matcherA, matcherB, matcherC, matcherD)
|
||||
// Got Foo(a, b, c, d) want Foo(matcherA, matcherB, matcherC, matcherD, matcherE)
|
||||
// Got Foo(a, b, c, d, e) want Foo(matcherA, matcherB, matcherC, matcherD)
|
||||
// Got Foo(a, b, c) want Foo(matcherA, matcherB)
|
||||
return fmt.Errorf("Expected call at %s doesn't match the argument at index %s.\nGot: %v\nWant: %v",
|
||||
c.origin, strconv.Itoa(i), args[i:], c.args[i])
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
// Check that all prerequisite calls have been satisfied.
|
||||
for _, preReqCall := range c.preReqs {
|
||||
if !preReqCall.satisfied() {
|
||||
return fmt.Errorf("Expected call at %s doesn't have a prerequisite call satisfied:\n%v\nshould be called before:\n%v",
|
||||
c.origin, preReqCall, c)
|
||||
}
|
||||
}
|
||||
|
||||
// Check that the call is not exhausted.
|
||||
if c.exhausted() {
|
||||
return fmt.Errorf("Expected call at %s has already been called the max number of times.", c.origin)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// dropPrereqs tells the expected Call to not re-check prerequisite calls any
|
||||
// longer, and to return its current set.
|
||||
func (c *Call) dropPrereqs() (preReqs []*Call) {
|
||||
preReqs = c.preReqs
|
||||
c.preReqs = nil
|
||||
return
|
||||
}
|
||||
|
||||
func (c *Call) call(args []interface{}) []func([]interface{}) []interface{} {
|
||||
c.numCalls++
|
||||
return c.actions
|
||||
}
|
||||
|
||||
// InOrder declares that the given calls should occur in order.
|
||||
func InOrder(calls ...*Call) {
|
||||
for i := 1; i < len(calls); i++ {
|
||||
calls[i].After(calls[i-1])
|
||||
}
|
||||
}
|
||||
|
||||
func setSlice(arg interface{}, v reflect.Value) {
|
||||
va := reflect.ValueOf(arg)
|
||||
for i := 0; i < v.Len(); i++ {
|
||||
va.Index(i).Set(v.Index(i))
|
||||
}
|
||||
}
|
||||
|
||||
func (c *Call) addAction(action func([]interface{}) []interface{}) {
|
||||
c.actions = append(c.actions, action)
|
||||
}
|
51
vendor/github.com/golang/mock/gomock/call_test.go
generated
vendored
Normal file
51
vendor/github.com/golang/mock/gomock/call_test.go
generated
vendored
Normal file
@@ -0,0 +1,51 @@
|
||||
package gomock
|
||||
|
||||
import (
|
||||
"testing"
|
||||
)
|
||||
|
||||
type mockTestReporter struct {
|
||||
errorCalls int
|
||||
fatalCalls int
|
||||
}
|
||||
|
||||
func (o *mockTestReporter) Errorf(format string, args ...interface{}) {
|
||||
o.errorCalls++
|
||||
}
|
||||
|
||||
func (o *mockTestReporter) Fatalf(format string, args ...interface{}) {
|
||||
o.fatalCalls++
|
||||
}
|
||||
|
||||
func (o *mockTestReporter) Helper() {}
|
||||
|
||||
func TestCall_After(t *testing.T) {
|
||||
t.Run("SelfPrereqCallsFatalf", func(t *testing.T) {
|
||||
tr1 := &mockTestReporter{}
|
||||
|
||||
c := &Call{t: tr1}
|
||||
c.After(c)
|
||||
|
||||
if tr1.fatalCalls != 1 {
|
||||
t.Errorf("number of fatal calls == %v, want 1", tr1.fatalCalls)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("LoopInCallOrderCallsFatalf", func(t *testing.T) {
|
||||
tr1 := &mockTestReporter{}
|
||||
tr2 := &mockTestReporter{}
|
||||
|
||||
c1 := &Call{t: tr1}
|
||||
c2 := &Call{t: tr2}
|
||||
c1.After(c2)
|
||||
c2.After(c1)
|
||||
|
||||
if tr1.errorCalls != 0 || tr1.fatalCalls != 0 {
|
||||
t.Error("unexpected errors")
|
||||
}
|
||||
|
||||
if tr2.fatalCalls != 1 {
|
||||
t.Errorf("number of fatal calls == %v, want 1", tr2.fatalCalls)
|
||||
}
|
||||
})
|
||||
}
|
108
vendor/github.com/golang/mock/gomock/callset.go
generated
vendored
Normal file
108
vendor/github.com/golang/mock/gomock/callset.go
generated
vendored
Normal file
@@ -0,0 +1,108 @@
|
||||
// Copyright 2011 Google Inc.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
package gomock
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
)
|
||||
|
||||
// callSet represents a set of expected calls, indexed by receiver and method
|
||||
// name.
|
||||
type callSet struct {
|
||||
// Calls that are still expected.
|
||||
expected map[callSetKey][]*Call
|
||||
// Calls that have been exhausted.
|
||||
exhausted map[callSetKey][]*Call
|
||||
}
|
||||
|
||||
// callSetKey is the key in the maps in callSet
|
||||
type callSetKey struct {
|
||||
receiver interface{}
|
||||
fname string
|
||||
}
|
||||
|
||||
func newCallSet() *callSet {
|
||||
return &callSet{make(map[callSetKey][]*Call), make(map[callSetKey][]*Call)}
|
||||
}
|
||||
|
||||
// Add adds a new expected call.
|
||||
func (cs callSet) Add(call *Call) {
|
||||
key := callSetKey{call.receiver, call.method}
|
||||
m := cs.expected
|
||||
if call.exhausted() {
|
||||
m = cs.exhausted
|
||||
}
|
||||
m[key] = append(m[key], call)
|
||||
}
|
||||
|
||||
// Remove removes an expected call.
|
||||
func (cs callSet) Remove(call *Call) {
|
||||
key := callSetKey{call.receiver, call.method}
|
||||
calls := cs.expected[key]
|
||||
for i, c := range calls {
|
||||
if c == call {
|
||||
// maintain order for remaining calls
|
||||
cs.expected[key] = append(calls[:i], calls[i+1:]...)
|
||||
cs.exhausted[key] = append(cs.exhausted[key], call)
|
||||
break
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// FindMatch searches for a matching call. Returns error with explanation message if no call matched.
|
||||
func (cs callSet) FindMatch(receiver interface{}, method string, args []interface{}) (*Call, error) {
|
||||
key := callSetKey{receiver, method}
|
||||
|
||||
// Search through the expected calls.
|
||||
expected := cs.expected[key]
|
||||
var callsErrors bytes.Buffer
|
||||
for _, call := range expected {
|
||||
err := call.matches(args)
|
||||
if err != nil {
|
||||
fmt.Fprintf(&callsErrors, "\n%v", err)
|
||||
} else {
|
||||
return call, nil
|
||||
}
|
||||
}
|
||||
|
||||
// If we haven't found a match then search through the exhausted calls so we
|
||||
// get useful error messages.
|
||||
exhausted := cs.exhausted[key]
|
||||
for _, call := range exhausted {
|
||||
if err := call.matches(args); err != nil {
|
||||
fmt.Fprintf(&callsErrors, "\n%v", err)
|
||||
}
|
||||
}
|
||||
|
||||
if len(expected)+len(exhausted) == 0 {
|
||||
fmt.Fprintf(&callsErrors, "there are no expected calls of the method %q for that receiver", method)
|
||||
}
|
||||
|
||||
return nil, fmt.Errorf(callsErrors.String())
|
||||
}
|
||||
|
||||
// Failures returns the calls that are not satisfied.
|
||||
func (cs callSet) Failures() []*Call {
|
||||
failures := make([]*Call, 0, len(cs.expected))
|
||||
for _, calls := range cs.expected {
|
||||
for _, call := range calls {
|
||||
if !call.satisfied() {
|
||||
failures = append(failures, call)
|
||||
}
|
||||
}
|
||||
}
|
||||
return failures
|
||||
}
|
76
vendor/github.com/golang/mock/gomock/callset_test.go
generated
vendored
Normal file
76
vendor/github.com/golang/mock/gomock/callset_test.go
generated
vendored
Normal file
@@ -0,0 +1,76 @@
|
||||
// Copyright 2011 Google Inc.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
package gomock
|
||||
|
||||
import (
|
||||
"reflect"
|
||||
"testing"
|
||||
)
|
||||
|
||||
type receiverType struct{}
|
||||
|
||||
func (receiverType) Func() {}
|
||||
|
||||
func TestCallSetAdd(t *testing.T) {
|
||||
method := "TestMethod"
|
||||
var receiver interface{} = "TestReceiver"
|
||||
cs := newCallSet()
|
||||
|
||||
numCalls := 10
|
||||
for i := 0; i < numCalls; i++ {
|
||||
cs.Add(newCall(t, receiver, method, reflect.TypeOf(receiverType{}.Func)))
|
||||
}
|
||||
|
||||
call, err := cs.FindMatch(receiver, method, []interface{}{})
|
||||
if err != nil {
|
||||
t.Fatalf("FindMatch: %v", err)
|
||||
}
|
||||
if call == nil {
|
||||
t.Fatalf("FindMatch: Got nil, want non-nil *Call")
|
||||
}
|
||||
}
|
||||
|
||||
func TestCallSetRemove(t *testing.T) {
|
||||
method := "TestMethod"
|
||||
var receiver interface{} = "TestReceiver"
|
||||
|
||||
cs := newCallSet()
|
||||
ourCalls := []*Call{}
|
||||
|
||||
numCalls := 10
|
||||
for i := 0; i < numCalls; i++ {
|
||||
// NOTE: abuse the `numCalls` value to convey initial ordering of mocked calls
|
||||
generatedCall := &Call{receiver: receiver, method: method, numCalls: i}
|
||||
cs.Add(generatedCall)
|
||||
ourCalls = append(ourCalls, generatedCall)
|
||||
}
|
||||
|
||||
// validateOrder validates that the calls in the array are ordered as they were added
|
||||
validateOrder := func(calls []*Call) {
|
||||
// lastNum tracks the last `numCalls` (call order) value seen
|
||||
lastNum := -1
|
||||
for _, c := range calls {
|
||||
if lastNum >= c.numCalls {
|
||||
t.Errorf("found call %d after call %d", c.numCalls, lastNum)
|
||||
}
|
||||
lastNum = c.numCalls
|
||||
}
|
||||
}
|
||||
|
||||
for _, c := range ourCalls {
|
||||
validateOrder(cs.expected[callSetKey{receiver, method}])
|
||||
cs.Remove(c)
|
||||
}
|
||||
}
|
217
vendor/github.com/golang/mock/gomock/controller.go
generated
vendored
Normal file
217
vendor/github.com/golang/mock/gomock/controller.go
generated
vendored
Normal file
@@ -0,0 +1,217 @@
|
||||
// Copyright 2010 Google Inc.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
// GoMock - a mock framework for Go.
|
||||
//
|
||||
// Standard usage:
|
||||
// (1) Define an interface that you wish to mock.
|
||||
// type MyInterface interface {
|
||||
// SomeMethod(x int64, y string)
|
||||
// }
|
||||
// (2) Use mockgen to generate a mock from the interface.
|
||||
// (3) Use the mock in a test:
|
||||
// func TestMyThing(t *testing.T) {
|
||||
// mockCtrl := gomock.NewController(t)
|
||||
// defer mockCtrl.Finish()
|
||||
//
|
||||
// mockObj := something.NewMockMyInterface(mockCtrl)
|
||||
// mockObj.EXPECT().SomeMethod(4, "blah")
|
||||
// // pass mockObj to a real object and play with it.
|
||||
// }
|
||||
//
|
||||
// By default, expected calls are not enforced to run in any particular order.
|
||||
// Call order dependency can be enforced by use of InOrder and/or Call.After.
|
||||
// Call.After can create more varied call order dependencies, but InOrder is
|
||||
// often more convenient.
|
||||
//
|
||||
// The following examples create equivalent call order dependencies.
|
||||
//
|
||||
// Example of using Call.After to chain expected call order:
|
||||
//
|
||||
// firstCall := mockObj.EXPECT().SomeMethod(1, "first")
|
||||
// secondCall := mockObj.EXPECT().SomeMethod(2, "second").After(firstCall)
|
||||
// mockObj.EXPECT().SomeMethod(3, "third").After(secondCall)
|
||||
//
|
||||
// Example of using InOrder to declare expected call order:
|
||||
//
|
||||
// gomock.InOrder(
|
||||
// mockObj.EXPECT().SomeMethod(1, "first"),
|
||||
// mockObj.EXPECT().SomeMethod(2, "second"),
|
||||
// mockObj.EXPECT().SomeMethod(3, "third"),
|
||||
// )
|
||||
//
|
||||
// TODO:
|
||||
// - Handle different argument/return types (e.g. ..., chan, map, interface).
|
||||
package gomock
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"golang.org/x/net/context"
|
||||
"reflect"
|
||||
"runtime"
|
||||
"sync"
|
||||
)
|
||||
|
||||
// A TestReporter is something that can be used to report test failures.
|
||||
// It is satisfied by the standard library's *testing.T.
|
||||
type TestReporter interface {
|
||||
Errorf(format string, args ...interface{})
|
||||
Fatalf(format string, args ...interface{})
|
||||
}
|
||||
|
||||
// A Controller represents the top-level control of a mock ecosystem.
|
||||
// It defines the scope and lifetime of mock objects, as well as their expectations.
|
||||
// It is safe to call Controller's methods from multiple goroutines.
|
||||
type Controller struct {
|
||||
mu sync.Mutex
|
||||
t TestReporter
|
||||
expectedCalls *callSet
|
||||
finished bool
|
||||
}
|
||||
|
||||
func NewController(t TestReporter) *Controller {
|
||||
return &Controller{
|
||||
t: t,
|
||||
expectedCalls: newCallSet(),
|
||||
}
|
||||
}
|
||||
|
||||
type cancelReporter struct {
|
||||
t TestReporter
|
||||
cancel func()
|
||||
}
|
||||
|
||||
func (r *cancelReporter) Errorf(format string, args ...interface{}) { r.t.Errorf(format, args...) }
|
||||
func (r *cancelReporter) Fatalf(format string, args ...interface{}) {
|
||||
defer r.cancel()
|
||||
r.t.Fatalf(format, args...)
|
||||
}
|
||||
|
||||
// WithContext returns a new Controller and a Context, which is cancelled on any
|
||||
// fatal failure.
|
||||
func WithContext(ctx context.Context, t TestReporter) (*Controller, context.Context) {
|
||||
ctx, cancel := context.WithCancel(ctx)
|
||||
return NewController(&cancelReporter{t, cancel}), ctx
|
||||
}
|
||||
|
||||
func (ctrl *Controller) RecordCall(receiver interface{}, method string, args ...interface{}) *Call {
|
||||
if h, ok := ctrl.t.(testHelper); ok {
|
||||
h.Helper()
|
||||
}
|
||||
|
||||
recv := reflect.ValueOf(receiver)
|
||||
for i := 0; i < recv.Type().NumMethod(); i++ {
|
||||
if recv.Type().Method(i).Name == method {
|
||||
return ctrl.RecordCallWithMethodType(receiver, method, recv.Method(i).Type(), args...)
|
||||
}
|
||||
}
|
||||
ctrl.t.Fatalf("gomock: failed finding method %s on %T", method, receiver)
|
||||
panic("unreachable")
|
||||
}
|
||||
|
||||
func (ctrl *Controller) RecordCallWithMethodType(receiver interface{}, method string, methodType reflect.Type, args ...interface{}) *Call {
|
||||
if h, ok := ctrl.t.(testHelper); ok {
|
||||
h.Helper()
|
||||
}
|
||||
|
||||
call := newCall(ctrl.t, receiver, method, methodType, args...)
|
||||
|
||||
ctrl.mu.Lock()
|
||||
defer ctrl.mu.Unlock()
|
||||
ctrl.expectedCalls.Add(call)
|
||||
|
||||
return call
|
||||
}
|
||||
|
||||
func (ctrl *Controller) Call(receiver interface{}, method string, args ...interface{}) []interface{} {
|
||||
if h, ok := ctrl.t.(testHelper); ok {
|
||||
h.Helper()
|
||||
}
|
||||
|
||||
// Nest this code so we can use defer to make sure the lock is released.
|
||||
actions := func() []func([]interface{}) []interface{} {
|
||||
ctrl.mu.Lock()
|
||||
defer ctrl.mu.Unlock()
|
||||
|
||||
expected, err := ctrl.expectedCalls.FindMatch(receiver, method, args)
|
||||
if err != nil {
|
||||
origin := callerInfo(2)
|
||||
ctrl.t.Fatalf("Unexpected call to %T.%v(%v) at %s because: %s", receiver, method, args, origin, err)
|
||||
}
|
||||
|
||||
// Two things happen here:
|
||||
// * the matching call no longer needs to check prerequite calls,
|
||||
// * and the prerequite calls are no longer expected, so remove them.
|
||||
preReqCalls := expected.dropPrereqs()
|
||||
for _, preReqCall := range preReqCalls {
|
||||
ctrl.expectedCalls.Remove(preReqCall)
|
||||
}
|
||||
|
||||
actions := expected.call(args)
|
||||
if expected.exhausted() {
|
||||
ctrl.expectedCalls.Remove(expected)
|
||||
}
|
||||
return actions
|
||||
}()
|
||||
|
||||
var rets []interface{}
|
||||
for _, action := range actions {
|
||||
if r := action(args); r != nil {
|
||||
rets = r
|
||||
}
|
||||
}
|
||||
|
||||
return rets
|
||||
}
|
||||
|
||||
func (ctrl *Controller) Finish() {
|
||||
if h, ok := ctrl.t.(testHelper); ok {
|
||||
h.Helper()
|
||||
}
|
||||
|
||||
ctrl.mu.Lock()
|
||||
defer ctrl.mu.Unlock()
|
||||
|
||||
if ctrl.finished {
|
||||
ctrl.t.Fatalf("Controller.Finish was called more than once. It has to be called exactly once.")
|
||||
}
|
||||
ctrl.finished = true
|
||||
|
||||
// If we're currently panicking, probably because this is a deferred call,
|
||||
// pass through the panic.
|
||||
if err := recover(); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
// Check that all remaining expected calls are satisfied.
|
||||
failures := ctrl.expectedCalls.Failures()
|
||||
for _, call := range failures {
|
||||
ctrl.t.Errorf("missing call(s) to %v", call)
|
||||
}
|
||||
if len(failures) != 0 {
|
||||
ctrl.t.Fatalf("aborting test due to missing call(s)")
|
||||
}
|
||||
}
|
||||
|
||||
func callerInfo(skip int) string {
|
||||
if _, file, line, ok := runtime.Caller(skip + 1); ok {
|
||||
return fmt.Sprintf("%s:%d", file, line)
|
||||
}
|
||||
return "unknown file"
|
||||
}
|
||||
|
||||
type testHelper interface {
|
||||
TestReporter
|
||||
Helper()
|
||||
}
|
712
vendor/github.com/golang/mock/gomock/controller_test.go
generated
vendored
Normal file
712
vendor/github.com/golang/mock/gomock/controller_test.go
generated
vendored
Normal file
@@ -0,0 +1,712 @@
|
||||
// Copyright 2011 Google Inc.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
package gomock_test
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"reflect"
|
||||
"testing"
|
||||
|
||||
"strings"
|
||||
|
||||
"github.com/golang/mock/gomock"
|
||||
)
|
||||
|
||||
type ErrorReporter struct {
|
||||
t *testing.T
|
||||
log []string
|
||||
failed bool
|
||||
fatalToken struct{}
|
||||
}
|
||||
|
||||
func NewErrorReporter(t *testing.T) *ErrorReporter {
|
||||
return &ErrorReporter{t: t}
|
||||
}
|
||||
|
||||
func (e *ErrorReporter) reportLog() {
|
||||
for _, entry := range e.log {
|
||||
e.t.Log(entry)
|
||||
}
|
||||
}
|
||||
|
||||
func (e *ErrorReporter) assertPass(msg string) {
|
||||
if e.failed {
|
||||
e.t.Errorf("Expected pass, but got failure(s): %s", msg)
|
||||
e.reportLog()
|
||||
}
|
||||
}
|
||||
|
||||
func (e *ErrorReporter) assertFail(msg string) {
|
||||
if !e.failed {
|
||||
e.t.Errorf("Expected failure, but got pass: %s", msg)
|
||||
}
|
||||
}
|
||||
|
||||
// Use to check that code triggers a fatal test failure.
|
||||
func (e *ErrorReporter) assertFatal(fn func(), expectedErrMsgs ...string) {
|
||||
defer func() {
|
||||
err := recover()
|
||||
if err == nil {
|
||||
var actual string
|
||||
if e.failed {
|
||||
actual = "non-fatal failure"
|
||||
} else {
|
||||
actual = "pass"
|
||||
}
|
||||
e.t.Error("Expected fatal failure, but got a", actual)
|
||||
} else if token, ok := err.(*struct{}); ok && token == &e.fatalToken {
|
||||
// This is okay - the panic is from Fatalf().
|
||||
if expectedErrMsgs != nil {
|
||||
// assert that the actual error message
|
||||
// contains expectedErrMsgs
|
||||
|
||||
// check the last actualErrMsg, because the previous messages come from previous errors
|
||||
actualErrMsg := e.log[len(e.log)-1]
|
||||
for _, expectedErrMsg := range expectedErrMsgs {
|
||||
if !strings.Contains(actualErrMsg, expectedErrMsg) {
|
||||
e.t.Errorf("Error message:\ngot: %q\nwant to contain: %q\n", actualErrMsg, expectedErrMsg)
|
||||
}
|
||||
}
|
||||
}
|
||||
return
|
||||
} else {
|
||||
// Some other panic.
|
||||
panic(err)
|
||||
}
|
||||
}()
|
||||
|
||||
fn()
|
||||
}
|
||||
|
||||
// recoverUnexpectedFatal can be used as a deferred call in test cases to
|
||||
// recover from and display a call to ErrorReporter.Fatalf().
|
||||
func (e *ErrorReporter) recoverUnexpectedFatal() {
|
||||
err := recover()
|
||||
if err == nil {
|
||||
// No panic.
|
||||
} else if token, ok := err.(*struct{}); ok && token == &e.fatalToken {
|
||||
// Unexpected fatal error happened.
|
||||
e.t.Error("Got unexpected fatal error(s). All errors up to this point:")
|
||||
e.reportLog()
|
||||
return
|
||||
} else {
|
||||
// Some other panic.
|
||||
panic(err)
|
||||
}
|
||||
}
|
||||
|
||||
func (e *ErrorReporter) Logf(format string, args ...interface{}) {
|
||||
e.log = append(e.log, fmt.Sprintf(format, args...))
|
||||
}
|
||||
|
||||
func (e *ErrorReporter) Errorf(format string, args ...interface{}) {
|
||||
e.Logf(format, args...)
|
||||
e.failed = true
|
||||
}
|
||||
|
||||
func (e *ErrorReporter) Fatalf(format string, args ...interface{}) {
|
||||
e.Logf(format, args...)
|
||||
e.failed = true
|
||||
panic(&e.fatalToken)
|
||||
}
|
||||
|
||||
// A type purely for use as a receiver in testing the Controller.
|
||||
type Subject struct{}
|
||||
|
||||
func (s *Subject) FooMethod(arg string) int {
|
||||
return 0
|
||||
}
|
||||
|
||||
func (s *Subject) BarMethod(arg string) int {
|
||||
return 0
|
||||
}
|
||||
|
||||
func (s *Subject) VariadicMethod(arg int, vararg ...string) {}
|
||||
|
||||
// A type purely for ActOnTestStructMethod
|
||||
type TestStruct struct {
|
||||
Number int
|
||||
Message string
|
||||
}
|
||||
|
||||
func (s *Subject) ActOnTestStructMethod(arg TestStruct, arg1 int) int {
|
||||
return 0
|
||||
}
|
||||
|
||||
func (s *Subject) SetArgMethod(sliceArg []byte, ptrArg *int) {}
|
||||
|
||||
func assertEqual(t *testing.T, expected interface{}, actual interface{}) {
|
||||
if !reflect.DeepEqual(expected, actual) {
|
||||
t.Errorf("Expected %+v, but got %+v", expected, actual)
|
||||
}
|
||||
}
|
||||
|
||||
func createFixtures(t *testing.T) (reporter *ErrorReporter, ctrl *gomock.Controller) {
|
||||
// reporter acts as a testing.T-like object that we pass to the
|
||||
// Controller. We use it to test that the mock considered tests
|
||||
// successful or failed.
|
||||
reporter = NewErrorReporter(t)
|
||||
ctrl = gomock.NewController(reporter)
|
||||
return
|
||||
}
|
||||
|
||||
func TestNoCalls(t *testing.T) {
|
||||
reporter, ctrl := createFixtures(t)
|
||||
ctrl.Finish()
|
||||
reporter.assertPass("No calls expected or made.")
|
||||
}
|
||||
|
||||
func TestNoRecordedCallsForAReceiver(t *testing.T) {
|
||||
reporter, ctrl := createFixtures(t)
|
||||
subject := new(Subject)
|
||||
|
||||
reporter.assertFatal(func() {
|
||||
ctrl.Call(subject, "NotRecordedMethod", "argument")
|
||||
}, "Unexpected call to", "there are no expected calls of the method \"NotRecordedMethod\" for that receiver")
|
||||
ctrl.Finish()
|
||||
}
|
||||
|
||||
func TestNoRecordedMatchingMethodNameForAReceiver(t *testing.T) {
|
||||
reporter, ctrl := createFixtures(t)
|
||||
subject := new(Subject)
|
||||
|
||||
ctrl.RecordCall(subject, "FooMethod", "argument")
|
||||
reporter.assertFatal(func() {
|
||||
ctrl.Call(subject, "NotRecordedMethod", "argument")
|
||||
}, "Unexpected call to", "there are no expected calls of the method \"NotRecordedMethod\" for that receiver")
|
||||
reporter.assertFatal(func() {
|
||||
// The expected call wasn't made.
|
||||
ctrl.Finish()
|
||||
})
|
||||
}
|
||||
|
||||
// This tests that a call with an arguments of some primitive type matches a recorded call.
|
||||
func TestExpectedMethodCall(t *testing.T) {
|
||||
reporter, ctrl := createFixtures(t)
|
||||
subject := new(Subject)
|
||||
|
||||
ctrl.RecordCall(subject, "FooMethod", "argument")
|
||||
ctrl.Call(subject, "FooMethod", "argument")
|
||||
ctrl.Finish()
|
||||
|
||||
reporter.assertPass("Expected method call made.")
|
||||
}
|
||||
|
||||
func TestUnexpectedMethodCall(t *testing.T) {
|
||||
reporter, ctrl := createFixtures(t)
|
||||
subject := new(Subject)
|
||||
|
||||
reporter.assertFatal(func() {
|
||||
ctrl.Call(subject, "FooMethod", "argument")
|
||||
})
|
||||
|
||||
ctrl.Finish()
|
||||
}
|
||||
|
||||
func TestRepeatedCall(t *testing.T) {
|
||||
reporter, ctrl := createFixtures(t)
|
||||
subject := new(Subject)
|
||||
|
||||
ctrl.RecordCall(subject, "FooMethod", "argument").Times(3)
|
||||
ctrl.Call(subject, "FooMethod", "argument")
|
||||
ctrl.Call(subject, "FooMethod", "argument")
|
||||
ctrl.Call(subject, "FooMethod", "argument")
|
||||
reporter.assertPass("After expected repeated method calls.")
|
||||
reporter.assertFatal(func() {
|
||||
ctrl.Call(subject, "FooMethod", "argument")
|
||||
})
|
||||
ctrl.Finish()
|
||||
reporter.assertFail("After calling one too many times.")
|
||||
}
|
||||
|
||||
func TestUnexpectedArgCount(t *testing.T) {
|
||||
reporter, ctrl := createFixtures(t)
|
||||
defer reporter.recoverUnexpectedFatal()
|
||||
subject := new(Subject)
|
||||
|
||||
ctrl.RecordCall(subject, "FooMethod", "argument")
|
||||
reporter.assertFatal(func() {
|
||||
// This call is made with the wrong number of arguments...
|
||||
ctrl.Call(subject, "FooMethod", "argument", "extra_argument")
|
||||
}, "Unexpected call to", "wrong number of arguments", "Got: 2, want: 1")
|
||||
reporter.assertFatal(func() {
|
||||
// ... so is this.
|
||||
ctrl.Call(subject, "FooMethod")
|
||||
}, "Unexpected call to", "wrong number of arguments", "Got: 0, want: 1")
|
||||
reporter.assertFatal(func() {
|
||||
// The expected call wasn't made.
|
||||
ctrl.Finish()
|
||||
})
|
||||
}
|
||||
|
||||
// This tests that a call with complex arguments (a struct and some primitive type) matches a recorded call.
|
||||
func TestExpectedMethodCall_CustomStruct(t *testing.T) {
|
||||
reporter, ctrl := createFixtures(t)
|
||||
subject := new(Subject)
|
||||
|
||||
expectedArg0 := TestStruct{Number: 123, Message: "hello"}
|
||||
ctrl.RecordCall(subject, "ActOnTestStructMethod", expectedArg0, 15)
|
||||
ctrl.Call(subject, "ActOnTestStructMethod", expectedArg0, 15)
|
||||
|
||||
reporter.assertPass("Expected method call made.")
|
||||
}
|
||||
|
||||
func TestUnexpectedArgValue_FirstArg(t *testing.T) {
|
||||
reporter, ctrl := createFixtures(t)
|
||||
defer reporter.recoverUnexpectedFatal()
|
||||
subject := new(Subject)
|
||||
|
||||
expectedArg0 := TestStruct{Number: 123, Message: "hello"}
|
||||
ctrl.RecordCall(subject, "ActOnTestStructMethod", expectedArg0, 15)
|
||||
|
||||
reporter.assertFatal(func() {
|
||||
// the method argument (of TestStruct type) has 1 unexpected value (for the Message field)
|
||||
ctrl.Call(subject, "ActOnTestStructMethod", TestStruct{Number: 123, Message: "no message"}, 15)
|
||||
}, "Unexpected call to", "doesn't match the argument at index 0",
|
||||
"Got: {123 no message}\nWant: is equal to {123 hello}")
|
||||
|
||||
reporter.assertFatal(func() {
|
||||
// the method argument (of TestStruct type) has 2 unexpected values (for both fields)
|
||||
ctrl.Call(subject, "ActOnTestStructMethod", TestStruct{Number: 11, Message: "no message"}, 15)
|
||||
}, "Unexpected call to", "doesn't match the argument at index 0",
|
||||
"Got: {11 no message}\nWant: is equal to {123 hello}")
|
||||
|
||||
reporter.assertFatal(func() {
|
||||
// The expected call wasn't made.
|
||||
ctrl.Finish()
|
||||
})
|
||||
}
|
||||
|
||||
func TestUnexpectedArgValue_SecondtArg(t *testing.T) {
|
||||
reporter, ctrl := createFixtures(t)
|
||||
defer reporter.recoverUnexpectedFatal()
|
||||
subject := new(Subject)
|
||||
|
||||
expectedArg0 := TestStruct{Number: 123, Message: "hello"}
|
||||
ctrl.RecordCall(subject, "ActOnTestStructMethod", expectedArg0, 15)
|
||||
|
||||
reporter.assertFatal(func() {
|
||||
ctrl.Call(subject, "ActOnTestStructMethod", TestStruct{Number: 123, Message: "hello"}, 3)
|
||||
}, "Unexpected call to", "doesn't match the argument at index 1",
|
||||
"Got: 3\nWant: is equal to 15")
|
||||
|
||||
reporter.assertFatal(func() {
|
||||
// The expected call wasn't made.
|
||||
ctrl.Finish()
|
||||
})
|
||||
}
|
||||
|
||||
func TestAnyTimes(t *testing.T) {
|
||||
reporter, ctrl := createFixtures(t)
|
||||
subject := new(Subject)
|
||||
|
||||
ctrl.RecordCall(subject, "FooMethod", "argument").AnyTimes()
|
||||
for i := 0; i < 100; i++ {
|
||||
ctrl.Call(subject, "FooMethod", "argument")
|
||||
}
|
||||
reporter.assertPass("After 100 method calls.")
|
||||
ctrl.Finish()
|
||||
}
|
||||
|
||||
func TestMinTimes1(t *testing.T) {
|
||||
// It fails if there are no calls
|
||||
reporter, ctrl := createFixtures(t)
|
||||
subject := new(Subject)
|
||||
ctrl.RecordCall(subject, "FooMethod", "argument").MinTimes(1)
|
||||
reporter.assertFatal(func() {
|
||||
ctrl.Finish()
|
||||
})
|
||||
|
||||
// It succeeds if there is one call
|
||||
reporter, ctrl = createFixtures(t)
|
||||
subject = new(Subject)
|
||||
ctrl.RecordCall(subject, "FooMethod", "argument").MinTimes(1)
|
||||
ctrl.Call(subject, "FooMethod", "argument")
|
||||
ctrl.Finish()
|
||||
|
||||
// It succeeds if there are many calls
|
||||
reporter, ctrl = createFixtures(t)
|
||||
subject = new(Subject)
|
||||
ctrl.RecordCall(subject, "FooMethod", "argument").MinTimes(1)
|
||||
for i := 0; i < 100; i++ {
|
||||
ctrl.Call(subject, "FooMethod", "argument")
|
||||
}
|
||||
ctrl.Finish()
|
||||
}
|
||||
|
||||
func TestMaxTimes1(t *testing.T) {
|
||||
// It succeeds if there are no calls
|
||||
_, ctrl := createFixtures(t)
|
||||
subject := new(Subject)
|
||||
ctrl.RecordCall(subject, "FooMethod", "argument").MaxTimes(1)
|
||||
ctrl.Finish()
|
||||
|
||||
// It succeeds if there is one call
|
||||
_, ctrl = createFixtures(t)
|
||||
subject = new(Subject)
|
||||
ctrl.RecordCall(subject, "FooMethod", "argument").MaxTimes(1)
|
||||
ctrl.Call(subject, "FooMethod", "argument")
|
||||
ctrl.Finish()
|
||||
|
||||
//It fails if there are more
|
||||
reporter, ctrl := createFixtures(t)
|
||||
subject = new(Subject)
|
||||
ctrl.RecordCall(subject, "FooMethod", "argument").MaxTimes(1)
|
||||
ctrl.Call(subject, "FooMethod", "argument")
|
||||
reporter.assertFatal(func() {
|
||||
ctrl.Call(subject, "FooMethod", "argument")
|
||||
})
|
||||
ctrl.Finish()
|
||||
}
|
||||
|
||||
func TestMinMaxTimes(t *testing.T) {
|
||||
// It fails if there are less calls than specified
|
||||
reporter, ctrl := createFixtures(t)
|
||||
subject := new(Subject)
|
||||
ctrl.RecordCall(subject, "FooMethod", "argument").MinTimes(2).MaxTimes(2)
|
||||
ctrl.Call(subject, "FooMethod", "argument")
|
||||
reporter.assertFatal(func() {
|
||||
ctrl.Finish()
|
||||
})
|
||||
|
||||
// It fails if there are more calls than specified
|
||||
reporter, ctrl = createFixtures(t)
|
||||
subject = new(Subject)
|
||||
ctrl.RecordCall(subject, "FooMethod", "argument").MinTimes(2).MaxTimes(2)
|
||||
ctrl.Call(subject, "FooMethod", "argument")
|
||||
ctrl.Call(subject, "FooMethod", "argument")
|
||||
reporter.assertFatal(func() {
|
||||
ctrl.Call(subject, "FooMethod", "argument")
|
||||
})
|
||||
|
||||
// It succeeds if there is just the right number of calls
|
||||
reporter, ctrl = createFixtures(t)
|
||||
subject = new(Subject)
|
||||
ctrl.RecordCall(subject, "FooMethod", "argument").MaxTimes(2).MinTimes(2)
|
||||
ctrl.Call(subject, "FooMethod", "argument")
|
||||
ctrl.Call(subject, "FooMethod", "argument")
|
||||
ctrl.Finish()
|
||||
}
|
||||
|
||||
func TestDo(t *testing.T) {
|
||||
_, ctrl := createFixtures(t)
|
||||
subject := new(Subject)
|
||||
|
||||
doCalled := false
|
||||
var argument string
|
||||
ctrl.RecordCall(subject, "FooMethod", "argument").Do(
|
||||
func(arg string) {
|
||||
doCalled = true
|
||||
argument = arg
|
||||
})
|
||||
if doCalled {
|
||||
t.Error("Do() callback called too early.")
|
||||
}
|
||||
|
||||
ctrl.Call(subject, "FooMethod", "argument")
|
||||
|
||||
if !doCalled {
|
||||
t.Error("Do() callback not called.")
|
||||
}
|
||||
if "argument" != argument {
|
||||
t.Error("Do callback received wrong argument.")
|
||||
}
|
||||
|
||||
ctrl.Finish()
|
||||
}
|
||||
|
||||
func TestDoAndReturn(t *testing.T) {
|
||||
_, ctrl := createFixtures(t)
|
||||
subject := new(Subject)
|
||||
|
||||
doCalled := false
|
||||
var argument string
|
||||
ctrl.RecordCall(subject, "FooMethod", "argument").DoAndReturn(
|
||||
func(arg string) int {
|
||||
doCalled = true
|
||||
argument = arg
|
||||
return 5
|
||||
})
|
||||
if doCalled {
|
||||
t.Error("Do() callback called too early.")
|
||||
}
|
||||
|
||||
rets := ctrl.Call(subject, "FooMethod", "argument")
|
||||
|
||||
if !doCalled {
|
||||
t.Error("Do() callback not called.")
|
||||
}
|
||||
if "argument" != argument {
|
||||
t.Error("Do callback received wrong argument.")
|
||||
}
|
||||
if len(rets) != 1 {
|
||||
t.Fatalf("Return values from Call: got %d, want 1", len(rets))
|
||||
}
|
||||
if ret, ok := rets[0].(int); !ok {
|
||||
t.Fatalf("Return value is not an int")
|
||||
} else if ret != 5 {
|
||||
t.Errorf("DoAndReturn return value: got %d, want 5", ret)
|
||||
}
|
||||
|
||||
ctrl.Finish()
|
||||
}
|
||||
|
||||
func TestSetArgSlice(t *testing.T) {
|
||||
_, ctrl := createFixtures(t)
|
||||
subject := new(Subject)
|
||||
|
||||
var in = []byte{4, 5, 6}
|
||||
var set = []byte{1, 2, 3}
|
||||
ctrl.RecordCall(subject, "SetArgMethod", in, nil).SetArg(0, set)
|
||||
ctrl.Call(subject, "SetArgMethod", in, nil)
|
||||
|
||||
if !reflect.DeepEqual(in, set) {
|
||||
t.Error("Expected SetArg() to modify input slice argument")
|
||||
}
|
||||
|
||||
ctrl.Finish()
|
||||
}
|
||||
|
||||
func TestSetArgPtr(t *testing.T) {
|
||||
_, ctrl := createFixtures(t)
|
||||
subject := new(Subject)
|
||||
|
||||
var in int = 43
|
||||
const set = 42
|
||||
ctrl.RecordCall(subject, "SetArgMethod", nil, &in).SetArg(1, set)
|
||||
ctrl.Call(subject, "SetArgMethod", nil, &in)
|
||||
|
||||
if in != set {
|
||||
t.Error("Expected SetArg() to modify value pointed to by argument")
|
||||
}
|
||||
|
||||
ctrl.Finish()
|
||||
}
|
||||
|
||||
func TestReturn(t *testing.T) {
|
||||
_, ctrl := createFixtures(t)
|
||||
subject := new(Subject)
|
||||
|
||||
// Unspecified return should produce "zero" result.
|
||||
ctrl.RecordCall(subject, "FooMethod", "zero")
|
||||
ctrl.RecordCall(subject, "FooMethod", "five").Return(5)
|
||||
|
||||
assertEqual(
|
||||
t,
|
||||
[]interface{}{0},
|
||||
ctrl.Call(subject, "FooMethod", "zero"))
|
||||
|
||||
assertEqual(
|
||||
t,
|
||||
[]interface{}{5},
|
||||
ctrl.Call(subject, "FooMethod", "five"))
|
||||
ctrl.Finish()
|
||||
}
|
||||
|
||||
func TestUnorderedCalls(t *testing.T) {
|
||||
reporter, ctrl := createFixtures(t)
|
||||
defer reporter.recoverUnexpectedFatal()
|
||||
subjectTwo := new(Subject)
|
||||
subjectOne := new(Subject)
|
||||
|
||||
ctrl.RecordCall(subjectOne, "FooMethod", "1")
|
||||
ctrl.RecordCall(subjectOne, "BarMethod", "2")
|
||||
ctrl.RecordCall(subjectTwo, "FooMethod", "3")
|
||||
ctrl.RecordCall(subjectTwo, "BarMethod", "4")
|
||||
|
||||
// Make the calls in a different order, which should be fine.
|
||||
ctrl.Call(subjectOne, "BarMethod", "2")
|
||||
ctrl.Call(subjectTwo, "FooMethod", "3")
|
||||
ctrl.Call(subjectTwo, "BarMethod", "4")
|
||||
ctrl.Call(subjectOne, "FooMethod", "1")
|
||||
|
||||
reporter.assertPass("After making all calls in different order")
|
||||
|
||||
ctrl.Finish()
|
||||
|
||||
reporter.assertPass("After finish")
|
||||
}
|
||||
|
||||
func commonTestOrderedCalls(t *testing.T) (reporter *ErrorReporter, ctrl *gomock.Controller, subjectOne, subjectTwo *Subject) {
|
||||
reporter, ctrl = createFixtures(t)
|
||||
|
||||
subjectOne = new(Subject)
|
||||
subjectTwo = new(Subject)
|
||||
|
||||
gomock.InOrder(
|
||||
ctrl.RecordCall(subjectOne, "FooMethod", "1").AnyTimes(),
|
||||
ctrl.RecordCall(subjectTwo, "FooMethod", "2"),
|
||||
ctrl.RecordCall(subjectTwo, "BarMethod", "3"),
|
||||
)
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func TestOrderedCallsCorrect(t *testing.T) {
|
||||
reporter, ctrl, subjectOne, subjectTwo := commonTestOrderedCalls(t)
|
||||
|
||||
ctrl.Call(subjectOne, "FooMethod", "1")
|
||||
ctrl.Call(subjectTwo, "FooMethod", "2")
|
||||
ctrl.Call(subjectTwo, "BarMethod", "3")
|
||||
|
||||
ctrl.Finish()
|
||||
|
||||
reporter.assertPass("After finish")
|
||||
}
|
||||
|
||||
func TestOrderedCallsInCorrect(t *testing.T) {
|
||||
reporter, ctrl, subjectOne, subjectTwo := commonTestOrderedCalls(t)
|
||||
|
||||
ctrl.Call(subjectOne, "FooMethod", "1")
|
||||
reporter.assertFatal(func() {
|
||||
// FooMethod(2) should be called before BarMethod(3)
|
||||
ctrl.Call(subjectTwo, "BarMethod", "3")
|
||||
}, "Unexpected call to", "Subject.BarMethod([3])", "doesn't have a prerequisite call satisfied")
|
||||
}
|
||||
|
||||
// Test that calls that are prerequisites to other calls but have maxCalls >
|
||||
// minCalls are removed from the expected call set.
|
||||
func TestOrderedCallsWithPreReqMaxUnbounded(t *testing.T) {
|
||||
reporter, ctrl, subjectOne, subjectTwo := commonTestOrderedCalls(t)
|
||||
|
||||
// Initially we should be able to call FooMethod("1") as many times as we
|
||||
// want.
|
||||
ctrl.Call(subjectOne, "FooMethod", "1")
|
||||
ctrl.Call(subjectOne, "FooMethod", "1")
|
||||
|
||||
// But calling something that has it as a prerequite should remove it from
|
||||
// the expected call set. This allows tests to ensure that FooMethod("1") is
|
||||
// *not* called after FooMethod("2").
|
||||
ctrl.Call(subjectTwo, "FooMethod", "2")
|
||||
|
||||
// Therefore this call should fail:
|
||||
reporter.assertFatal(func() {
|
||||
ctrl.Call(subjectOne, "FooMethod", "1")
|
||||
})
|
||||
}
|
||||
|
||||
func TestCallAfterLoopPanic(t *testing.T) {
|
||||
_, ctrl := createFixtures(t)
|
||||
|
||||
subject := new(Subject)
|
||||
|
||||
firstCall := ctrl.RecordCall(subject, "FooMethod", "1")
|
||||
secondCall := ctrl.RecordCall(subject, "FooMethod", "2")
|
||||
thirdCall := ctrl.RecordCall(subject, "FooMethod", "3")
|
||||
|
||||
gomock.InOrder(firstCall, secondCall, thirdCall)
|
||||
|
||||
defer func() {
|
||||
err := recover()
|
||||
if err == nil {
|
||||
t.Error("Call.After creation of dependency loop did not panic.")
|
||||
}
|
||||
}()
|
||||
|
||||
// This should panic due to dependency loop.
|
||||
firstCall.After(thirdCall)
|
||||
}
|
||||
|
||||
func TestPanicOverridesExpectationChecks(t *testing.T) {
|
||||
ctrl := gomock.NewController(t)
|
||||
reporter := NewErrorReporter(t)
|
||||
|
||||
reporter.assertFatal(func() {
|
||||
ctrl.RecordCall(new(Subject), "FooMethod", "1")
|
||||
defer ctrl.Finish()
|
||||
reporter.Fatalf("Intentional panic")
|
||||
})
|
||||
}
|
||||
|
||||
func TestSetArgWithBadType(t *testing.T) {
|
||||
rep, ctrl := createFixtures(t)
|
||||
defer ctrl.Finish()
|
||||
|
||||
s := new(Subject)
|
||||
// This should catch a type error:
|
||||
rep.assertFatal(func() {
|
||||
ctrl.RecordCall(s, "FooMethod", "1").SetArg(0, "blah")
|
||||
})
|
||||
ctrl.Call(s, "FooMethod", "1")
|
||||
}
|
||||
|
||||
func TestTimes0(t *testing.T) {
|
||||
rep, ctrl := createFixtures(t)
|
||||
defer ctrl.Finish()
|
||||
|
||||
s := new(Subject)
|
||||
ctrl.RecordCall(s, "FooMethod", "arg").Times(0)
|
||||
rep.assertFatal(func() {
|
||||
ctrl.Call(s, "FooMethod", "arg")
|
||||
})
|
||||
}
|
||||
|
||||
func TestVariadicMatching(t *testing.T) {
|
||||
rep, ctrl := createFixtures(t)
|
||||
defer rep.recoverUnexpectedFatal()
|
||||
|
||||
s := new(Subject)
|
||||
ctrl.RecordCall(s, "VariadicMethod", 0, "1", "2")
|
||||
ctrl.Call(s, "VariadicMethod", 0, "1", "2")
|
||||
ctrl.Finish()
|
||||
rep.assertPass("variadic matching works")
|
||||
}
|
||||
|
||||
func TestVariadicNoMatch(t *testing.T) {
|
||||
rep, ctrl := createFixtures(t)
|
||||
defer rep.recoverUnexpectedFatal()
|
||||
|
||||
s := new(Subject)
|
||||
ctrl.RecordCall(s, "VariadicMethod", 0)
|
||||
rep.assertFatal(func() {
|
||||
ctrl.Call(s, "VariadicMethod", 1)
|
||||
}, "Expected call at", "doesn't match the argument at index 0",
|
||||
"Got: 1\nWant: is equal to 0")
|
||||
ctrl.Call(s, "VariadicMethod", 0)
|
||||
ctrl.Finish()
|
||||
}
|
||||
|
||||
func TestVariadicMatchingWithSlice(t *testing.T) {
|
||||
testCases := [][]string{
|
||||
{"1"},
|
||||
{"1", "2"},
|
||||
}
|
||||
for _, tc := range testCases {
|
||||
t.Run(fmt.Sprintf("%d arguments", len(tc)), func(t *testing.T) {
|
||||
rep, ctrl := createFixtures(t)
|
||||
defer rep.recoverUnexpectedFatal()
|
||||
|
||||
s := new(Subject)
|
||||
ctrl.RecordCall(s, "VariadicMethod", 1, tc)
|
||||
args := make([]interface{}, len(tc)+1)
|
||||
args[0] = 1
|
||||
for i, arg := range tc {
|
||||
args[i+1] = arg
|
||||
}
|
||||
ctrl.Call(s, "VariadicMethod", args...)
|
||||
ctrl.Finish()
|
||||
rep.assertPass("slices can be used as matchers for variadic arguments")
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func TestDuplicateFinishCallFails(t *testing.T) {
|
||||
rep, ctrl := createFixtures(t)
|
||||
|
||||
ctrl.Finish()
|
||||
rep.assertPass("the first Finish call should succeed")
|
||||
|
||||
rep.assertFatal(ctrl.Finish, "Controller.Finish was called more than once. It has to be called exactly once.")
|
||||
}
|
99
vendor/github.com/golang/mock/gomock/matchers.go
generated
vendored
Normal file
99
vendor/github.com/golang/mock/gomock/matchers.go
generated
vendored
Normal file
@@ -0,0 +1,99 @@
|
||||
//go:generate mockgen -destination mock_matcher/mock_matcher.go github.com/golang/mock/gomock Matcher
|
||||
|
||||
// Copyright 2010 Google Inc.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
package gomock
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"reflect"
|
||||
)
|
||||
|
||||
// A Matcher is a representation of a class of values.
|
||||
// It is used to represent the valid or expected arguments to a mocked method.
|
||||
type Matcher interface {
|
||||
// Matches returns whether x is a match.
|
||||
Matches(x interface{}) bool
|
||||
|
||||
// String describes what the matcher matches.
|
||||
String() string
|
||||
}
|
||||
|
||||
type anyMatcher struct{}
|
||||
|
||||
func (anyMatcher) Matches(x interface{}) bool {
|
||||
return true
|
||||
}
|
||||
|
||||
func (anyMatcher) String() string {
|
||||
return "is anything"
|
||||
}
|
||||
|
||||
type eqMatcher struct {
|
||||
x interface{}
|
||||
}
|
||||
|
||||
func (e eqMatcher) Matches(x interface{}) bool {
|
||||
return reflect.DeepEqual(e.x, x)
|
||||
}
|
||||
|
||||
func (e eqMatcher) String() string {
|
||||
return fmt.Sprintf("is equal to %v", e.x)
|
||||
}
|
||||
|
||||
type nilMatcher struct{}
|
||||
|
||||
func (nilMatcher) Matches(x interface{}) bool {
|
||||
if x == nil {
|
||||
return true
|
||||
}
|
||||
|
||||
v := reflect.ValueOf(x)
|
||||
switch v.Kind() {
|
||||
case reflect.Chan, reflect.Func, reflect.Interface, reflect.Map,
|
||||
reflect.Ptr, reflect.Slice:
|
||||
return v.IsNil()
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
func (nilMatcher) String() string {
|
||||
return "is nil"
|
||||
}
|
||||
|
||||
type notMatcher struct {
|
||||
m Matcher
|
||||
}
|
||||
|
||||
func (n notMatcher) Matches(x interface{}) bool {
|
||||
return !n.m.Matches(x)
|
||||
}
|
||||
|
||||
func (n notMatcher) String() string {
|
||||
// TODO: Improve this if we add a NotString method to the Matcher interface.
|
||||
return "not(" + n.m.String() + ")"
|
||||
}
|
||||
|
||||
// Constructors
|
||||
func Any() Matcher { return anyMatcher{} }
|
||||
func Eq(x interface{}) Matcher { return eqMatcher{x} }
|
||||
func Nil() Matcher { return nilMatcher{} }
|
||||
func Not(x interface{}) Matcher {
|
||||
if m, ok := x.(Matcher); ok {
|
||||
return notMatcher{m}
|
||||
}
|
||||
return notMatcher{Eq(x)}
|
||||
}
|
70
vendor/github.com/golang/mock/gomock/matchers_test.go
generated
vendored
Normal file
70
vendor/github.com/golang/mock/gomock/matchers_test.go
generated
vendored
Normal file
@@ -0,0 +1,70 @@
|
||||
// Copyright 2010 Google Inc.
|
||||
//
|
||||
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||
// you may not use this file except in compliance with the License.
|
||||
// You may obtain a copy of the License at
|
||||
//
|
||||
// http://www.apache.org/licenses/LICENSE-2.0
|
||||
//
|
||||
// Unless required by applicable law or agreed to in writing, software
|
||||
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
|
||||
package gomock_test
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"testing"
|
||||
|
||||
"github.com/golang/mock/gomock"
|
||||
mock_matcher "github.com/golang/mock/gomock/mock_matcher"
|
||||
)
|
||||
|
||||
func TestMatchers(t *testing.T) {
|
||||
type e interface{}
|
||||
type testCase struct {
|
||||
matcher gomock.Matcher
|
||||
yes, no []e
|
||||
}
|
||||
tests := []testCase{
|
||||
testCase{gomock.Any(), []e{3, nil, "foo"}, nil},
|
||||
testCase{gomock.Eq(4), []e{4}, []e{3, "blah", nil, int64(4)}},
|
||||
testCase{gomock.Nil(),
|
||||
[]e{nil, (error)(nil), (chan bool)(nil), (*int)(nil)},
|
||||
[]e{"", 0, make(chan bool), errors.New("err"), new(int)}},
|
||||
testCase{gomock.Not(gomock.Eq(4)), []e{3, "blah", nil, int64(4)}, []e{4}},
|
||||
}
|
||||
for i, test := range tests {
|
||||
for _, x := range test.yes {
|
||||
if !test.matcher.Matches(x) {
|
||||
t.Errorf(`test %d: "%v %s" should be true.`, i, x, test.matcher)
|
||||
}
|
||||
}
|
||||
for _, x := range test.no {
|
||||
if test.matcher.Matches(x) {
|
||||
t.Errorf(`test %d: "%v %s" should be false.`, i, x, test.matcher)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// A more thorough test of notMatcher
|
||||
func TestNotMatcher(t *testing.T) {
|
||||
ctrl := gomock.NewController(t)
|
||||
defer ctrl.Finish()
|
||||
|
||||
mockMatcher := mock_matcher.NewMockMatcher(ctrl)
|
||||
notMatcher := gomock.Not(mockMatcher)
|
||||
|
||||
mockMatcher.EXPECT().Matches(4).Return(true)
|
||||
if match := notMatcher.Matches(4); match {
|
||||
t.Errorf("notMatcher should not match 4")
|
||||
}
|
||||
|
||||
mockMatcher.EXPECT().Matches(5).Return(false)
|
||||
if match := notMatcher.Matches(5); !match {
|
||||
t.Errorf("notMatcher should match 5")
|
||||
}
|
||||
}
|
57
vendor/github.com/golang/mock/gomock/mock_matcher/mock_matcher.go
generated
vendored
Normal file
57
vendor/github.com/golang/mock/gomock/mock_matcher/mock_matcher.go
generated
vendored
Normal file
@@ -0,0 +1,57 @@
|
||||
// Code generated by MockGen. DO NOT EDIT.
|
||||
// Source: github.com/golang/mock/gomock (interfaces: Matcher)
|
||||
|
||||
// Package mock_gomock is a generated GoMock package.
|
||||
package mock_gomock
|
||||
|
||||
import (
|
||||
gomock "github.com/golang/mock/gomock"
|
||||
reflect "reflect"
|
||||
)
|
||||
|
||||
// MockMatcher is a mock of Matcher interface
|
||||
type MockMatcher struct {
|
||||
ctrl *gomock.Controller
|
||||
recorder *MockMatcherMockRecorder
|
||||
}
|
||||
|
||||
// MockMatcherMockRecorder is the mock recorder for MockMatcher
|
||||
type MockMatcherMockRecorder struct {
|
||||
mock *MockMatcher
|
||||
}
|
||||
|
||||
// NewMockMatcher creates a new mock instance
|
||||
func NewMockMatcher(ctrl *gomock.Controller) *MockMatcher {
|
||||
mock := &MockMatcher{ctrl: ctrl}
|
||||
mock.recorder = &MockMatcherMockRecorder{mock}
|
||||
return mock
|
||||
}
|
||||
|
||||
// EXPECT returns an object that allows the caller to indicate expected use
|
||||
func (m *MockMatcher) EXPECT() *MockMatcherMockRecorder {
|
||||
return m.recorder
|
||||
}
|
||||
|
||||
// Matches mocks base method
|
||||
func (m *MockMatcher) Matches(arg0 interface{}) bool {
|
||||
ret := m.ctrl.Call(m, "Matches", arg0)
|
||||
ret0, _ := ret[0].(bool)
|
||||
return ret0
|
||||
}
|
||||
|
||||
// Matches indicates an expected call of Matches
|
||||
func (mr *MockMatcherMockRecorder) Matches(arg0 interface{}) *gomock.Call {
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Matches", reflect.TypeOf((*MockMatcher)(nil).Matches), arg0)
|
||||
}
|
||||
|
||||
// String mocks base method
|
||||
func (m *MockMatcher) String() string {
|
||||
ret := m.ctrl.Call(m, "String")
|
||||
ret0, _ := ret[0].(string)
|
||||
return ret0
|
||||
}
|
||||
|
||||
// String indicates an expected call of String
|
||||
func (mr *MockMatcherMockRecorder) String() *gomock.Call {
|
||||
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "String", reflect.TypeOf((*MockMatcher)(nil).String))
|
||||
}
|
Reference in New Issue
Block a user