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