prysm-pulse/shared/runutil/every_test.go

package runutil_test

import (
	"context"
	"testing"
	"time"

	"github.com/prysmaticlabs/prysm/shared/runutil"
)

func TestEveryRuns(t *testing.T) {
	ctx, cancel := context.WithCancel(context.Background())

	i := 0
	runutil.RunEvery(ctx, time.Duration(100*time.Millisecond), func() {
		i++
	})

	// Sleep for a bit and ensure the value has increased.
	time.Sleep(time.Duration(200 * time.Millisecond))

	if i == 0 {
		t.Error("Counter failed to increment with ticker")
	}

	cancel()

	// Sleep for a bit to let the cancel take place.
	time.Sleep(time.Duration(100 * time.Millisecond))

	last := i

	// Sleep for a bit and ensure the value has not increased.
	time.Sleep(time.Duration(200 * time.Millisecond))

	if i != last {
		t.Error("Counter incremented after stop")
	}
}