49 lines
859 B
Go
49 lines
859 B
Go
|
package utils
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestTestingClock(t *testing.T) {
|
||
|
|
c := &TestingClock{
|
||
|
|
now: time.Unix(0, 0),
|
||
|
|
}
|
||
|
|
|
||
|
|
assert.Equal(t, int64(0), c.Now().Unix())
|
||
|
|
c.now = time.Unix(20, 0)
|
||
|
|
|
||
|
|
assert.Equal(t, int64(20), c.Now().Unix())
|
||
|
|
assert.Equal(t, int64(20000000000), c.Now().UnixNano())
|
||
|
|
|
||
|
|
c.Set(time.Unix(16000000, 0))
|
||
|
|
|
||
|
|
assert.Equal(t, int64(16000000), c.Now().Unix())
|
||
|
|
|
||
|
|
before := c.Now()
|
||
|
|
|
||
|
|
<-c.After(time.Millisecond * 100)
|
||
|
|
|
||
|
|
assert.Equal(t, before, c.Now())
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestRealClock(t *testing.T) {
|
||
|
|
c := &RealClock{}
|
||
|
|
|
||
|
|
assert.WithinDuration(t, time.Now(), c.Now(), time.Second)
|
||
|
|
|
||
|
|
before := c.Now()
|
||
|
|
|
||
|
|
<-c.After(time.Millisecond * 100)
|
||
|
|
|
||
|
|
after := c.Now()
|
||
|
|
|
||
|
|
assert.WithinDuration(t, before, after, time.Millisecond*120)
|
||
|
|
|
||
|
|
diff := after.Sub(before)
|
||
|
|
|
||
|
|
assert.GreaterOrEqual(t, diff, time.Millisecond*100)
|
||
|
|
}
|