2ae606f132
* Internal registry for disambiguated imports, vars - Move functionality in the moq package partially into internal/{registry,template}. - Leverage registry to assign unique package and variable/method parameter names. Use import aliases if present in interface source package. BREAKING CHANGE: When the interface definition does not mention the parameter names, the field names in call info anonymous struct will be different. The new field names are generated using the type info (string -> s, int -> n, chan int -> intCh, []MyType -> myTypes, map[string]int -> stringToInt etc.). For example, for a string parameter previously if the field name was 'In1', the new field could be 'S' or 'S1' (depends on number of string method parameters). * Refactor golden file tests to be table-driven * Fix sync pkg alias handling for moq generation * Improve, add tests (increase coverage) * Use $.Foo in template, avoid declaring variables $ is set to the data argument passed to Execute, that is, to the starting value of dot. Variables were declared to be able to refer to the parent context. * Consistent template field formatting * Use tabs in generated Godoc comments' example code * Minor simplification * go generate * Fix conflict for generated param name of pointer type Excellent work by @sudo-suhas.
136 lines
3.3 KiB
Go
Executable File
136 lines
3.3 KiB
Go
Executable File
// Code generated by moq; DO NOT EDIT.
|
|
// github.com/matryer/moq
|
|
|
|
package example
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
)
|
|
|
|
// Ensure, that PersonStoreMock does implement PersonStore.
|
|
// If this is not the case, regenerate this file with moq.
|
|
var _ PersonStore = &PersonStoreMock{}
|
|
|
|
// PersonStoreMock is a mock implementation of PersonStore.
|
|
//
|
|
// func TestSomethingThatUsesPersonStore(t *testing.T) {
|
|
//
|
|
// // make and configure a mocked PersonStore
|
|
// mockedPersonStore := &PersonStoreMock{
|
|
// CreateFunc: func(ctx context.Context, person *Person, confirm bool) error {
|
|
// panic("mock out the Create method")
|
|
// },
|
|
// GetFunc: func(ctx context.Context, id string) (*Person, error) {
|
|
// panic("mock out the Get method")
|
|
// },
|
|
// }
|
|
//
|
|
// // use mockedPersonStore in code that requires PersonStore
|
|
// // and then make assertions.
|
|
//
|
|
// }
|
|
type PersonStoreMock struct {
|
|
// CreateFunc mocks the Create method.
|
|
CreateFunc func(ctx context.Context, person *Person, confirm bool) error
|
|
|
|
// GetFunc mocks the Get method.
|
|
GetFunc func(ctx context.Context, id string) (*Person, error)
|
|
|
|
// calls tracks calls to the methods.
|
|
calls struct {
|
|
// Create holds details about calls to the Create method.
|
|
Create []struct {
|
|
// Ctx is the ctx argument value.
|
|
Ctx context.Context
|
|
// Person is the person argument value.
|
|
Person *Person
|
|
// Confirm is the confirm argument value.
|
|
Confirm bool
|
|
}
|
|
// Get holds details about calls to the Get method.
|
|
Get []struct {
|
|
// Ctx is the ctx argument value.
|
|
Ctx context.Context
|
|
// ID is the id argument value.
|
|
ID string
|
|
}
|
|
}
|
|
lockCreate sync.RWMutex
|
|
lockGet sync.RWMutex
|
|
}
|
|
|
|
// Create calls CreateFunc.
|
|
func (mock *PersonStoreMock) Create(ctx context.Context, person *Person, confirm bool) error {
|
|
if mock.CreateFunc == nil {
|
|
panic("PersonStoreMock.CreateFunc: method is nil but PersonStore.Create was just called")
|
|
}
|
|
callInfo := struct {
|
|
Ctx context.Context
|
|
Person *Person
|
|
Confirm bool
|
|
}{
|
|
Ctx: ctx,
|
|
Person: person,
|
|
Confirm: confirm,
|
|
}
|
|
mock.lockCreate.Lock()
|
|
mock.calls.Create = append(mock.calls.Create, callInfo)
|
|
mock.lockCreate.Unlock()
|
|
return mock.CreateFunc(ctx, person, confirm)
|
|
}
|
|
|
|
// CreateCalls gets all the calls that were made to Create.
|
|
// Check the length with:
|
|
// len(mockedPersonStore.CreateCalls())
|
|
func (mock *PersonStoreMock) CreateCalls() []struct {
|
|
Ctx context.Context
|
|
Person *Person
|
|
Confirm bool
|
|
} {
|
|
var calls []struct {
|
|
Ctx context.Context
|
|
Person *Person
|
|
Confirm bool
|
|
}
|
|
mock.lockCreate.RLock()
|
|
calls = mock.calls.Create
|
|
mock.lockCreate.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// Get calls GetFunc.
|
|
func (mock *PersonStoreMock) Get(ctx context.Context, id string) (*Person, error) {
|
|
if mock.GetFunc == nil {
|
|
panic("PersonStoreMock.GetFunc: method is nil but PersonStore.Get was just called")
|
|
}
|
|
callInfo := struct {
|
|
Ctx context.Context
|
|
ID string
|
|
}{
|
|
Ctx: ctx,
|
|
ID: id,
|
|
}
|
|
mock.lockGet.Lock()
|
|
mock.calls.Get = append(mock.calls.Get, callInfo)
|
|
mock.lockGet.Unlock()
|
|
return mock.GetFunc(ctx, id)
|
|
}
|
|
|
|
// GetCalls gets all the calls that were made to Get.
|
|
// Check the length with:
|
|
// len(mockedPersonStore.GetCalls())
|
|
func (mock *PersonStoreMock) GetCalls() []struct {
|
|
Ctx context.Context
|
|
ID string
|
|
} {
|
|
var calls []struct {
|
|
Ctx context.Context
|
|
ID string
|
|
}
|
|
mock.lockGet.RLock()
|
|
calls = mock.calls.Get
|
|
mock.lockGet.RUnlock()
|
|
return calls
|
|
}
|