27 lines
627 B
Go
27 lines
627 B
Go
package timewheel
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func Test_timeWheel(t *testing.T) {
|
|
timeWheel := New(10, 100*time.Millisecond)
|
|
defer timeWheel.Stop()
|
|
|
|
<-time.After(10 * time.Millisecond)
|
|
|
|
t.Errorf("test2, %v", time.Now())
|
|
timeWheel.AddTask("test1", TimeTypeLoop, func() {
|
|
t.Errorf("test1, %v", time.Now())
|
|
}, time.Now().Add(time.Second))
|
|
timeWheel.AddTask("test2", TimeTypeLoop, func() {
|
|
t.Errorf("test2, %v", time.Now())
|
|
}, time.Now().Add(5*time.Second))
|
|
timeWheel.AddTask("test2", TimerTypeOnce, func() {
|
|
t.Errorf("test2, %v", time.Now())
|
|
}, time.Now().Add(3*time.Second))
|
|
|
|
<-time.After(6 * time.Second)
|
|
}
|