Function TestWatchKeyWithNoStartValue

pkg/ring/kv/consul/client_test.go:229–258

(t *testing.T)

}

func TestWatchKeyWithNoStartValue(t *testing.T) {
	c, closer := NewInMemoryClient(codec.String{}, testLogger{}, prometheus.NewPedanticRegistry())
	t.Cleanup(func() {
		assert.NoError(t, closer.Close())
	})

	const key = "test"

	go func() {
		time.Sleep(100 * time.Millisecond)
		_, err := c.kv.Put(&consul.KVPair{Key: key, Value: []byte("start")}, nil)
		require.NoError(t, err)

		time.Sleep(100 * time.Millisecond)
		_, err = c.kv.Put(&consul.KVPair{Key: key, Value: []byte("end")}, nil)
		require.NoError(t, err)
	}()

	ctx, fn := context.WithTimeout(context.Background(), 300*time.Millisecond)
	defer fn()

	reported := 0
	c.WatchKey(ctx, key, func(i any) bool {
		reported++
		return reported != 2
	})

	// we should see both start and end values.
	require.Equal(t, 2, reported)
}

type testLogger struct {
}