mirror of
https://github.com/woodpecker-ci/woodpecker.git
synced 2024-12-11 19:16:32 +00:00
795 lines
22 KiB
Go
795 lines
22 KiB
Go
|
package mock
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"github.com/drone/drone/Godeps/_workspace/src/github.com/stretchr/testify/assert"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
/*
|
||
|
|
Test objects
|
||
|
|
*/
|
||
|
|
|
||
|
|
// ExampleInterface represents an example interface.
|
||
|
|
type ExampleInterface interface {
|
||
|
|
TheExampleMethod(a, b, c int) (int, error)
|
||
|
|
}
|
||
|
|
|
||
|
|
// TestExampleImplementation is a test implementation of ExampleInterface
|
||
|
|
type TestExampleImplementation struct {
|
||
|
|
Mock
|
||
|
|
}
|
||
|
|
|
||
|
|
func (i *TestExampleImplementation) TheExampleMethod(a, b, c int) (int, error) {
|
||
|
|
args := i.Called(a, b, c)
|
||
|
|
return args.Int(0), errors.New("Whoops")
|
||
|
|
}
|
||
|
|
|
||
|
|
func (i *TestExampleImplementation) TheExampleMethod2(yesorno bool) {
|
||
|
|
i.Called(yesorno)
|
||
|
|
}
|
||
|
|
|
||
|
|
type ExampleType struct {
|
||
|
|
ran bool
|
||
|
|
}
|
||
|
|
|
||
|
|
func (i *TestExampleImplementation) TheExampleMethod3(et *ExampleType) error {
|
||
|
|
args := i.Called(et)
|
||
|
|
return args.Error(0)
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
Mock
|
||
|
|
*/
|
||
|
|
|
||
|
|
func Test_Mock_TestData(t *testing.T) {
|
||
|
|
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
if assert.NotNil(t, mockedService.TestData()) {
|
||
|
|
|
||
|
|
mockedService.TestData().Set("something", 123)
|
||
|
|
assert.Equal(t, 123, mockedService.TestData().Get("something").Data())
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_On(t *testing.T) {
|
||
|
|
|
||
|
|
// make a test impl object
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
assert.Equal(t, mockedService.On("TheExampleMethod"), &mockedService.Mock)
|
||
|
|
assert.Equal(t, "TheExampleMethod", mockedService.onMethodName)
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_On_WithArgs(t *testing.T) {
|
||
|
|
|
||
|
|
// make a test impl object
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
assert.Equal(t, mockedService.On("TheExampleMethod", 1, 2, 3), &mockedService.Mock)
|
||
|
|
assert.Equal(t, "TheExampleMethod", mockedService.onMethodName)
|
||
|
|
assert.Equal(t, 1, mockedService.onMethodArguments[0])
|
||
|
|
assert.Equal(t, 2, mockedService.onMethodArguments[1])
|
||
|
|
assert.Equal(t, 3, mockedService.onMethodArguments[2])
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_Return(t *testing.T) {
|
||
|
|
|
||
|
|
// make a test impl object
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
assert.Equal(t, mockedService.On("TheExampleMethod", "A", "B", true).Return(1, "two", true), &mockedService.Mock)
|
||
|
|
|
||
|
|
// ensure the call was created
|
||
|
|
if assert.Equal(t, 1, len(mockedService.ExpectedCalls)) {
|
||
|
|
call := mockedService.ExpectedCalls[0]
|
||
|
|
|
||
|
|
assert.Equal(t, "TheExampleMethod", call.Method)
|
||
|
|
assert.Equal(t, "A", call.Arguments[0])
|
||
|
|
assert.Equal(t, "B", call.Arguments[1])
|
||
|
|
assert.Equal(t, true, call.Arguments[2])
|
||
|
|
assert.Equal(t, 1, call.ReturnArguments[0])
|
||
|
|
assert.Equal(t, "two", call.ReturnArguments[1])
|
||
|
|
assert.Equal(t, true, call.ReturnArguments[2])
|
||
|
|
assert.Equal(t, 0, call.Repeatability)
|
||
|
|
assert.Nil(t, call.WaitFor)
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_Return_WaitUntil(t *testing.T) {
|
||
|
|
|
||
|
|
// make a test impl object
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
ch := time.After(time.Second)
|
||
|
|
|
||
|
|
assert.Equal(t, mockedService.Mock.On("TheExampleMethod", "A", "B", true).Return(1, "two", true).WaitUntil(ch), &mockedService.Mock)
|
||
|
|
|
||
|
|
// ensure the call was created
|
||
|
|
if assert.Equal(t, 1, len(mockedService.Mock.ExpectedCalls)) {
|
||
|
|
call := mockedService.Mock.ExpectedCalls[0]
|
||
|
|
|
||
|
|
assert.Equal(t, "TheExampleMethod", call.Method)
|
||
|
|
assert.Equal(t, "A", call.Arguments[0])
|
||
|
|
assert.Equal(t, "B", call.Arguments[1])
|
||
|
|
assert.Equal(t, true, call.Arguments[2])
|
||
|
|
assert.Equal(t, 1, call.ReturnArguments[0])
|
||
|
|
assert.Equal(t, "two", call.ReturnArguments[1])
|
||
|
|
assert.Equal(t, true, call.ReturnArguments[2])
|
||
|
|
assert.Equal(t, 0, call.Repeatability)
|
||
|
|
assert.Equal(t, ch, call.WaitFor)
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_Return_After(t *testing.T) {
|
||
|
|
|
||
|
|
// make a test impl object
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
assert.Equal(t, mockedService.Mock.On("TheExampleMethod", "A", "B", true).Return(1, "two", true).After(time.Second), &mockedService.Mock)
|
||
|
|
|
||
|
|
// ensure the call was created
|
||
|
|
if assert.Equal(t, 1, len(mockedService.Mock.ExpectedCalls)) {
|
||
|
|
call := mockedService.Mock.ExpectedCalls[0]
|
||
|
|
|
||
|
|
assert.Equal(t, "TheExampleMethod", call.Method)
|
||
|
|
assert.Equal(t, "A", call.Arguments[0])
|
||
|
|
assert.Equal(t, "B", call.Arguments[1])
|
||
|
|
assert.Equal(t, true, call.Arguments[2])
|
||
|
|
assert.Equal(t, 1, call.ReturnArguments[0])
|
||
|
|
assert.Equal(t, "two", call.ReturnArguments[1])
|
||
|
|
assert.Equal(t, true, call.ReturnArguments[2])
|
||
|
|
assert.Equal(t, 0, call.Repeatability)
|
||
|
|
assert.NotEqual(t, nil, call.WaitFor)
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_Return_Run(t *testing.T) {
|
||
|
|
|
||
|
|
// make a test impl object
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
assert.Equal(t, mockedService.Mock.On("TheExampleMethod3", AnythingOfType("*mock.ExampleType")).Return(nil).Run(func(args Arguments) {
|
||
|
|
arg := args.Get(0).(*ExampleType)
|
||
|
|
arg.ran = true
|
||
|
|
}), &mockedService.Mock)
|
||
|
|
|
||
|
|
// ensure the call was created
|
||
|
|
if assert.Equal(t, 1, len(mockedService.Mock.ExpectedCalls)) {
|
||
|
|
call := mockedService.Mock.ExpectedCalls[0]
|
||
|
|
|
||
|
|
assert.Equal(t, "TheExampleMethod3", call.Method)
|
||
|
|
assert.Equal(t, AnythingOfType("*mock.ExampleType"), call.Arguments[0])
|
||
|
|
assert.Equal(t, nil, call.ReturnArguments[0])
|
||
|
|
assert.Equal(t, 0, call.Repeatability)
|
||
|
|
assert.NotEqual(t, nil, call.WaitFor)
|
||
|
|
assert.NotNil(t, call.Run)
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
et := ExampleType{}
|
||
|
|
assert.Equal(t, false, et.ran)
|
||
|
|
mockedService.TheExampleMethod3(&et)
|
||
|
|
assert.Equal(t, true, et.ran)
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_Return_Once(t *testing.T) {
|
||
|
|
|
||
|
|
// make a test impl object
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
mockedService.On("TheExampleMethod", "A", "B", true).Return(1, "two", true).Once()
|
||
|
|
|
||
|
|
// ensure the call was created
|
||
|
|
if assert.Equal(t, 1, len(mockedService.ExpectedCalls)) {
|
||
|
|
call := mockedService.ExpectedCalls[0]
|
||
|
|
|
||
|
|
assert.Equal(t, "TheExampleMethod", call.Method)
|
||
|
|
assert.Equal(t, "A", call.Arguments[0])
|
||
|
|
assert.Equal(t, "B", call.Arguments[1])
|
||
|
|
assert.Equal(t, true, call.Arguments[2])
|
||
|
|
assert.Equal(t, 1, call.ReturnArguments[0])
|
||
|
|
assert.Equal(t, "two", call.ReturnArguments[1])
|
||
|
|
assert.Equal(t, true, call.ReturnArguments[2])
|
||
|
|
assert.Equal(t, 1, call.Repeatability)
|
||
|
|
assert.Nil(t, call.WaitFor)
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_Return_Twice(t *testing.T) {
|
||
|
|
|
||
|
|
// make a test impl object
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
mockedService.On("TheExampleMethod", "A", "B", true).Return(1, "two", true).Twice()
|
||
|
|
|
||
|
|
// ensure the call was created
|
||
|
|
if assert.Equal(t, 1, len(mockedService.ExpectedCalls)) {
|
||
|
|
call := mockedService.ExpectedCalls[0]
|
||
|
|
|
||
|
|
assert.Equal(t, "TheExampleMethod", call.Method)
|
||
|
|
assert.Equal(t, "A", call.Arguments[0])
|
||
|
|
assert.Equal(t, "B", call.Arguments[1])
|
||
|
|
assert.Equal(t, true, call.Arguments[2])
|
||
|
|
assert.Equal(t, 1, call.ReturnArguments[0])
|
||
|
|
assert.Equal(t, "two", call.ReturnArguments[1])
|
||
|
|
assert.Equal(t, true, call.ReturnArguments[2])
|
||
|
|
assert.Equal(t, 2, call.Repeatability)
|
||
|
|
assert.Nil(t, call.WaitFor)
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_Return_Times(t *testing.T) {
|
||
|
|
|
||
|
|
// make a test impl object
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
mockedService.On("TheExampleMethod", "A", "B", true).Return(1, "two", true).Times(5)
|
||
|
|
|
||
|
|
// ensure the call was created
|
||
|
|
if assert.Equal(t, 1, len(mockedService.ExpectedCalls)) {
|
||
|
|
call := mockedService.ExpectedCalls[0]
|
||
|
|
|
||
|
|
assert.Equal(t, "TheExampleMethod", call.Method)
|
||
|
|
assert.Equal(t, "A", call.Arguments[0])
|
||
|
|
assert.Equal(t, "B", call.Arguments[1])
|
||
|
|
assert.Equal(t, true, call.Arguments[2])
|
||
|
|
assert.Equal(t, 1, call.ReturnArguments[0])
|
||
|
|
assert.Equal(t, "two", call.ReturnArguments[1])
|
||
|
|
assert.Equal(t, true, call.ReturnArguments[2])
|
||
|
|
assert.Equal(t, 5, call.Repeatability)
|
||
|
|
assert.Nil(t, call.WaitFor)
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_Return_Nothing(t *testing.T) {
|
||
|
|
|
||
|
|
// make a test impl object
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
assert.Equal(t, mockedService.On("TheExampleMethod", "A", "B", true).Return(), &mockedService.Mock)
|
||
|
|
|
||
|
|
// ensure the call was created
|
||
|
|
if assert.Equal(t, 1, len(mockedService.ExpectedCalls)) {
|
||
|
|
call := mockedService.ExpectedCalls[0]
|
||
|
|
|
||
|
|
assert.Equal(t, "TheExampleMethod", call.Method)
|
||
|
|
assert.Equal(t, "A", call.Arguments[0])
|
||
|
|
assert.Equal(t, "B", call.Arguments[1])
|
||
|
|
assert.Equal(t, true, call.Arguments[2])
|
||
|
|
assert.Equal(t, 0, len(call.ReturnArguments))
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_findExpectedCall(t *testing.T) {
|
||
|
|
|
||
|
|
m := new(Mock)
|
||
|
|
m.On("One", 1).Return("one")
|
||
|
|
m.On("Two", 2).Return("two")
|
||
|
|
m.On("Two", 3).Return("three")
|
||
|
|
|
||
|
|
f, c := m.findExpectedCall("Two", 3)
|
||
|
|
|
||
|
|
if assert.Equal(t, 2, f) {
|
||
|
|
if assert.NotNil(t, c) {
|
||
|
|
assert.Equal(t, "Two", c.Method)
|
||
|
|
assert.Equal(t, 3, c.Arguments[0])
|
||
|
|
assert.Equal(t, "three", c.ReturnArguments[0])
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_findExpectedCall_For_Unknown_Method(t *testing.T) {
|
||
|
|
|
||
|
|
m := new(Mock)
|
||
|
|
m.On("One", 1).Return("one")
|
||
|
|
m.On("Two", 2).Return("two")
|
||
|
|
m.On("Two", 3).Return("three")
|
||
|
|
|
||
|
|
f, _ := m.findExpectedCall("Two")
|
||
|
|
|
||
|
|
assert.Equal(t, -1, f)
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_findExpectedCall_Respects_Repeatability(t *testing.T) {
|
||
|
|
|
||
|
|
m := new(Mock)
|
||
|
|
m.On("One", 1).Return("one")
|
||
|
|
m.On("Two", 2).Return("two").Once()
|
||
|
|
m.On("Two", 3).Return("three").Twice()
|
||
|
|
m.On("Two", 3).Return("three").Times(8)
|
||
|
|
|
||
|
|
f, c := m.findExpectedCall("Two", 3)
|
||
|
|
|
||
|
|
if assert.Equal(t, 2, f) {
|
||
|
|
if assert.NotNil(t, c) {
|
||
|
|
assert.Equal(t, "Two", c.Method)
|
||
|
|
assert.Equal(t, 3, c.Arguments[0])
|
||
|
|
assert.Equal(t, "three", c.ReturnArguments[0])
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_callString(t *testing.T) {
|
||
|
|
|
||
|
|
assert.Equal(t, `Method(int,bool,string)`, callString("Method", []interface{}{1, true, "something"}, false))
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_Called(t *testing.T) {
|
||
|
|
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
mockedService.On("Test_Mock_Called", 1, 2, 3).Return(5, "6", true)
|
||
|
|
|
||
|
|
returnArguments := mockedService.Called(1, 2, 3)
|
||
|
|
|
||
|
|
if assert.Equal(t, 1, len(mockedService.Calls)) {
|
||
|
|
assert.Equal(t, "Test_Mock_Called", mockedService.Calls[0].Method)
|
||
|
|
assert.Equal(t, 1, mockedService.Calls[0].Arguments[0])
|
||
|
|
assert.Equal(t, 2, mockedService.Calls[0].Arguments[1])
|
||
|
|
assert.Equal(t, 3, mockedService.Calls[0].Arguments[2])
|
||
|
|
}
|
||
|
|
|
||
|
|
if assert.Equal(t, 3, len(returnArguments)) {
|
||
|
|
assert.Equal(t, 5, returnArguments[0])
|
||
|
|
assert.Equal(t, "6", returnArguments[1])
|
||
|
|
assert.Equal(t, true, returnArguments[2])
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func asyncCall(m *Mock, ch chan Arguments) {
|
||
|
|
ch <- m.Called(1, 2, 3)
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_Called_blocks(t *testing.T) {
|
||
|
|
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
mockedService.Mock.On("asyncCall", 1, 2, 3).Return(5, "6", true).After(2 * time.Millisecond)
|
||
|
|
|
||
|
|
ch := make(chan Arguments)
|
||
|
|
|
||
|
|
go asyncCall(&mockedService.Mock, ch)
|
||
|
|
|
||
|
|
select {
|
||
|
|
case <-ch:
|
||
|
|
t.Fatal("should have waited")
|
||
|
|
case <-time.After(1 * time.Millisecond):
|
||
|
|
}
|
||
|
|
|
||
|
|
returnArguments := <-ch
|
||
|
|
|
||
|
|
if assert.Equal(t, 1, len(mockedService.Mock.Calls)) {
|
||
|
|
assert.Equal(t, "asyncCall", mockedService.Mock.Calls[0].Method)
|
||
|
|
assert.Equal(t, 1, mockedService.Mock.Calls[0].Arguments[0])
|
||
|
|
assert.Equal(t, 2, mockedService.Mock.Calls[0].Arguments[1])
|
||
|
|
assert.Equal(t, 3, mockedService.Mock.Calls[0].Arguments[2])
|
||
|
|
}
|
||
|
|
|
||
|
|
if assert.Equal(t, 3, len(returnArguments)) {
|
||
|
|
assert.Equal(t, 5, returnArguments[0])
|
||
|
|
assert.Equal(t, "6", returnArguments[1])
|
||
|
|
assert.Equal(t, true, returnArguments[2])
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_Called_For_Bounded_Repeatability(t *testing.T) {
|
||
|
|
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
mockedService.On("Test_Mock_Called_For_Bounded_Repeatability", 1, 2, 3).Return(5, "6", true).Once()
|
||
|
|
mockedService.On("Test_Mock_Called_For_Bounded_Repeatability", 1, 2, 3).Return(-1, "hi", false)
|
||
|
|
|
||
|
|
returnArguments1 := mockedService.Called(1, 2, 3)
|
||
|
|
returnArguments2 := mockedService.Called(1, 2, 3)
|
||
|
|
|
||
|
|
if assert.Equal(t, 2, len(mockedService.Calls)) {
|
||
|
|
assert.Equal(t, "Test_Mock_Called_For_Bounded_Repeatability", mockedService.Calls[0].Method)
|
||
|
|
assert.Equal(t, 1, mockedService.Calls[0].Arguments[0])
|
||
|
|
assert.Equal(t, 2, mockedService.Calls[0].Arguments[1])
|
||
|
|
assert.Equal(t, 3, mockedService.Calls[0].Arguments[2])
|
||
|
|
|
||
|
|
assert.Equal(t, "Test_Mock_Called_For_Bounded_Repeatability", mockedService.Calls[1].Method)
|
||
|
|
assert.Equal(t, 1, mockedService.Calls[1].Arguments[0])
|
||
|
|
assert.Equal(t, 2, mockedService.Calls[1].Arguments[1])
|
||
|
|
assert.Equal(t, 3, mockedService.Calls[1].Arguments[2])
|
||
|
|
}
|
||
|
|
|
||
|
|
if assert.Equal(t, 3, len(returnArguments1)) {
|
||
|
|
assert.Equal(t, 5, returnArguments1[0])
|
||
|
|
assert.Equal(t, "6", returnArguments1[1])
|
||
|
|
assert.Equal(t, true, returnArguments1[2])
|
||
|
|
}
|
||
|
|
|
||
|
|
if assert.Equal(t, 3, len(returnArguments2)) {
|
||
|
|
assert.Equal(t, -1, returnArguments2[0])
|
||
|
|
assert.Equal(t, "hi", returnArguments2[1])
|
||
|
|
assert.Equal(t, false, returnArguments2[2])
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_Called_For_SetTime_Expectation(t *testing.T) {
|
||
|
|
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
mockedService.On("TheExampleMethod", 1, 2, 3).Return(5, "6", true).Times(4)
|
||
|
|
|
||
|
|
mockedService.TheExampleMethod(1, 2, 3)
|
||
|
|
mockedService.TheExampleMethod(1, 2, 3)
|
||
|
|
mockedService.TheExampleMethod(1, 2, 3)
|
||
|
|
mockedService.TheExampleMethod(1, 2, 3)
|
||
|
|
assert.Panics(t, func() {
|
||
|
|
mockedService.TheExampleMethod(1, 2, 3)
|
||
|
|
})
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_Called_Unexpected(t *testing.T) {
|
||
|
|
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
// make sure it panics if no expectation was made
|
||
|
|
assert.Panics(t, func() {
|
||
|
|
mockedService.Called(1, 2, 3)
|
||
|
|
}, "Calling unexpected method should panic")
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_AssertExpectationsForObjects_Helper(t *testing.T) {
|
||
|
|
|
||
|
|
var mockedService1 *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
var mockedService2 *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
var mockedService3 *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
mockedService1.On("Test_AssertExpectationsForObjects_Helper", 1).Return()
|
||
|
|
mockedService2.On("Test_AssertExpectationsForObjects_Helper", 2).Return()
|
||
|
|
mockedService3.On("Test_AssertExpectationsForObjects_Helper", 3).Return()
|
||
|
|
|
||
|
|
mockedService1.Called(1)
|
||
|
|
mockedService2.Called(2)
|
||
|
|
mockedService3.Called(3)
|
||
|
|
|
||
|
|
assert.True(t, AssertExpectationsForObjects(t, mockedService1.Mock, mockedService2.Mock, mockedService3.Mock))
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_AssertExpectationsForObjects_Helper_Failed(t *testing.T) {
|
||
|
|
|
||
|
|
var mockedService1 *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
var mockedService2 *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
var mockedService3 *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
mockedService1.On("Test_AssertExpectationsForObjects_Helper_Failed", 1).Return()
|
||
|
|
mockedService2.On("Test_AssertExpectationsForObjects_Helper_Failed", 2).Return()
|
||
|
|
mockedService3.On("Test_AssertExpectationsForObjects_Helper_Failed", 3).Return()
|
||
|
|
|
||
|
|
mockedService1.Called(1)
|
||
|
|
mockedService3.Called(3)
|
||
|
|
|
||
|
|
tt := new(testing.T)
|
||
|
|
assert.False(t, AssertExpectationsForObjects(tt, mockedService1.Mock, mockedService2.Mock, mockedService3.Mock))
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_AssertExpectations(t *testing.T) {
|
||
|
|
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
mockedService.On("Test_Mock_AssertExpectations", 1, 2, 3).Return(5, 6, 7)
|
||
|
|
|
||
|
|
tt := new(testing.T)
|
||
|
|
assert.False(t, mockedService.AssertExpectations(tt))
|
||
|
|
|
||
|
|
// make the call now
|
||
|
|
mockedService.Called(1, 2, 3)
|
||
|
|
|
||
|
|
// now assert expectations
|
||
|
|
assert.True(t, mockedService.AssertExpectations(tt))
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_AssertExpectationsCustomType(t *testing.T) {
|
||
|
|
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
mockedService.On("TheExampleMethod3", AnythingOfType("*mock.ExampleType")).Return(nil).Once()
|
||
|
|
|
||
|
|
tt := new(testing.T)
|
||
|
|
assert.False(t, mockedService.AssertExpectations(tt))
|
||
|
|
|
||
|
|
// make the call now
|
||
|
|
mockedService.TheExampleMethod3(&ExampleType{})
|
||
|
|
|
||
|
|
// now assert expectations
|
||
|
|
assert.True(t, mockedService.AssertExpectations(tt))
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_AssertExpectations_With_Repeatability(t *testing.T) {
|
||
|
|
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
mockedService.On("Test_Mock_AssertExpectations_With_Repeatability", 1, 2, 3).Return(5, 6, 7).Twice()
|
||
|
|
|
||
|
|
tt := new(testing.T)
|
||
|
|
assert.False(t, mockedService.AssertExpectations(tt))
|
||
|
|
|
||
|
|
// make the call now
|
||
|
|
mockedService.Called(1, 2, 3)
|
||
|
|
|
||
|
|
assert.False(t, mockedService.AssertExpectations(tt))
|
||
|
|
|
||
|
|
mockedService.Called(1, 2, 3)
|
||
|
|
|
||
|
|
// now assert expectations
|
||
|
|
assert.True(t, mockedService.AssertExpectations(tt))
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_TwoCallsWithDifferentArguments(t *testing.T) {
|
||
|
|
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
mockedService.On("Test_Mock_TwoCallsWithDifferentArguments", 1, 2, 3).Return(5, 6, 7)
|
||
|
|
mockedService.On("Test_Mock_TwoCallsWithDifferentArguments", 4, 5, 6).Return(5, 6, 7)
|
||
|
|
|
||
|
|
args1 := mockedService.Called(1, 2, 3)
|
||
|
|
assert.Equal(t, 5, args1.Int(0))
|
||
|
|
assert.Equal(t, 6, args1.Int(1))
|
||
|
|
assert.Equal(t, 7, args1.Int(2))
|
||
|
|
|
||
|
|
args2 := mockedService.Called(4, 5, 6)
|
||
|
|
assert.Equal(t, 5, args2.Int(0))
|
||
|
|
assert.Equal(t, 6, args2.Int(1))
|
||
|
|
assert.Equal(t, 7, args2.Int(2))
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_AssertNumberOfCalls(t *testing.T) {
|
||
|
|
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
mockedService.On("Test_Mock_AssertNumberOfCalls", 1, 2, 3).Return(5, 6, 7)
|
||
|
|
|
||
|
|
mockedService.Called(1, 2, 3)
|
||
|
|
assert.True(t, mockedService.AssertNumberOfCalls(t, "Test_Mock_AssertNumberOfCalls", 1))
|
||
|
|
|
||
|
|
mockedService.Called(1, 2, 3)
|
||
|
|
assert.True(t, mockedService.AssertNumberOfCalls(t, "Test_Mock_AssertNumberOfCalls", 2))
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_AssertCalled(t *testing.T) {
|
||
|
|
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
mockedService.On("Test_Mock_AssertCalled", 1, 2, 3).Return(5, 6, 7)
|
||
|
|
|
||
|
|
mockedService.Called(1, 2, 3)
|
||
|
|
|
||
|
|
assert.True(t, mockedService.AssertCalled(t, "Test_Mock_AssertCalled", 1, 2, 3))
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_AssertCalled_WithAnythingOfTypeArgument(t *testing.T) {
|
||
|
|
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
mockedService.On("Test_Mock_AssertCalled_WithAnythingOfTypeArgument", Anything, Anything, Anything).Return()
|
||
|
|
|
||
|
|
mockedService.Called(1, "two", []uint8("three"))
|
||
|
|
|
||
|
|
assert.True(t, mockedService.AssertCalled(t, "Test_Mock_AssertCalled_WithAnythingOfTypeArgument", AnythingOfType("int"), AnythingOfType("string"), AnythingOfType("[]uint8")))
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_AssertCalled_WithArguments(t *testing.T) {
|
||
|
|
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
mockedService.On("Test_Mock_AssertCalled_WithArguments", 1, 2, 3).Return(5, 6, 7)
|
||
|
|
|
||
|
|
mockedService.Called(1, 2, 3)
|
||
|
|
|
||
|
|
tt := new(testing.T)
|
||
|
|
assert.True(t, mockedService.AssertCalled(tt, "Test_Mock_AssertCalled_WithArguments", 1, 2, 3))
|
||
|
|
assert.False(t, mockedService.AssertCalled(tt, "Test_Mock_AssertCalled_WithArguments", 2, 3, 4))
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_AssertCalled_WithArguments_With_Repeatability(t *testing.T) {
|
||
|
|
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
mockedService.On("Test_Mock_AssertCalled_WithArguments_With_Repeatability", 1, 2, 3).Return(5, 6, 7).Once()
|
||
|
|
mockedService.On("Test_Mock_AssertCalled_WithArguments_With_Repeatability", 2, 3, 4).Return(5, 6, 7).Once()
|
||
|
|
|
||
|
|
mockedService.Called(1, 2, 3)
|
||
|
|
mockedService.Called(2, 3, 4)
|
||
|
|
|
||
|
|
tt := new(testing.T)
|
||
|
|
assert.True(t, mockedService.AssertCalled(tt, "Test_Mock_AssertCalled_WithArguments_With_Repeatability", 1, 2, 3))
|
||
|
|
assert.True(t, mockedService.AssertCalled(tt, "Test_Mock_AssertCalled_WithArguments_With_Repeatability", 2, 3, 4))
|
||
|
|
assert.False(t, mockedService.AssertCalled(tt, "Test_Mock_AssertCalled_WithArguments_With_Repeatability", 3, 4, 5))
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Mock_AssertNotCalled(t *testing.T) {
|
||
|
|
|
||
|
|
var mockedService *TestExampleImplementation = new(TestExampleImplementation)
|
||
|
|
|
||
|
|
mockedService.On("Test_Mock_AssertNotCalled", 1, 2, 3).Return(5, 6, 7)
|
||
|
|
|
||
|
|
mockedService.Called(1, 2, 3)
|
||
|
|
|
||
|
|
assert.True(t, mockedService.AssertNotCalled(t, "Test_Mock_NotCalled"))
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
/*
|
||
|
|
Arguments helper methods
|
||
|
|
*/
|
||
|
|
func Test_Arguments_Get(t *testing.T) {
|
||
|
|
|
||
|
|
var args Arguments = []interface{}{"string", 123, true}
|
||
|
|
|
||
|
|
assert.Equal(t, "string", args.Get(0).(string))
|
||
|
|
assert.Equal(t, 123, args.Get(1).(int))
|
||
|
|
assert.Equal(t, true, args.Get(2).(bool))
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Arguments_Is(t *testing.T) {
|
||
|
|
|
||
|
|
var args Arguments = []interface{}{"string", 123, true}
|
||
|
|
|
||
|
|
assert.True(t, args.Is("string", 123, true))
|
||
|
|
assert.False(t, args.Is("wrong", 456, false))
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Arguments_Diff(t *testing.T) {
|
||
|
|
|
||
|
|
var args Arguments = []interface{}{"Hello World", 123, true}
|
||
|
|
var diff string
|
||
|
|
var count int
|
||
|
|
diff, count = args.Diff([]interface{}{"Hello World", 456, "false"})
|
||
|
|
|
||
|
|
assert.Equal(t, 2, count)
|
||
|
|
assert.Contains(t, diff, `%!s(int=456) != %!s(int=123)`)
|
||
|
|
assert.Contains(t, diff, `false != %!s(bool=true)`)
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Arguments_Diff_DifferentNumberOfArgs(t *testing.T) {
|
||
|
|
|
||
|
|
var args Arguments = []interface{}{"string", 123, true}
|
||
|
|
var diff string
|
||
|
|
var count int
|
||
|
|
diff, count = args.Diff([]interface{}{"string", 456, "false", "extra"})
|
||
|
|
|
||
|
|
assert.Equal(t, 3, count)
|
||
|
|
assert.Contains(t, diff, `extra != (Missing)`)
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Arguments_Diff_WithAnythingArgument(t *testing.T) {
|
||
|
|
|
||
|
|
var args Arguments = []interface{}{"string", 123, true}
|
||
|
|
var count int
|
||
|
|
_, count = args.Diff([]interface{}{"string", Anything, true})
|
||
|
|
|
||
|
|
assert.Equal(t, 0, count)
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Arguments_Diff_WithAnythingArgument_InActualToo(t *testing.T) {
|
||
|
|
|
||
|
|
var args Arguments = []interface{}{"string", Anything, true}
|
||
|
|
var count int
|
||
|
|
_, count = args.Diff([]interface{}{"string", 123, true})
|
||
|
|
|
||
|
|
assert.Equal(t, 0, count)
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Arguments_Diff_WithAnythingOfTypeArgument(t *testing.T) {
|
||
|
|
|
||
|
|
var args Arguments = []interface{}{"string", AnythingOfType("int"), true}
|
||
|
|
var count int
|
||
|
|
_, count = args.Diff([]interface{}{"string", 123, true})
|
||
|
|
|
||
|
|
assert.Equal(t, 0, count)
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Arguments_Diff_WithAnythingOfTypeArgument_Failing(t *testing.T) {
|
||
|
|
|
||
|
|
var args Arguments = []interface{}{"string", AnythingOfType("string"), true}
|
||
|
|
var count int
|
||
|
|
var diff string
|
||
|
|
diff, count = args.Diff([]interface{}{"string", 123, true})
|
||
|
|
|
||
|
|
assert.Equal(t, 1, count)
|
||
|
|
assert.Contains(t, diff, `string != type int - %!s(int=123)`)
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Arguments_Assert(t *testing.T) {
|
||
|
|
|
||
|
|
var args Arguments = []interface{}{"string", 123, true}
|
||
|
|
|
||
|
|
assert.True(t, args.Assert(t, "string", 123, true))
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Arguments_String_Representation(t *testing.T) {
|
||
|
|
|
||
|
|
var args Arguments = []interface{}{"string", 123, true}
|
||
|
|
assert.Equal(t, `string,int,bool`, args.String())
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Arguments_String(t *testing.T) {
|
||
|
|
|
||
|
|
var args Arguments = []interface{}{"string", 123, true}
|
||
|
|
assert.Equal(t, "string", args.String(0))
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Arguments_Error(t *testing.T) {
|
||
|
|
|
||
|
|
var err error = errors.New("An Error")
|
||
|
|
var args Arguments = []interface{}{"string", 123, true, err}
|
||
|
|
assert.Equal(t, err, args.Error(3))
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Arguments_Error_Nil(t *testing.T) {
|
||
|
|
|
||
|
|
var args Arguments = []interface{}{"string", 123, true, nil}
|
||
|
|
assert.Equal(t, nil, args.Error(3))
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Arguments_Int(t *testing.T) {
|
||
|
|
|
||
|
|
var args Arguments = []interface{}{"string", 123, true}
|
||
|
|
assert.Equal(t, 123, args.Int(1))
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func Test_Arguments_Bool(t *testing.T) {
|
||
|
|
|
||
|
|
var args Arguments = []interface{}{"string", 123, true}
|
||
|
|
assert.Equal(t, true, args.Bool(2))
|
||
|
|
|
||
|
|
}
|