timerMock.go

package mock

import (
	"sync"
	"time"
)

// TimerMock -
type TimerMock struct {
	secondsMutex *sync.RWMutex
	seconds      int64
	NowCalled    func() time.Time
}

// NewTimerMock -
func NewTimerMock() *TimerMock {
	return &TimerMock{
		seconds:      0,
		secondsMutex: &sync.RWMutex{},
	}
}

// Now -
func (tm *TimerMock) Now() time.Time {
	if tm.NowCalled != nil {
		return tm.NowCalled()
	}

	tm.secondsMutex.RLock()
	currentSeconds := time.Unix(tm.seconds, 0)
	tm.secondsMutex.RUnlock()
	return currentSeconds
}

// IsInterfaceNil -
func (tm *TimerMock) IsInterfaceNil() bool {
	return tm == nil
}

// IncrementSeconds -
func (tm *TimerMock) IncrementSeconds(value int) {
	tm.secondsMutex.Lock()
	tm.seconds += int64(value)
	tm.secondsMutex.Unlock()
}

// SetSeconds -
func (tm *TimerMock) SetSeconds(value int) {
	tm.secondsMutex.Lock()
	tm.seconds = int64(value)
	tm.secondsMutex.Unlock()
}