timer_test.go

package incr

import (
	"context"
	"testing"
	"time"

	"github.com/wcharczuk/go-incr/testutil"
)

func Test_Timer(t *testing.T) {
	ctx := testContext()
	clock := time.Now()
	g := New()

	timer := Timer(g, Return(g, 0), 500*time.Millisecond)
	timer.Node().SetLabel("timer-a")
	timer.(*timerIncr[int]).clockSource = func(_ context.Context) time.Time {
		return clock
	}

	testutil.Matches(t, `timer\[(.*)\]:timer-a@-1`, timer.(*timerIncr[int]).String())

	// nop
	timer.(*timerIncr[int]).Always()

	var counterTimed int
	timed := Map(g, timer, func(base int) int {
		counterTimed++
		return base + counterTimed + 1
	})
	timed.Node().SetLabel("timed")

	var counterUntimed int
	untimed := Map(g, Return(g, 0), func(base int) int {
		counterUntimed++
		return base + counterUntimed + 1
	})
	untimed.Node().SetLabel("untimed")

	final := Map2(g, timed, untimed, func(a, b int) int {
		return a + b
	})
	final.Node().SetLabel("final")

	o := MustObserve(g, final)

	err := g.Stabilize(ctx)
	testutil.Nil(t, err)
	testutil.Equal(t, 0, timer.Value())
	testutil.Equal(t, 2, timed.Value())
	testutil.Equal(t, 2, untimed.Value())
	testutil.Equal(t, 4, o.Value())

	err = g.Stabilize(ctx)
	testutil.Nil(t, err)
	testutil.Equal(t, 4, o.Value())

	clock = clock.Add(time.Second)

	err = g.Stabilize(ctx)
	testutil.Nil(t, err)
	testutil.Equal(t, 5, o.Value())
}