|
|
@@ -1,6 +1,7 @@
|
|
|
package task_test
|
|
|
|
|
|
import (
|
|
|
+ "sync/atomic"
|
|
|
"testing"
|
|
|
"time"
|
|
|
|
|
|
@@ -9,28 +10,33 @@ import (
|
|
|
)
|
|
|
|
|
|
func TestPeriodicTaskStop(t *testing.T) {
|
|
|
- value := 0
|
|
|
+ var value uint64
|
|
|
task := &Periodic{
|
|
|
Interval: time.Second * 2,
|
|
|
Execute: func() error {
|
|
|
- value++
|
|
|
+ atomic.AddUint64(&value, 1)
|
|
|
return nil
|
|
|
},
|
|
|
}
|
|
|
common.Must(task.Start())
|
|
|
time.Sleep(time.Second * 5)
|
|
|
common.Must(task.Close())
|
|
|
- if value != 3 {
|
|
|
- t.Fatal("expected 3, but got ", value)
|
|
|
+ value1 := atomic.LoadUint64(&value)
|
|
|
+ if value1 != 3 {
|
|
|
+ t.Fatal("expected 3, but got ", value1)
|
|
|
}
|
|
|
+
|
|
|
time.Sleep(time.Second * 4)
|
|
|
- if value != 3 {
|
|
|
- t.Fatal("expected 3, but got ", value)
|
|
|
+ value2 := atomic.LoadUint64(&value)
|
|
|
+ if value2 != 3 {
|
|
|
+ t.Fatal("expected 3, but got ", value2)
|
|
|
}
|
|
|
+
|
|
|
common.Must(task.Start())
|
|
|
time.Sleep(time.Second * 3)
|
|
|
- if value != 5 {
|
|
|
- t.Fatal("Expected 5, but ", value)
|
|
|
+ value3 := atomic.LoadUint64(&value)
|
|
|
+ if value3 != 5 {
|
|
|
+ t.Fatal("Expected 5, but ", value3)
|
|
|
}
|
|
|
common.Must(task.Close())
|
|
|
}
|