config/provider/watcher/provider_test.go

package watcher_test

import (
	"context"
	"strconv"
	"sync"
	"sync/atomic"
	"testing"
	"time"

	"gitoa.ru/go-4devs/config"
	"gitoa.ru/go-4devs/config/provider/watcher"
	"gitoa.ru/go-4devs/config/test/require"
	"gitoa.ru/go-4devs/config/value"
)

var _ config.Provider = (*provider)(nil)

type provider struct {
	cnt int32
}

func (p *provider) Name() string {
	return "test"
}

func (p *provider) Value(context.Context, ...string) (config.Value, error) {
	p.cnt++

	return value.JString(strconv.Itoa(int(p.cnt))), nil
}

func TestWatcher(t *testing.T) {
	t.Parallel()

	ctx, cancel := context.WithTimeout(context.Background(), time.Second*2)
	defer func() {
		cancel()
	}()

	prov := &provider{}

	w := watcher.New(time.Second, prov)
	wg := sync.WaitGroup{}
	wg.Add(2)

	var cnt int32

	err := w.Watch(
		ctx,
		func(ctx context.Context, oldVar, newVar config.Value) error {
			atomic.AddInt32(&cnt, 1)
			wg.Done()
			if atomic.LoadInt32(&cnt) == 2 {
				return config.ErrStopWatch
			}

			return nil
		},
		"tmpname",
	)

	wg.Wait()

	require.NoError(t, err)
	require.Equal(t, int32(2), cnt)
}