vendor: add hashstructure and mock
This commit is contained in:
parent
46c570c414
commit
17af5e6a24
|
@ -0,0 +1,21 @@
|
||||||
|
The MIT License (MIT)
|
||||||
|
|
||||||
|
Copyright (c) 2016 Mitchell Hashimoto
|
||||||
|
|
||||||
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
||||||
|
of this software and associated documentation files (the "Software"), to deal
|
||||||
|
in the Software without restriction, including without limitation the rights
|
||||||
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
||||||
|
copies of the Software, and to permit persons to whom the Software is
|
||||||
|
furnished to do so, subject to the following conditions:
|
||||||
|
|
||||||
|
The above copyright notice and this permission notice shall be included in
|
||||||
|
all copies or substantial portions of the Software.
|
||||||
|
|
||||||
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||||
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||||
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
||||||
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
||||||
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
||||||
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
||||||
|
THE SOFTWARE.
|
|
@ -0,0 +1,65 @@
|
||||||
|
# hashstructure [![GoDoc](https://godoc.org/github.com/mitchellh/hashstructure?status.svg)](https://godoc.org/github.com/mitchellh/hashstructure)
|
||||||
|
|
||||||
|
hashstructure is a Go library for creating a unique hash value
|
||||||
|
for arbitrary values in Go.
|
||||||
|
|
||||||
|
This can be used to key values in a hash (for use in a map, set, etc.)
|
||||||
|
that are complex. The most common use case is comparing two values without
|
||||||
|
sending data across the network, caching values locally (de-dup), and so on.
|
||||||
|
|
||||||
|
## Features
|
||||||
|
|
||||||
|
* Hash any arbitrary Go value, including complex types.
|
||||||
|
|
||||||
|
* Tag a struct field to ignore it and not affect the hash value.
|
||||||
|
|
||||||
|
* Tag a slice type struct field to treat it as a set where ordering
|
||||||
|
doesn't affect the hash code but the field itself is still taken into
|
||||||
|
account to create the hash value.
|
||||||
|
|
||||||
|
* Optionally specify a custom hash function to optimize for speed, collision
|
||||||
|
avoidance for your data set, etc.
|
||||||
|
|
||||||
|
* Optionally hash the output of `.String()` on structs that implement fmt.Stringer,
|
||||||
|
allowing effective hashing of time.Time
|
||||||
|
|
||||||
|
## Installation
|
||||||
|
|
||||||
|
Standard `go get`:
|
||||||
|
|
||||||
|
```
|
||||||
|
$ go get github.com/mitchellh/hashstructure
|
||||||
|
```
|
||||||
|
|
||||||
|
## Usage & Example
|
||||||
|
|
||||||
|
For usage and examples see the [Godoc](http://godoc.org/github.com/mitchellh/hashstructure).
|
||||||
|
|
||||||
|
A quick code example is shown below:
|
||||||
|
|
||||||
|
```go
|
||||||
|
type ComplexStruct struct {
|
||||||
|
Name string
|
||||||
|
Age uint
|
||||||
|
Metadata map[string]interface{}
|
||||||
|
}
|
||||||
|
|
||||||
|
v := ComplexStruct{
|
||||||
|
Name: "mitchellh",
|
||||||
|
Age: 64,
|
||||||
|
Metadata: map[string]interface{}{
|
||||||
|
"car": true,
|
||||||
|
"location": "California",
|
||||||
|
"siblings": []string{"Bob", "John"},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
hash, err := hashstructure.Hash(v, nil)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Printf("%d", hash)
|
||||||
|
// Output:
|
||||||
|
// 2307517237273902113
|
||||||
|
```
|
|
@ -0,0 +1,358 @@
|
||||||
|
package hashstructure
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/binary"
|
||||||
|
"fmt"
|
||||||
|
"hash"
|
||||||
|
"hash/fnv"
|
||||||
|
"reflect"
|
||||||
|
)
|
||||||
|
|
||||||
|
// ErrNotStringer is returned when there's an error with hash:"string"
|
||||||
|
type ErrNotStringer struct {
|
||||||
|
Field string
|
||||||
|
}
|
||||||
|
|
||||||
|
// Error implements error for ErrNotStringer
|
||||||
|
func (ens *ErrNotStringer) Error() string {
|
||||||
|
return fmt.Sprintf("hashstructure: %s has hash:\"string\" set, but does not implement fmt.Stringer", ens.Field)
|
||||||
|
}
|
||||||
|
|
||||||
|
// HashOptions are options that are available for hashing.
|
||||||
|
type HashOptions struct {
|
||||||
|
// Hasher is the hash function to use. If this isn't set, it will
|
||||||
|
// default to FNV.
|
||||||
|
Hasher hash.Hash64
|
||||||
|
|
||||||
|
// TagName is the struct tag to look at when hashing the structure.
|
||||||
|
// By default this is "hash".
|
||||||
|
TagName string
|
||||||
|
|
||||||
|
// ZeroNil is flag determining if nil pointer should be treated equal
|
||||||
|
// to a zero value of pointed type. By default this is false.
|
||||||
|
ZeroNil bool
|
||||||
|
}
|
||||||
|
|
||||||
|
// Hash returns the hash value of an arbitrary value.
|
||||||
|
//
|
||||||
|
// If opts is nil, then default options will be used. See HashOptions
|
||||||
|
// for the default values. The same *HashOptions value cannot be used
|
||||||
|
// concurrently. None of the values within a *HashOptions struct are
|
||||||
|
// safe to read/write while hashing is being done.
|
||||||
|
//
|
||||||
|
// Notes on the value:
|
||||||
|
//
|
||||||
|
// * Unexported fields on structs are ignored and do not affect the
|
||||||
|
// hash value.
|
||||||
|
//
|
||||||
|
// * Adding an exported field to a struct with the zero value will change
|
||||||
|
// the hash value.
|
||||||
|
//
|
||||||
|
// For structs, the hashing can be controlled using tags. For example:
|
||||||
|
//
|
||||||
|
// struct {
|
||||||
|
// Name string
|
||||||
|
// UUID string `hash:"ignore"`
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// The available tag values are:
|
||||||
|
//
|
||||||
|
// * "ignore" or "-" - The field will be ignored and not affect the hash code.
|
||||||
|
//
|
||||||
|
// * "set" - The field will be treated as a set, where ordering doesn't
|
||||||
|
// affect the hash code. This only works for slices.
|
||||||
|
//
|
||||||
|
// * "string" - The field will be hashed as a string, only works when the
|
||||||
|
// field implements fmt.Stringer
|
||||||
|
//
|
||||||
|
func Hash(v interface{}, opts *HashOptions) (uint64, error) {
|
||||||
|
// Create default options
|
||||||
|
if opts == nil {
|
||||||
|
opts = &HashOptions{}
|
||||||
|
}
|
||||||
|
if opts.Hasher == nil {
|
||||||
|
opts.Hasher = fnv.New64()
|
||||||
|
}
|
||||||
|
if opts.TagName == "" {
|
||||||
|
opts.TagName = "hash"
|
||||||
|
}
|
||||||
|
|
||||||
|
// Reset the hash
|
||||||
|
opts.Hasher.Reset()
|
||||||
|
|
||||||
|
// Create our walker and walk the structure
|
||||||
|
w := &walker{
|
||||||
|
h: opts.Hasher,
|
||||||
|
tag: opts.TagName,
|
||||||
|
zeronil: opts.ZeroNil,
|
||||||
|
}
|
||||||
|
return w.visit(reflect.ValueOf(v), nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
type walker struct {
|
||||||
|
h hash.Hash64
|
||||||
|
tag string
|
||||||
|
zeronil bool
|
||||||
|
}
|
||||||
|
|
||||||
|
type visitOpts struct {
|
||||||
|
// Flags are a bitmask of flags to affect behavior of this visit
|
||||||
|
Flags visitFlag
|
||||||
|
|
||||||
|
// Information about the struct containing this field
|
||||||
|
Struct interface{}
|
||||||
|
StructField string
|
||||||
|
}
|
||||||
|
|
||||||
|
func (w *walker) visit(v reflect.Value, opts *visitOpts) (uint64, error) {
|
||||||
|
t := reflect.TypeOf(0)
|
||||||
|
|
||||||
|
// Loop since these can be wrapped in multiple layers of pointers
|
||||||
|
// and interfaces.
|
||||||
|
for {
|
||||||
|
// If we have an interface, dereference it. We have to do this up
|
||||||
|
// here because it might be a nil in there and the check below must
|
||||||
|
// catch that.
|
||||||
|
if v.Kind() == reflect.Interface {
|
||||||
|
v = v.Elem()
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if v.Kind() == reflect.Ptr {
|
||||||
|
if w.zeronil {
|
||||||
|
t = v.Type().Elem()
|
||||||
|
}
|
||||||
|
v = reflect.Indirect(v)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
// If it is nil, treat it like a zero.
|
||||||
|
if !v.IsValid() {
|
||||||
|
v = reflect.Zero(t)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Binary writing can use raw ints, we have to convert to
|
||||||
|
// a sized-int, we'll choose the largest...
|
||||||
|
switch v.Kind() {
|
||||||
|
case reflect.Int:
|
||||||
|
v = reflect.ValueOf(int64(v.Int()))
|
||||||
|
case reflect.Uint:
|
||||||
|
v = reflect.ValueOf(uint64(v.Uint()))
|
||||||
|
case reflect.Bool:
|
||||||
|
var tmp int8
|
||||||
|
if v.Bool() {
|
||||||
|
tmp = 1
|
||||||
|
}
|
||||||
|
v = reflect.ValueOf(tmp)
|
||||||
|
}
|
||||||
|
|
||||||
|
k := v.Kind()
|
||||||
|
|
||||||
|
// We can shortcut numeric values by directly binary writing them
|
||||||
|
if k >= reflect.Int && k <= reflect.Complex64 {
|
||||||
|
// A direct hash calculation
|
||||||
|
w.h.Reset()
|
||||||
|
err := binary.Write(w.h, binary.LittleEndian, v.Interface())
|
||||||
|
return w.h.Sum64(), err
|
||||||
|
}
|
||||||
|
|
||||||
|
switch k {
|
||||||
|
case reflect.Array:
|
||||||
|
var h uint64
|
||||||
|
l := v.Len()
|
||||||
|
for i := 0; i < l; i++ {
|
||||||
|
current, err := w.visit(v.Index(i), nil)
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
|
||||||
|
h = hashUpdateOrdered(w.h, h, current)
|
||||||
|
}
|
||||||
|
|
||||||
|
return h, nil
|
||||||
|
|
||||||
|
case reflect.Map:
|
||||||
|
var includeMap IncludableMap
|
||||||
|
if opts != nil && opts.Struct != nil {
|
||||||
|
if v, ok := opts.Struct.(IncludableMap); ok {
|
||||||
|
includeMap = v
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Build the hash for the map. We do this by XOR-ing all the key
|
||||||
|
// and value hashes. This makes it deterministic despite ordering.
|
||||||
|
var h uint64
|
||||||
|
for _, k := range v.MapKeys() {
|
||||||
|
v := v.MapIndex(k)
|
||||||
|
if includeMap != nil {
|
||||||
|
incl, err := includeMap.HashIncludeMap(
|
||||||
|
opts.StructField, k.Interface(), v.Interface())
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
if !incl {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
kh, err := w.visit(k, nil)
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
vh, err := w.visit(v, nil)
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
|
||||||
|
fieldHash := hashUpdateOrdered(w.h, kh, vh)
|
||||||
|
h = hashUpdateUnordered(h, fieldHash)
|
||||||
|
}
|
||||||
|
|
||||||
|
return h, nil
|
||||||
|
|
||||||
|
case reflect.Struct:
|
||||||
|
parent := v.Interface()
|
||||||
|
var include Includable
|
||||||
|
if impl, ok := parent.(Includable); ok {
|
||||||
|
include = impl
|
||||||
|
}
|
||||||
|
|
||||||
|
t := v.Type()
|
||||||
|
h, err := w.visit(reflect.ValueOf(t.Name()), nil)
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
|
||||||
|
l := v.NumField()
|
||||||
|
for i := 0; i < l; i++ {
|
||||||
|
if innerV := v.Field(i); v.CanSet() || t.Field(i).Name != "_" {
|
||||||
|
var f visitFlag
|
||||||
|
fieldType := t.Field(i)
|
||||||
|
if fieldType.PkgPath != "" {
|
||||||
|
// Unexported
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
tag := fieldType.Tag.Get(w.tag)
|
||||||
|
if tag == "ignore" || tag == "-" {
|
||||||
|
// Ignore this field
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// if string is set, use the string value
|
||||||
|
if tag == "string" {
|
||||||
|
if impl, ok := innerV.Interface().(fmt.Stringer); ok {
|
||||||
|
innerV = reflect.ValueOf(impl.String())
|
||||||
|
} else {
|
||||||
|
return 0, &ErrNotStringer{
|
||||||
|
Field: v.Type().Field(i).Name,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if we implement includable and check it
|
||||||
|
if include != nil {
|
||||||
|
incl, err := include.HashInclude(fieldType.Name, innerV)
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
if !incl {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
switch tag {
|
||||||
|
case "set":
|
||||||
|
f |= visitFlagSet
|
||||||
|
}
|
||||||
|
|
||||||
|
kh, err := w.visit(reflect.ValueOf(fieldType.Name), nil)
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
|
||||||
|
vh, err := w.visit(innerV, &visitOpts{
|
||||||
|
Flags: f,
|
||||||
|
Struct: parent,
|
||||||
|
StructField: fieldType.Name,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
|
||||||
|
fieldHash := hashUpdateOrdered(w.h, kh, vh)
|
||||||
|
h = hashUpdateUnordered(h, fieldHash)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return h, nil
|
||||||
|
|
||||||
|
case reflect.Slice:
|
||||||
|
// We have two behaviors here. If it isn't a set, then we just
|
||||||
|
// visit all the elements. If it is a set, then we do a deterministic
|
||||||
|
// hash code.
|
||||||
|
var h uint64
|
||||||
|
var set bool
|
||||||
|
if opts != nil {
|
||||||
|
set = (opts.Flags & visitFlagSet) != 0
|
||||||
|
}
|
||||||
|
l := v.Len()
|
||||||
|
for i := 0; i < l; i++ {
|
||||||
|
current, err := w.visit(v.Index(i), nil)
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if set {
|
||||||
|
h = hashUpdateUnordered(h, current)
|
||||||
|
} else {
|
||||||
|
h = hashUpdateOrdered(w.h, h, current)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return h, nil
|
||||||
|
|
||||||
|
case reflect.String:
|
||||||
|
// Directly hash
|
||||||
|
w.h.Reset()
|
||||||
|
_, err := w.h.Write([]byte(v.String()))
|
||||||
|
return w.h.Sum64(), err
|
||||||
|
|
||||||
|
default:
|
||||||
|
return 0, fmt.Errorf("unknown kind to hash: %s", k)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func hashUpdateOrdered(h hash.Hash64, a, b uint64) uint64 {
|
||||||
|
// For ordered updates, use a real hash function
|
||||||
|
h.Reset()
|
||||||
|
|
||||||
|
// We just panic if the binary writes fail because we are writing
|
||||||
|
// an int64 which should never be fail-able.
|
||||||
|
e1 := binary.Write(h, binary.LittleEndian, a)
|
||||||
|
e2 := binary.Write(h, binary.LittleEndian, b)
|
||||||
|
if e1 != nil {
|
||||||
|
panic(e1)
|
||||||
|
}
|
||||||
|
if e2 != nil {
|
||||||
|
panic(e2)
|
||||||
|
}
|
||||||
|
|
||||||
|
return h.Sum64()
|
||||||
|
}
|
||||||
|
|
||||||
|
func hashUpdateUnordered(a, b uint64) uint64 {
|
||||||
|
return a ^ b
|
||||||
|
}
|
||||||
|
|
||||||
|
// visitFlag is used as a bitmask for affecting visit behavior
|
||||||
|
type visitFlag uint
|
||||||
|
|
||||||
|
const (
|
||||||
|
visitFlagInvalid visitFlag = iota
|
||||||
|
visitFlagSet = iota << 1
|
||||||
|
)
|
|
@ -0,0 +1,15 @@
|
||||||
|
package hashstructure
|
||||||
|
|
||||||
|
// Includable is an interface that can optionally be implemented by
|
||||||
|
// a struct. It will be called for each field in the struct to check whether
|
||||||
|
// it should be included in the hash.
|
||||||
|
type Includable interface {
|
||||||
|
HashInclude(field string, v interface{}) (bool, error)
|
||||||
|
}
|
||||||
|
|
||||||
|
// IncludableMap is an interface that can optionally be implemented by
|
||||||
|
// a struct. It will be called when a map-type field is found to ask the
|
||||||
|
// struct if the map item should be included in the hash.
|
||||||
|
type IncludableMap interface {
|
||||||
|
HashIncludeMap(field string, k, v interface{}) (bool, error)
|
||||||
|
}
|
|
@ -0,0 +1,44 @@
|
||||||
|
// Package mock provides a system by which it is possible to mock your objects
|
||||||
|
// and verify calls are happening as expected.
|
||||||
|
//
|
||||||
|
// Example Usage
|
||||||
|
//
|
||||||
|
// The mock package provides an object, Mock, that tracks activity on another object. It is usually
|
||||||
|
// embedded into a test object as shown below:
|
||||||
|
//
|
||||||
|
// type MyTestObject struct {
|
||||||
|
// // add a Mock object instance
|
||||||
|
// mock.Mock
|
||||||
|
//
|
||||||
|
// // other fields go here as normal
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// When implementing the methods of an interface, you wire your functions up
|
||||||
|
// to call the Mock.Called(args...) method, and return the appropriate values.
|
||||||
|
//
|
||||||
|
// For example, to mock a method that saves the name and age of a person and returns
|
||||||
|
// the year of their birth or an error, you might write this:
|
||||||
|
//
|
||||||
|
// func (o *MyTestObject) SavePersonDetails(firstname, lastname string, age int) (int, error) {
|
||||||
|
// args := o.Called(firstname, lastname, age)
|
||||||
|
// return args.Int(0), args.Error(1)
|
||||||
|
// }
|
||||||
|
//
|
||||||
|
// The Int, Error and Bool methods are examples of strongly typed getters that take the argument
|
||||||
|
// index position. Given this argument list:
|
||||||
|
//
|
||||||
|
// (12, true, "Something")
|
||||||
|
//
|
||||||
|
// You could read them out strongly typed like this:
|
||||||
|
//
|
||||||
|
// args.Int(0)
|
||||||
|
// args.Bool(1)
|
||||||
|
// args.String(2)
|
||||||
|
//
|
||||||
|
// For objects of your own type, use the generic Arguments.Get(index) method and make a type assertion:
|
||||||
|
//
|
||||||
|
// return args.Get(0).(*MyObject), args.Get(1).(*AnotherObjectOfMine)
|
||||||
|
//
|
||||||
|
// This may cause a panic if the object you are getting is nil (the type assertion will fail), in those
|
||||||
|
// cases you should check for nil first.
|
||||||
|
package mock
|
|
@ -0,0 +1,880 @@
|
||||||
|
package mock
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"reflect"
|
||||||
|
"regexp"
|
||||||
|
"runtime"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/davecgh/go-spew/spew"
|
||||||
|
"github.com/pmezard/go-difflib/difflib"
|
||||||
|
"github.com/stretchr/objx"
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
)
|
||||||
|
|
||||||
|
// TestingT is an interface wrapper around *testing.T
|
||||||
|
type TestingT interface {
|
||||||
|
Logf(format string, args ...interface{})
|
||||||
|
Errorf(format string, args ...interface{})
|
||||||
|
FailNow()
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
Call
|
||||||
|
*/
|
||||||
|
|
||||||
|
// Call represents a method call and is used for setting expectations,
|
||||||
|
// as well as recording activity.
|
||||||
|
type Call struct {
|
||||||
|
Parent *Mock
|
||||||
|
|
||||||
|
// The name of the method that was or will be called.
|
||||||
|
Method string
|
||||||
|
|
||||||
|
// Holds the arguments of the method.
|
||||||
|
Arguments Arguments
|
||||||
|
|
||||||
|
// Holds the arguments that should be returned when
|
||||||
|
// this method is called.
|
||||||
|
ReturnArguments Arguments
|
||||||
|
|
||||||
|
// Holds the caller info for the On() call
|
||||||
|
callerInfo []string
|
||||||
|
|
||||||
|
// The number of times to return the return arguments when setting
|
||||||
|
// expectations. 0 means to always return the value.
|
||||||
|
Repeatability int
|
||||||
|
|
||||||
|
// Amount of times this call has been called
|
||||||
|
totalCalls int
|
||||||
|
|
||||||
|
// Call to this method can be optional
|
||||||
|
optional bool
|
||||||
|
|
||||||
|
// Holds a channel that will be used to block the Return until it either
|
||||||
|
// receives a message or is closed. nil means it returns immediately.
|
||||||
|
WaitFor <-chan time.Time
|
||||||
|
|
||||||
|
waitTime time.Duration
|
||||||
|
|
||||||
|
// Holds a handler used to manipulate arguments content that are passed by
|
||||||
|
// reference. It's useful when mocking methods such as unmarshalers or
|
||||||
|
// decoders.
|
||||||
|
RunFn func(Arguments)
|
||||||
|
}
|
||||||
|
|
||||||
|
func newCall(parent *Mock, methodName string, callerInfo []string, methodArguments ...interface{}) *Call {
|
||||||
|
return &Call{
|
||||||
|
Parent: parent,
|
||||||
|
Method: methodName,
|
||||||
|
Arguments: methodArguments,
|
||||||
|
ReturnArguments: make([]interface{}, 0),
|
||||||
|
callerInfo: callerInfo,
|
||||||
|
Repeatability: 0,
|
||||||
|
WaitFor: nil,
|
||||||
|
RunFn: nil,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Call) lock() {
|
||||||
|
c.Parent.mutex.Lock()
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *Call) unlock() {
|
||||||
|
c.Parent.mutex.Unlock()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Return specifies the return arguments for the expectation.
|
||||||
|
//
|
||||||
|
// Mock.On("DoSomething").Return(errors.New("failed"))
|
||||||
|
func (c *Call) Return(returnArguments ...interface{}) *Call {
|
||||||
|
c.lock()
|
||||||
|
defer c.unlock()
|
||||||
|
|
||||||
|
c.ReturnArguments = returnArguments
|
||||||
|
|
||||||
|
return c
|
||||||
|
}
|
||||||
|
|
||||||
|
// Once indicates that that the mock should only return the value once.
|
||||||
|
//
|
||||||
|
// Mock.On("MyMethod", arg1, arg2).Return(returnArg1, returnArg2).Once()
|
||||||
|
func (c *Call) Once() *Call {
|
||||||
|
return c.Times(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Twice indicates that that the mock should only return the value twice.
|
||||||
|
//
|
||||||
|
// Mock.On("MyMethod", arg1, arg2).Return(returnArg1, returnArg2).Twice()
|
||||||
|
func (c *Call) Twice() *Call {
|
||||||
|
return c.Times(2)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Times indicates that that the mock should only return the indicated number
|
||||||
|
// of times.
|
||||||
|
//
|
||||||
|
// Mock.On("MyMethod", arg1, arg2).Return(returnArg1, returnArg2).Times(5)
|
||||||
|
func (c *Call) Times(i int) *Call {
|
||||||
|
c.lock()
|
||||||
|
defer c.unlock()
|
||||||
|
c.Repeatability = i
|
||||||
|
return c
|
||||||
|
}
|
||||||
|
|
||||||
|
// WaitUntil sets the channel that will block the mock's return until its closed
|
||||||
|
// or a message is received.
|
||||||
|
//
|
||||||
|
// Mock.On("MyMethod", arg1, arg2).WaitUntil(time.After(time.Second))
|
||||||
|
func (c *Call) WaitUntil(w <-chan time.Time) *Call {
|
||||||
|
c.lock()
|
||||||
|
defer c.unlock()
|
||||||
|
c.WaitFor = w
|
||||||
|
return c
|
||||||
|
}
|
||||||
|
|
||||||
|
// After sets how long to block until the call returns
|
||||||
|
//
|
||||||
|
// Mock.On("MyMethod", arg1, arg2).After(time.Second)
|
||||||
|
func (c *Call) After(d time.Duration) *Call {
|
||||||
|
c.lock()
|
||||||
|
defer c.unlock()
|
||||||
|
c.waitTime = d
|
||||||
|
return c
|
||||||
|
}
|
||||||
|
|
||||||
|
// Run sets a handler to be called before returning. It can be used when
|
||||||
|
// mocking a method such as unmarshalers that takes a pointer to a struct and
|
||||||
|
// sets properties in such struct
|
||||||
|
//
|
||||||
|
// Mock.On("Unmarshal", AnythingOfType("*map[string]interface{}").Return().Run(func(args Arguments) {
|
||||||
|
// arg := args.Get(0).(*map[string]interface{})
|
||||||
|
// arg["foo"] = "bar"
|
||||||
|
// })
|
||||||
|
func (c *Call) Run(fn func(args Arguments)) *Call {
|
||||||
|
c.lock()
|
||||||
|
defer c.unlock()
|
||||||
|
c.RunFn = fn
|
||||||
|
return c
|
||||||
|
}
|
||||||
|
|
||||||
|
// Maybe allows the method call to be optional. Not calling an optional method
|
||||||
|
// will not cause an error while asserting expectations
|
||||||
|
func (c *Call) Maybe() *Call {
|
||||||
|
c.lock()
|
||||||
|
defer c.unlock()
|
||||||
|
c.optional = true
|
||||||
|
return c
|
||||||
|
}
|
||||||
|
|
||||||
|
// On chains a new expectation description onto the mocked interface. This
|
||||||
|
// allows syntax like.
|
||||||
|
//
|
||||||
|
// Mock.
|
||||||
|
// On("MyMethod", 1).Return(nil).
|
||||||
|
// On("MyOtherMethod", 'a', 'b', 'c').Return(errors.New("Some Error"))
|
||||||
|
func (c *Call) On(methodName string, arguments ...interface{}) *Call {
|
||||||
|
return c.Parent.On(methodName, arguments...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Mock is the workhorse used to track activity on another object.
|
||||||
|
// For an example of its usage, refer to the "Example Usage" section at the top
|
||||||
|
// of this document.
|
||||||
|
type Mock struct {
|
||||||
|
// Represents the calls that are expected of
|
||||||
|
// an object.
|
||||||
|
ExpectedCalls []*Call
|
||||||
|
|
||||||
|
// Holds the calls that were made to this mocked object.
|
||||||
|
Calls []Call
|
||||||
|
|
||||||
|
// test is An optional variable that holds the test struct, to be used when an
|
||||||
|
// invalid mock call was made.
|
||||||
|
test TestingT
|
||||||
|
|
||||||
|
// TestData holds any data that might be useful for testing. Testify ignores
|
||||||
|
// this data completely allowing you to do whatever you like with it.
|
||||||
|
testData objx.Map
|
||||||
|
|
||||||
|
mutex sync.Mutex
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestData holds any data that might be useful for testing. Testify ignores
|
||||||
|
// this data completely allowing you to do whatever you like with it.
|
||||||
|
func (m *Mock) TestData() objx.Map {
|
||||||
|
|
||||||
|
if m.testData == nil {
|
||||||
|
m.testData = make(objx.Map)
|
||||||
|
}
|
||||||
|
|
||||||
|
return m.testData
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
Setting expectations
|
||||||
|
*/
|
||||||
|
|
||||||
|
// Test sets the test struct variable of the mock object
|
||||||
|
func (m *Mock) Test(t TestingT) {
|
||||||
|
m.mutex.Lock()
|
||||||
|
defer m.mutex.Unlock()
|
||||||
|
m.test = t
|
||||||
|
}
|
||||||
|
|
||||||
|
// fail fails the current test with the given formatted format and args.
|
||||||
|
// In case that a test was defined, it uses the test APIs for failing a test,
|
||||||
|
// otherwise it uses panic.
|
||||||
|
func (m *Mock) fail(format string, args ...interface{}) {
|
||||||
|
m.mutex.Lock()
|
||||||
|
defer m.mutex.Unlock()
|
||||||
|
|
||||||
|
if m.test == nil {
|
||||||
|
panic(fmt.Sprintf(format, args...))
|
||||||
|
}
|
||||||
|
m.test.Errorf(format, args...)
|
||||||
|
m.test.FailNow()
|
||||||
|
}
|
||||||
|
|
||||||
|
// On starts a description of an expectation of the specified method
|
||||||
|
// being called.
|
||||||
|
//
|
||||||
|
// Mock.On("MyMethod", arg1, arg2)
|
||||||
|
func (m *Mock) On(methodName string, arguments ...interface{}) *Call {
|
||||||
|
for _, arg := range arguments {
|
||||||
|
if v := reflect.ValueOf(arg); v.Kind() == reflect.Func {
|
||||||
|
panic(fmt.Sprintf("cannot use Func in expectations. Use mock.AnythingOfType(\"%T\")", arg))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
m.mutex.Lock()
|
||||||
|
defer m.mutex.Unlock()
|
||||||
|
c := newCall(m, methodName, assert.CallerInfo(), arguments...)
|
||||||
|
m.ExpectedCalls = append(m.ExpectedCalls, c)
|
||||||
|
return c
|
||||||
|
}
|
||||||
|
|
||||||
|
// /*
|
||||||
|
// Recording and responding to activity
|
||||||
|
// */
|
||||||
|
|
||||||
|
func (m *Mock) findExpectedCall(method string, arguments ...interface{}) (int, *Call) {
|
||||||
|
for i, call := range m.ExpectedCalls {
|
||||||
|
if call.Method == method && call.Repeatability > -1 {
|
||||||
|
|
||||||
|
_, diffCount := call.Arguments.Diff(arguments)
|
||||||
|
if diffCount == 0 {
|
||||||
|
return i, call
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return -1, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Mock) findClosestCall(method string, arguments ...interface{}) (*Call, string) {
|
||||||
|
var diffCount int
|
||||||
|
var closestCall *Call
|
||||||
|
var err string
|
||||||
|
|
||||||
|
for _, call := range m.expectedCalls() {
|
||||||
|
if call.Method == method {
|
||||||
|
|
||||||
|
errInfo, tempDiffCount := call.Arguments.Diff(arguments)
|
||||||
|
if tempDiffCount < diffCount || diffCount == 0 {
|
||||||
|
diffCount = tempDiffCount
|
||||||
|
closestCall = call
|
||||||
|
err = errInfo
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return closestCall, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func callString(method string, arguments Arguments, includeArgumentValues bool) string {
|
||||||
|
|
||||||
|
var argValsString string
|
||||||
|
if includeArgumentValues {
|
||||||
|
var argVals []string
|
||||||
|
for argIndex, arg := range arguments {
|
||||||
|
argVals = append(argVals, fmt.Sprintf("%d: %#v", argIndex, arg))
|
||||||
|
}
|
||||||
|
argValsString = fmt.Sprintf("\n\t\t%s", strings.Join(argVals, "\n\t\t"))
|
||||||
|
}
|
||||||
|
|
||||||
|
return fmt.Sprintf("%s(%s)%s", method, arguments.String(), argValsString)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Called tells the mock object that a method has been called, and gets an array
|
||||||
|
// of arguments to return. Panics if the call is unexpected (i.e. not preceded by
|
||||||
|
// appropriate .On .Return() calls)
|
||||||
|
// If Call.WaitFor is set, blocks until the channel is closed or receives a message.
|
||||||
|
func (m *Mock) Called(arguments ...interface{}) Arguments {
|
||||||
|
// get the calling function's name
|
||||||
|
pc, _, _, ok := runtime.Caller(1)
|
||||||
|
if !ok {
|
||||||
|
panic("Couldn't get the caller information")
|
||||||
|
}
|
||||||
|
functionPath := runtime.FuncForPC(pc).Name()
|
||||||
|
//Next four lines are required to use GCCGO function naming conventions.
|
||||||
|
//For Ex: github_com_docker_libkv_store_mock.WatchTree.pN39_github_com_docker_libkv_store_mock.Mock
|
||||||
|
//uses interface information unlike golang github.com/docker/libkv/store/mock.(*Mock).WatchTree
|
||||||
|
//With GCCGO we need to remove interface information starting from pN<dd>.
|
||||||
|
re := regexp.MustCompile("\\.pN\\d+_")
|
||||||
|
if re.MatchString(functionPath) {
|
||||||
|
functionPath = re.Split(functionPath, -1)[0]
|
||||||
|
}
|
||||||
|
parts := strings.Split(functionPath, ".")
|
||||||
|
functionName := parts[len(parts)-1]
|
||||||
|
return m.MethodCalled(functionName, arguments...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// MethodCalled tells the mock object that the given method has been called, and gets
|
||||||
|
// an array of arguments to return. Panics if the call is unexpected (i.e. not preceded
|
||||||
|
// by appropriate .On .Return() calls)
|
||||||
|
// If Call.WaitFor is set, blocks until the channel is closed or receives a message.
|
||||||
|
func (m *Mock) MethodCalled(methodName string, arguments ...interface{}) Arguments {
|
||||||
|
m.mutex.Lock()
|
||||||
|
//TODO: could combine expected and closes in single loop
|
||||||
|
found, call := m.findExpectedCall(methodName, arguments...)
|
||||||
|
|
||||||
|
if found < 0 {
|
||||||
|
// we have to fail here - because we don't know what to do
|
||||||
|
// as the return arguments. This is because:
|
||||||
|
//
|
||||||
|
// a) this is a totally unexpected call to this method,
|
||||||
|
// b) the arguments are not what was expected, or
|
||||||
|
// c) the developer has forgotten to add an accompanying On...Return pair.
|
||||||
|
|
||||||
|
closestCall, mismatch := m.findClosestCall(methodName, arguments...)
|
||||||
|
m.mutex.Unlock()
|
||||||
|
|
||||||
|
if closestCall != nil {
|
||||||
|
m.fail("\n\nmock: Unexpected Method Call\n-----------------------------\n\n%s\n\nThe closest call I have is: \n\n%s\n\n%s\nDiff: %s",
|
||||||
|
callString(methodName, arguments, true),
|
||||||
|
callString(methodName, closestCall.Arguments, true),
|
||||||
|
diffArguments(closestCall.Arguments, arguments),
|
||||||
|
strings.TrimSpace(mismatch),
|
||||||
|
)
|
||||||
|
} else {
|
||||||
|
m.fail("\nassert: mock: I don't know what to return because the method call was unexpected.\n\tEither do Mock.On(\"%s\").Return(...) first, or remove the %s() call.\n\tThis method was unexpected:\n\t\t%s\n\tat: %s", methodName, methodName, callString(methodName, arguments, true), assert.CallerInfo())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if call.Repeatability == 1 {
|
||||||
|
call.Repeatability = -1
|
||||||
|
} else if call.Repeatability > 1 {
|
||||||
|
call.Repeatability--
|
||||||
|
}
|
||||||
|
call.totalCalls++
|
||||||
|
|
||||||
|
// add the call
|
||||||
|
m.Calls = append(m.Calls, *newCall(m, methodName, assert.CallerInfo(), arguments...))
|
||||||
|
m.mutex.Unlock()
|
||||||
|
|
||||||
|
// block if specified
|
||||||
|
if call.WaitFor != nil {
|
||||||
|
<-call.WaitFor
|
||||||
|
} else {
|
||||||
|
time.Sleep(call.waitTime)
|
||||||
|
}
|
||||||
|
|
||||||
|
m.mutex.Lock()
|
||||||
|
runFn := call.RunFn
|
||||||
|
m.mutex.Unlock()
|
||||||
|
|
||||||
|
if runFn != nil {
|
||||||
|
runFn(arguments)
|
||||||
|
}
|
||||||
|
|
||||||
|
m.mutex.Lock()
|
||||||
|
returnArgs := call.ReturnArguments
|
||||||
|
m.mutex.Unlock()
|
||||||
|
|
||||||
|
return returnArgs
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
Assertions
|
||||||
|
*/
|
||||||
|
|
||||||
|
type assertExpectationser interface {
|
||||||
|
AssertExpectations(TestingT) bool
|
||||||
|
}
|
||||||
|
|
||||||
|
// AssertExpectationsForObjects asserts that everything specified with On and Return
|
||||||
|
// of the specified objects was in fact called as expected.
|
||||||
|
//
|
||||||
|
// Calls may have occurred in any order.
|
||||||
|
func AssertExpectationsForObjects(t TestingT, testObjects ...interface{}) bool {
|
||||||
|
if h, ok := t.(tHelper); ok {
|
||||||
|
h.Helper()
|
||||||
|
}
|
||||||
|
for _, obj := range testObjects {
|
||||||
|
if m, ok := obj.(Mock); ok {
|
||||||
|
t.Logf("Deprecated mock.AssertExpectationsForObjects(myMock.Mock) use mock.AssertExpectationsForObjects(myMock)")
|
||||||
|
obj = &m
|
||||||
|
}
|
||||||
|
m := obj.(assertExpectationser)
|
||||||
|
if !m.AssertExpectations(t) {
|
||||||
|
t.Logf("Expectations didn't match for Mock: %+v", reflect.TypeOf(m))
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// AssertExpectations asserts that everything specified with On and Return was
|
||||||
|
// in fact called as expected. Calls may have occurred in any order.
|
||||||
|
func (m *Mock) AssertExpectations(t TestingT) bool {
|
||||||
|
if h, ok := t.(tHelper); ok {
|
||||||
|
h.Helper()
|
||||||
|
}
|
||||||
|
m.mutex.Lock()
|
||||||
|
defer m.mutex.Unlock()
|
||||||
|
var somethingMissing bool
|
||||||
|
var failedExpectations int
|
||||||
|
|
||||||
|
// iterate through each expectation
|
||||||
|
expectedCalls := m.expectedCalls()
|
||||||
|
for _, expectedCall := range expectedCalls {
|
||||||
|
if !expectedCall.optional && !m.methodWasCalled(expectedCall.Method, expectedCall.Arguments) && expectedCall.totalCalls == 0 {
|
||||||
|
somethingMissing = true
|
||||||
|
failedExpectations++
|
||||||
|
t.Logf("FAIL:\t%s(%s)\n\t\tat: %s", expectedCall.Method, expectedCall.Arguments.String(), expectedCall.callerInfo)
|
||||||
|
} else {
|
||||||
|
if expectedCall.Repeatability > 0 {
|
||||||
|
somethingMissing = true
|
||||||
|
failedExpectations++
|
||||||
|
t.Logf("FAIL:\t%s(%s)\n\t\tat: %s", expectedCall.Method, expectedCall.Arguments.String(), expectedCall.callerInfo)
|
||||||
|
} else {
|
||||||
|
t.Logf("PASS:\t%s(%s)", expectedCall.Method, expectedCall.Arguments.String())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if somethingMissing {
|
||||||
|
t.Errorf("FAIL: %d out of %d expectation(s) were met.\n\tThe code you are testing needs to make %d more call(s).\n\tat: %s", len(expectedCalls)-failedExpectations, len(expectedCalls), failedExpectations, assert.CallerInfo())
|
||||||
|
}
|
||||||
|
|
||||||
|
return !somethingMissing
|
||||||
|
}
|
||||||
|
|
||||||
|
// AssertNumberOfCalls asserts that the method was called expectedCalls times.
|
||||||
|
func (m *Mock) AssertNumberOfCalls(t TestingT, methodName string, expectedCalls int) bool {
|
||||||
|
if h, ok := t.(tHelper); ok {
|
||||||
|
h.Helper()
|
||||||
|
}
|
||||||
|
m.mutex.Lock()
|
||||||
|
defer m.mutex.Unlock()
|
||||||
|
var actualCalls int
|
||||||
|
for _, call := range m.calls() {
|
||||||
|
if call.Method == methodName {
|
||||||
|
actualCalls++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return assert.Equal(t, expectedCalls, actualCalls, fmt.Sprintf("Expected number of calls (%d) does not match the actual number of calls (%d).", expectedCalls, actualCalls))
|
||||||
|
}
|
||||||
|
|
||||||
|
// AssertCalled asserts that the method was called.
|
||||||
|
// It can produce a false result when an argument is a pointer type and the underlying value changed after calling the mocked method.
|
||||||
|
func (m *Mock) AssertCalled(t TestingT, methodName string, arguments ...interface{}) bool {
|
||||||
|
if h, ok := t.(tHelper); ok {
|
||||||
|
h.Helper()
|
||||||
|
}
|
||||||
|
m.mutex.Lock()
|
||||||
|
defer m.mutex.Unlock()
|
||||||
|
if !m.methodWasCalled(methodName, arguments) {
|
||||||
|
var calledWithArgs []string
|
||||||
|
for _, call := range m.calls() {
|
||||||
|
calledWithArgs = append(calledWithArgs, fmt.Sprintf("%v", call.Arguments))
|
||||||
|
}
|
||||||
|
if len(calledWithArgs) == 0 {
|
||||||
|
return assert.Fail(t, "Should have called with given arguments",
|
||||||
|
fmt.Sprintf("Expected %q to have been called with:\n%v\nbut no actual calls happened", methodName, arguments))
|
||||||
|
}
|
||||||
|
return assert.Fail(t, "Should have called with given arguments",
|
||||||
|
fmt.Sprintf("Expected %q to have been called with:\n%v\nbut actual calls were:\n %v", methodName, arguments, strings.Join(calledWithArgs, "\n")))
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// AssertNotCalled asserts that the method was not called.
|
||||||
|
// It can produce a false result when an argument is a pointer type and the underlying value changed after calling the mocked method.
|
||||||
|
func (m *Mock) AssertNotCalled(t TestingT, methodName string, arguments ...interface{}) bool {
|
||||||
|
if h, ok := t.(tHelper); ok {
|
||||||
|
h.Helper()
|
||||||
|
}
|
||||||
|
m.mutex.Lock()
|
||||||
|
defer m.mutex.Unlock()
|
||||||
|
if m.methodWasCalled(methodName, arguments) {
|
||||||
|
return assert.Fail(t, "Should not have called with given arguments",
|
||||||
|
fmt.Sprintf("Expected %q to not have been called with:\n%v\nbut actually it was.", methodName, arguments))
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Mock) methodWasCalled(methodName string, expected []interface{}) bool {
|
||||||
|
for _, call := range m.calls() {
|
||||||
|
if call.Method == methodName {
|
||||||
|
|
||||||
|
_, differences := Arguments(expected).Diff(call.Arguments)
|
||||||
|
|
||||||
|
if differences == 0 {
|
||||||
|
// found the expected call
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// we didn't find the expected call
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Mock) expectedCalls() []*Call {
|
||||||
|
return append([]*Call{}, m.ExpectedCalls...)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (m *Mock) calls() []Call {
|
||||||
|
return append([]Call{}, m.Calls...)
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
Arguments
|
||||||
|
*/
|
||||||
|
|
||||||
|
// Arguments holds an array of method arguments or return values.
|
||||||
|
type Arguments []interface{}
|
||||||
|
|
||||||
|
const (
|
||||||
|
// Anything is used in Diff and Assert when the argument being tested
|
||||||
|
// shouldn't be taken into consideration.
|
||||||
|
Anything = "mock.Anything"
|
||||||
|
)
|
||||||
|
|
||||||
|
// AnythingOfTypeArgument is a string that contains the type of an argument
|
||||||
|
// for use when type checking. Used in Diff and Assert.
|
||||||
|
type AnythingOfTypeArgument string
|
||||||
|
|
||||||
|
// AnythingOfType returns an AnythingOfTypeArgument object containing the
|
||||||
|
// name of the type to check for. Used in Diff and Assert.
|
||||||
|
//
|
||||||
|
// For example:
|
||||||
|
// Assert(t, AnythingOfType("string"), AnythingOfType("int"))
|
||||||
|
func AnythingOfType(t string) AnythingOfTypeArgument {
|
||||||
|
return AnythingOfTypeArgument(t)
|
||||||
|
}
|
||||||
|
|
||||||
|
// argumentMatcher performs custom argument matching, returning whether or
|
||||||
|
// not the argument is matched by the expectation fixture function.
|
||||||
|
type argumentMatcher struct {
|
||||||
|
// fn is a function which accepts one argument, and returns a bool.
|
||||||
|
fn reflect.Value
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f argumentMatcher) Matches(argument interface{}) bool {
|
||||||
|
expectType := f.fn.Type().In(0)
|
||||||
|
expectTypeNilSupported := false
|
||||||
|
switch expectType.Kind() {
|
||||||
|
case reflect.Interface, reflect.Chan, reflect.Func, reflect.Map, reflect.Slice, reflect.Ptr:
|
||||||
|
expectTypeNilSupported = true
|
||||||
|
}
|
||||||
|
|
||||||
|
argType := reflect.TypeOf(argument)
|
||||||
|
var arg reflect.Value
|
||||||
|
if argType == nil {
|
||||||
|
arg = reflect.New(expectType).Elem()
|
||||||
|
} else {
|
||||||
|
arg = reflect.ValueOf(argument)
|
||||||
|
}
|
||||||
|
|
||||||
|
if argType == nil && !expectTypeNilSupported {
|
||||||
|
panic(errors.New("attempting to call matcher with nil for non-nil expected type"))
|
||||||
|
}
|
||||||
|
if argType == nil || argType.AssignableTo(expectType) {
|
||||||
|
result := f.fn.Call([]reflect.Value{arg})
|
||||||
|
return result[0].Bool()
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
func (f argumentMatcher) String() string {
|
||||||
|
return fmt.Sprintf("func(%s) bool", f.fn.Type().In(0).Name())
|
||||||
|
}
|
||||||
|
|
||||||
|
// MatchedBy can be used to match a mock call based on only certain properties
|
||||||
|
// from a complex struct or some calculation. It takes a function that will be
|
||||||
|
// evaluated with the called argument and will return true when there's a match
|
||||||
|
// and false otherwise.
|
||||||
|
//
|
||||||
|
// Example:
|
||||||
|
// m.On("Do", MatchedBy(func(req *http.Request) bool { return req.Host == "example.com" }))
|
||||||
|
//
|
||||||
|
// |fn|, must be a function accepting a single argument (of the expected type)
|
||||||
|
// which returns a bool. If |fn| doesn't match the required signature,
|
||||||
|
// MatchedBy() panics.
|
||||||
|
func MatchedBy(fn interface{}) argumentMatcher {
|
||||||
|
fnType := reflect.TypeOf(fn)
|
||||||
|
|
||||||
|
if fnType.Kind() != reflect.Func {
|
||||||
|
panic(fmt.Sprintf("assert: arguments: %s is not a func", fn))
|
||||||
|
}
|
||||||
|
if fnType.NumIn() != 1 {
|
||||||
|
panic(fmt.Sprintf("assert: arguments: %s does not take exactly one argument", fn))
|
||||||
|
}
|
||||||
|
if fnType.NumOut() != 1 || fnType.Out(0).Kind() != reflect.Bool {
|
||||||
|
panic(fmt.Sprintf("assert: arguments: %s does not return a bool", fn))
|
||||||
|
}
|
||||||
|
|
||||||
|
return argumentMatcher{fn: reflect.ValueOf(fn)}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get Returns the argument at the specified index.
|
||||||
|
func (args Arguments) Get(index int) interface{} {
|
||||||
|
if index+1 > len(args) {
|
||||||
|
panic(fmt.Sprintf("assert: arguments: Cannot call Get(%d) because there are %d argument(s).", index, len(args)))
|
||||||
|
}
|
||||||
|
return args[index]
|
||||||
|
}
|
||||||
|
|
||||||
|
// Is gets whether the objects match the arguments specified.
|
||||||
|
func (args Arguments) Is(objects ...interface{}) bool {
|
||||||
|
for i, obj := range args {
|
||||||
|
if obj != objects[i] {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// Diff gets a string describing the differences between the arguments
|
||||||
|
// and the specified objects.
|
||||||
|
//
|
||||||
|
// Returns the diff string and number of differences found.
|
||||||
|
func (args Arguments) Diff(objects []interface{}) (string, int) {
|
||||||
|
//TODO: could return string as error and nil for No difference
|
||||||
|
|
||||||
|
var output = "\n"
|
||||||
|
var differences int
|
||||||
|
|
||||||
|
var maxArgCount = len(args)
|
||||||
|
if len(objects) > maxArgCount {
|
||||||
|
maxArgCount = len(objects)
|
||||||
|
}
|
||||||
|
|
||||||
|
for i := 0; i < maxArgCount; i++ {
|
||||||
|
var actual, expected interface{}
|
||||||
|
|
||||||
|
if len(objects) <= i {
|
||||||
|
actual = "(Missing)"
|
||||||
|
} else {
|
||||||
|
actual = objects[i]
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(args) <= i {
|
||||||
|
expected = "(Missing)"
|
||||||
|
} else {
|
||||||
|
expected = args[i]
|
||||||
|
}
|
||||||
|
|
||||||
|
if matcher, ok := expected.(argumentMatcher); ok {
|
||||||
|
if matcher.Matches(actual) {
|
||||||
|
output = fmt.Sprintf("%s\t%d: PASS: %s matched by %s\n", output, i, actual, matcher)
|
||||||
|
} else {
|
||||||
|
differences++
|
||||||
|
output = fmt.Sprintf("%s\t%d: PASS: %s not matched by %s\n", output, i, actual, matcher)
|
||||||
|
}
|
||||||
|
} else if reflect.TypeOf(expected) == reflect.TypeOf((*AnythingOfTypeArgument)(nil)).Elem() {
|
||||||
|
|
||||||
|
// type checking
|
||||||
|
if reflect.TypeOf(actual).Name() != string(expected.(AnythingOfTypeArgument)) && reflect.TypeOf(actual).String() != string(expected.(AnythingOfTypeArgument)) {
|
||||||
|
// not match
|
||||||
|
differences++
|
||||||
|
output = fmt.Sprintf("%s\t%d: FAIL: type %s != type %s - %s\n", output, i, expected, reflect.TypeOf(actual).Name(), actual)
|
||||||
|
}
|
||||||
|
|
||||||
|
} else {
|
||||||
|
|
||||||
|
// normal checking
|
||||||
|
|
||||||
|
if assert.ObjectsAreEqual(expected, Anything) || assert.ObjectsAreEqual(actual, Anything) || assert.ObjectsAreEqual(actual, expected) {
|
||||||
|
// match
|
||||||
|
output = fmt.Sprintf("%s\t%d: PASS: %s == %s\n", output, i, actual, expected)
|
||||||
|
} else {
|
||||||
|
// not match
|
||||||
|
differences++
|
||||||
|
output = fmt.Sprintf("%s\t%d: FAIL: %s != %s\n", output, i, actual, expected)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
if differences == 0 {
|
||||||
|
return "No differences.", differences
|
||||||
|
}
|
||||||
|
|
||||||
|
return output, differences
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// Assert compares the arguments with the specified objects and fails if
|
||||||
|
// they do not exactly match.
|
||||||
|
func (args Arguments) Assert(t TestingT, objects ...interface{}) bool {
|
||||||
|
if h, ok := t.(tHelper); ok {
|
||||||
|
h.Helper()
|
||||||
|
}
|
||||||
|
|
||||||
|
// get the differences
|
||||||
|
diff, diffCount := args.Diff(objects)
|
||||||
|
|
||||||
|
if diffCount == 0 {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
// there are differences... report them...
|
||||||
|
t.Logf(diff)
|
||||||
|
t.Errorf("%sArguments do not match.", assert.CallerInfo())
|
||||||
|
|
||||||
|
return false
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// String gets the argument at the specified index. Panics if there is no argument, or
|
||||||
|
// if the argument is of the wrong type.
|
||||||
|
//
|
||||||
|
// If no index is provided, String() returns a complete string representation
|
||||||
|
// of the arguments.
|
||||||
|
func (args Arguments) String(indexOrNil ...int) string {
|
||||||
|
|
||||||
|
if len(indexOrNil) == 0 {
|
||||||
|
// normal String() method - return a string representation of the args
|
||||||
|
var argsStr []string
|
||||||
|
for _, arg := range args {
|
||||||
|
argsStr = append(argsStr, fmt.Sprintf("%s", reflect.TypeOf(arg)))
|
||||||
|
}
|
||||||
|
return strings.Join(argsStr, ",")
|
||||||
|
} else if len(indexOrNil) == 1 {
|
||||||
|
// Index has been specified - get the argument at that index
|
||||||
|
var index = indexOrNil[0]
|
||||||
|
var s string
|
||||||
|
var ok bool
|
||||||
|
if s, ok = args.Get(index).(string); !ok {
|
||||||
|
panic(fmt.Sprintf("assert: arguments: String(%d) failed because object wasn't correct type: %s", index, args.Get(index)))
|
||||||
|
}
|
||||||
|
return s
|
||||||
|
}
|
||||||
|
|
||||||
|
panic(fmt.Sprintf("assert: arguments: Wrong number of arguments passed to String. Must be 0 or 1, not %d", len(indexOrNil)))
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
// Int gets the argument at the specified index. Panics if there is no argument, or
|
||||||
|
// if the argument is of the wrong type.
|
||||||
|
func (args Arguments) Int(index int) int {
|
||||||
|
var s int
|
||||||
|
var ok bool
|
||||||
|
if s, ok = args.Get(index).(int); !ok {
|
||||||
|
panic(fmt.Sprintf("assert: arguments: Int(%d) failed because object wasn't correct type: %v", index, args.Get(index)))
|
||||||
|
}
|
||||||
|
return s
|
||||||
|
}
|
||||||
|
|
||||||
|
// Error gets the argument at the specified index. Panics if there is no argument, or
|
||||||
|
// if the argument is of the wrong type.
|
||||||
|
func (args Arguments) Error(index int) error {
|
||||||
|
obj := args.Get(index)
|
||||||
|
var s error
|
||||||
|
var ok bool
|
||||||
|
if obj == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
if s, ok = obj.(error); !ok {
|
||||||
|
panic(fmt.Sprintf("assert: arguments: Error(%d) failed because object wasn't correct type: %v", index, args.Get(index)))
|
||||||
|
}
|
||||||
|
return s
|
||||||
|
}
|
||||||
|
|
||||||
|
// Bool gets the argument at the specified index. Panics if there is no argument, or
|
||||||
|
// if the argument is of the wrong type.
|
||||||
|
func (args Arguments) Bool(index int) bool {
|
||||||
|
var s bool
|
||||||
|
var ok bool
|
||||||
|
if s, ok = args.Get(index).(bool); !ok {
|
||||||
|
panic(fmt.Sprintf("assert: arguments: Bool(%d) failed because object wasn't correct type: %v", index, args.Get(index)))
|
||||||
|
}
|
||||||
|
return s
|
||||||
|
}
|
||||||
|
|
||||||
|
func typeAndKind(v interface{}) (reflect.Type, reflect.Kind) {
|
||||||
|
t := reflect.TypeOf(v)
|
||||||
|
k := t.Kind()
|
||||||
|
|
||||||
|
if k == reflect.Ptr {
|
||||||
|
t = t.Elem()
|
||||||
|
k = t.Kind()
|
||||||
|
}
|
||||||
|
return t, k
|
||||||
|
}
|
||||||
|
|
||||||
|
func diffArguments(expected Arguments, actual Arguments) string {
|
||||||
|
if len(expected) != len(actual) {
|
||||||
|
return fmt.Sprintf("Provided %v arguments, mocked for %v arguments", len(expected), len(actual))
|
||||||
|
}
|
||||||
|
|
||||||
|
for x := range expected {
|
||||||
|
if diffString := diff(expected[x], actual[x]); diffString != "" {
|
||||||
|
return fmt.Sprintf("Difference found in argument %v:\n\n%s", x, diffString)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
// diff returns a diff of both values as long as both are of the same type and
|
||||||
|
// are a struct, map, slice or array. Otherwise it returns an empty string.
|
||||||
|
func diff(expected interface{}, actual interface{}) string {
|
||||||
|
if expected == nil || actual == nil {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
et, ek := typeAndKind(expected)
|
||||||
|
at, _ := typeAndKind(actual)
|
||||||
|
|
||||||
|
if et != at {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
if ek != reflect.Struct && ek != reflect.Map && ek != reflect.Slice && ek != reflect.Array {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
e := spewConfig.Sdump(expected)
|
||||||
|
a := spewConfig.Sdump(actual)
|
||||||
|
|
||||||
|
diff, _ := difflib.GetUnifiedDiffString(difflib.UnifiedDiff{
|
||||||
|
A: difflib.SplitLines(e),
|
||||||
|
B: difflib.SplitLines(a),
|
||||||
|
FromFile: "Expected",
|
||||||
|
FromDate: "",
|
||||||
|
ToFile: "Actual",
|
||||||
|
ToDate: "",
|
||||||
|
Context: 1,
|
||||||
|
})
|
||||||
|
|
||||||
|
return diff
|
||||||
|
}
|
||||||
|
|
||||||
|
var spewConfig = spew.ConfigState{
|
||||||
|
Indent: " ",
|
||||||
|
DisablePointerAddresses: true,
|
||||||
|
DisableCapacities: true,
|
||||||
|
SortKeys: true,
|
||||||
|
}
|
||||||
|
|
||||||
|
type tHelper interface {
|
||||||
|
Helper()
|
||||||
|
}
|
|
@ -80,6 +80,7 @@
|
||||||
{"path":"github.com/mitchellh/copystructure","checksumSHA1":"86nE93o1VIND0Doe8PuhCXnhUx0=","revision":"cdac8253d00f2ecf0a0b19fbff173a9a72de4f82","revisionTime":"2016-08-04T03:23:30Z"},
|
{"path":"github.com/mitchellh/copystructure","checksumSHA1":"86nE93o1VIND0Doe8PuhCXnhUx0=","revision":"cdac8253d00f2ecf0a0b19fbff173a9a72de4f82","revisionTime":"2016-08-04T03:23:30Z"},
|
||||||
{"path":"github.com/mitchellh/go-homedir","checksumSHA1":"V/quM7+em2ByJbWBLOsEwnY3j/Q=","revision":"b8bc1bf767474819792c23f32d8286a45736f1c6","revisionTime":"2016-12-03T19:45:07Z"},
|
{"path":"github.com/mitchellh/go-homedir","checksumSHA1":"V/quM7+em2ByJbWBLOsEwnY3j/Q=","revision":"b8bc1bf767474819792c23f32d8286a45736f1c6","revisionTime":"2016-12-03T19:45:07Z"},
|
||||||
{"path":"github.com/mitchellh/go-testing-interface","checksumSHA1":"bDdhmDk8q6utWrccBhEOa6IoGkE=","revision":"a61a99592b77c9ba629d254a693acffaeb4b7e28","revisionTime":"2017-10-04T22:19:16Z"},
|
{"path":"github.com/mitchellh/go-testing-interface","checksumSHA1":"bDdhmDk8q6utWrccBhEOa6IoGkE=","revision":"a61a99592b77c9ba629d254a693acffaeb4b7e28","revisionTime":"2017-10-04T22:19:16Z"},
|
||||||
|
{"path":"github.com/mitchellh/hashstructure","checksumSHA1":"tWUjKyFOGJtYExocPWVYiXBYsfE=","revision":"2bca23e0e452137f789efbc8610126fd8b94f73b","revisionTime":"2017-06-09T04:59:27Z"},
|
||||||
{"path":"github.com/mitchellh/mapstructure","checksumSHA1":"gILp4IL+xwXLH6tJtRLrnZ56F24=","revision":"06020f85339e21b2478f756a78e295255ffa4d6a","revisionTime":"2017-10-17T17:18:08Z"},
|
{"path":"github.com/mitchellh/mapstructure","checksumSHA1":"gILp4IL+xwXLH6tJtRLrnZ56F24=","revision":"06020f85339e21b2478f756a78e295255ffa4d6a","revisionTime":"2017-10-17T17:18:08Z"},
|
||||||
{"path":"github.com/mitchellh/reflectwalk","checksumSHA1":"mrqMlK6gqe//WsJSrJ1HgkPM0lM=","revision":"eecf4c70c626c7cfbb95c90195bc34d386c74ac6","revisionTime":"2015-05-27T15:31:53Z"},
|
{"path":"github.com/mitchellh/reflectwalk","checksumSHA1":"mrqMlK6gqe//WsJSrJ1HgkPM0lM=","revision":"eecf4c70c626c7cfbb95c90195bc34d386c74ac6","revisionTime":"2015-05-27T15:31:53Z"},
|
||||||
{"path":"github.com/pascaldekloe/goe/verify","checksumSHA1":"5h+ERzHw3Rl2G0kFPxoJzxiA9s0=","revision":"07ebd1e2481f616a278ab431cf04cc5cf5ab3ebe","revisionTime":"2017-03-28T18:37:59Z"},
|
{"path":"github.com/pascaldekloe/goe/verify","checksumSHA1":"5h+ERzHw3Rl2G0kFPxoJzxiA9s0=","revision":"07ebd1e2481f616a278ab431cf04cc5cf5ab3ebe","revisionTime":"2017-03-28T18:37:59Z"},
|
||||||
|
@ -97,6 +98,7 @@
|
||||||
{"path":"github.com/shirou/gopsutil/process","checksumSHA1":"JX0bRK/BdKVfbm4XOxMducVdY58=","revision":"32b6636de04b303274daac3ca2b10d3b0e4afc35","revisionTime":"2017-02-04T05:36:48Z"},
|
{"path":"github.com/shirou/gopsutil/process","checksumSHA1":"JX0bRK/BdKVfbm4XOxMducVdY58=","revision":"32b6636de04b303274daac3ca2b10d3b0e4afc35","revisionTime":"2017-02-04T05:36:48Z"},
|
||||||
{"path":"github.com/shirou/w32","checksumSHA1":"Nve7SpDmjsv6+rhkXAkfg/UQx94=","revision":"bb4de0191aa41b5507caa14b0650cdbddcd9280b","revisionTime":"2016-09-30T03:27:40Z"},
|
{"path":"github.com/shirou/w32","checksumSHA1":"Nve7SpDmjsv6+rhkXAkfg/UQx94=","revision":"bb4de0191aa41b5507caa14b0650cdbddcd9280b","revisionTime":"2016-09-30T03:27:40Z"},
|
||||||
{"path":"github.com/stretchr/testify/assert","checksumSHA1":"6LwXZI7kXm1C0h4Ui0Y52p9uQhk=","revision":"c679ae2cc0cb27ec3293fea7e254e47386f05d69","revisionTime":"2018-03-14T08:05:35Z"},
|
{"path":"github.com/stretchr/testify/assert","checksumSHA1":"6LwXZI7kXm1C0h4Ui0Y52p9uQhk=","revision":"c679ae2cc0cb27ec3293fea7e254e47386f05d69","revisionTime":"2018-03-14T08:05:35Z"},
|
||||||
|
{"path":"github.com/stretchr/testify/mock","checksumSHA1":"Qloi2PTvZv+D9FDHXM/banCoaFY=","revision":"c679ae2cc0cb27ec3293fea7e254e47386f05d69","revisionTime":"2018-03-14T08:05:35Z"},
|
||||||
{"path":"github.com/stretchr/testify/require","checksumSHA1":"KqYmXUcuGwsvBL6XVsQnXsFb3LI=","revision":"c679ae2cc0cb27ec3293fea7e254e47386f05d69","revisionTime":"2018-03-14T08:05:35Z"},
|
{"path":"github.com/stretchr/testify/require","checksumSHA1":"KqYmXUcuGwsvBL6XVsQnXsFb3LI=","revision":"c679ae2cc0cb27ec3293fea7e254e47386f05d69","revisionTime":"2018-03-14T08:05:35Z"},
|
||||||
{"path":"github.com/tonnerre/golang-text","checksumSHA1":"t24KnvC9jRxiANVhpw2pqFpmEu8=","revision":"048ed3d792f7104850acbc8cfc01e5a6070f4c04","revisionTime":"2013-09-25T19:58:46Z"},
|
{"path":"github.com/tonnerre/golang-text","checksumSHA1":"t24KnvC9jRxiANVhpw2pqFpmEu8=","revision":"048ed3d792f7104850acbc8cfc01e5a6070f4c04","revisionTime":"2013-09-25T19:58:46Z"},
|
||||||
{"path":"golang.org/x/net/context","checksumSHA1":"9jjO5GjLa0XF/nfWihF02RoH4qc=","revision":"075e191f18186a8ff2becaf64478e30f4545cdad","revisionTime":"2016-08-05T06:12:51Z"},
|
{"path":"golang.org/x/net/context","checksumSHA1":"9jjO5GjLa0XF/nfWihF02RoH4qc=","revision":"075e191f18186a8ff2becaf64478e30f4545cdad","revisionTime":"2016-08-05T06:12:51Z"},
|
||||||
|
|
Loading…
Reference in New Issue