// Code generated by counterfeiter. DO NOT EDIT. package fake import ( "sync" "time" ) type Sleeper struct { SleepStub func(time.Duration) sleepMutex sync.RWMutex sleepArgsForCall []struct { arg1 time.Duration } invocations map[string][][]interface{} invocationsMutex sync.RWMutex } func (fake *Sleeper) Sleep(arg1 time.Duration) { fake.sleepMutex.Lock() fake.sleepArgsForCall = append(fake.sleepArgsForCall, struct { arg1 time.Duration }{arg1}) stub := fake.SleepStub fake.recordInvocation("Sleep", []interface{}{arg1}) fake.sleepMutex.Unlock() if stub != nil { fake.SleepStub(arg1) } } func (fake *Sleeper) SleepCallCount() int { fake.sleepMutex.RLock() defer fake.sleepMutex.RUnlock() return len(fake.sleepArgsForCall) } func (fake *Sleeper) SleepCalls(stub func(time.Duration)) { fake.sleepMutex.Lock() defer fake.sleepMutex.Unlock() fake.SleepStub = stub } func (fake *Sleeper) SleepArgsForCall(i int) time.Duration { fake.sleepMutex.RLock() defer fake.sleepMutex.RUnlock() argsForCall := fake.sleepArgsForCall[i] return argsForCall.arg1 } func (fake *Sleeper) Invocations() map[string][][]interface{} { fake.invocationsMutex.RLock() defer fake.invocationsMutex.RUnlock() fake.sleepMutex.RLock() defer fake.sleepMutex.RUnlock() copiedInvocations := map[string][][]interface{}{} for key, value := range fake.invocations { copiedInvocations[key] = value } return copiedInvocations } func (fake *Sleeper) recordInvocation(key string, args []interface{}) { fake.invocationsMutex.Lock() defer fake.invocationsMutex.Unlock() if fake.invocations == nil { fake.invocations = map[string][][]interface{}{} } if fake.invocations[key] == nil { fake.invocations[key] = [][]interface{}{} } fake.invocations[key] = append(fake.invocations[key], args) }