generated mockdb as Invites.Count signature changed
This commit is contained in:
parent
5d3ccc7184
commit
b3b2f59f90
|
@ -25,10 +25,11 @@ type FakeInvitesService struct {
|
||||||
result1 roomdb.Invite
|
result1 roomdb.Invite
|
||||||
result2 error
|
result2 error
|
||||||
}
|
}
|
||||||
CountStub func(context.Context) (uint, error)
|
CountStub func(context.Context, bool) (uint, error)
|
||||||
countMutex sync.RWMutex
|
countMutex sync.RWMutex
|
||||||
countArgsForCall []struct {
|
countArgsForCall []struct {
|
||||||
arg1 context.Context
|
arg1 context.Context
|
||||||
|
arg2 bool
|
||||||
}
|
}
|
||||||
countReturns struct {
|
countReturns struct {
|
||||||
result1 uint
|
result1 uint
|
||||||
|
@ -175,18 +176,19 @@ func (fake *FakeInvitesService) ConsumeReturnsOnCall(i int, result1 roomdb.Invit
|
||||||
}{result1, result2}
|
}{result1, result2}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (fake *FakeInvitesService) Count(arg1 context.Context) (uint, error) {
|
func (fake *FakeInvitesService) Count(arg1 context.Context, arg2 bool) (uint, error) {
|
||||||
fake.countMutex.Lock()
|
fake.countMutex.Lock()
|
||||||
ret, specificReturn := fake.countReturnsOnCall[len(fake.countArgsForCall)]
|
ret, specificReturn := fake.countReturnsOnCall[len(fake.countArgsForCall)]
|
||||||
fake.countArgsForCall = append(fake.countArgsForCall, struct {
|
fake.countArgsForCall = append(fake.countArgsForCall, struct {
|
||||||
arg1 context.Context
|
arg1 context.Context
|
||||||
}{arg1})
|
arg2 bool
|
||||||
|
}{arg1, arg2})
|
||||||
stub := fake.CountStub
|
stub := fake.CountStub
|
||||||
fakeReturns := fake.countReturns
|
fakeReturns := fake.countReturns
|
||||||
fake.recordInvocation("Count", []interface{}{arg1})
|
fake.recordInvocation("Count", []interface{}{arg1, arg2})
|
||||||
fake.countMutex.Unlock()
|
fake.countMutex.Unlock()
|
||||||
if stub != nil {
|
if stub != nil {
|
||||||
return stub(arg1)
|
return stub(arg1, arg2)
|
||||||
}
|
}
|
||||||
if specificReturn {
|
if specificReturn {
|
||||||
return ret.result1, ret.result2
|
return ret.result1, ret.result2
|
||||||
|
@ -200,17 +202,17 @@ func (fake *FakeInvitesService) CountCallCount() int {
|
||||||
return len(fake.countArgsForCall)
|
return len(fake.countArgsForCall)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (fake *FakeInvitesService) CountCalls(stub func(context.Context) (uint, error)) {
|
func (fake *FakeInvitesService) CountCalls(stub func(context.Context, bool) (uint, error)) {
|
||||||
fake.countMutex.Lock()
|
fake.countMutex.Lock()
|
||||||
defer fake.countMutex.Unlock()
|
defer fake.countMutex.Unlock()
|
||||||
fake.CountStub = stub
|
fake.CountStub = stub
|
||||||
}
|
}
|
||||||
|
|
||||||
func (fake *FakeInvitesService) CountArgsForCall(i int) context.Context {
|
func (fake *FakeInvitesService) CountArgsForCall(i int) (context.Context, bool) {
|
||||||
fake.countMutex.RLock()
|
fake.countMutex.RLock()
|
||||||
defer fake.countMutex.RUnlock()
|
defer fake.countMutex.RUnlock()
|
||||||
argsForCall := fake.countArgsForCall[i]
|
argsForCall := fake.countArgsForCall[i]
|
||||||
return argsForCall.arg1
|
return argsForCall.arg1, argsForCall.arg2
|
||||||
}
|
}
|
||||||
|
|
||||||
func (fake *FakeInvitesService) CountReturns(result1 uint, result2 error) {
|
func (fake *FakeInvitesService) CountReturns(result1 uint, result2 error) {
|
||||||
|
|
Loading…
Reference in New Issue