mirror of
https://github.com/OffchainLabs/prysm.git
synced 2026-01-07 06:34:01 -05:00
41 lines
793 B
Go
41 lines
793 B
Go
package async_test
|
|
|
|
import (
|
|
"context"
|
|
"sync/atomic"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/OffchainLabs/prysm/v7/async"
|
|
)
|
|
|
|
func TestEveryRuns(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(t.Context())
|
|
|
|
i := int32(0)
|
|
async.RunEvery(ctx, 100*time.Millisecond, func() {
|
|
atomic.AddInt32(&i, 1)
|
|
})
|
|
|
|
// Sleep for a bit and ensure the value has increased.
|
|
time.Sleep(200 * time.Millisecond)
|
|
|
|
if atomic.LoadInt32(&i) == 0 {
|
|
t.Error("Counter failed to increment with ticker")
|
|
}
|
|
|
|
cancel()
|
|
|
|
// Sleep for a bit to let the cancel take place.
|
|
time.Sleep(100 * time.Millisecond)
|
|
|
|
last := atomic.LoadInt32(&i)
|
|
|
|
// Sleep for a bit and ensure the value has not increased.
|
|
time.Sleep(200 * time.Millisecond)
|
|
|
|
if atomic.LoadInt32(&i) != last {
|
|
t.Error("Counter incremented after stop")
|
|
}
|
|
}
|