pond_test.go

package koi_test

import (
	"errors"
	"sync"
	"testing"
	"time"

	"github.com/1995parham/koi"
)

const (
	queueSize       = 0
	concurrentCount = 10
)

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

	pond := koi.NewPond[int, koi.NoReturn]()

	var wg sync.WaitGroup

	printer := func(_ int) koi.NoReturn {
		time.Sleep(1 * time.Second)

		wg.Done()

		return koi.None
	}

	printWorker := koi.MustNewWoker(printer, queueSize, concurrentCount)

	pond.MustRegisterWorker("printer", printWorker)

	for i := 0; i < concurrentCount; i++ {
		wg.Add(1)

		if _, err := pond.AddWork("printer", i); err != nil {
			t.Errorf("error while adding job: %s", err)
		}
	}

	wg.Wait()
}

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

	pond := koi.NewPond[int, koi.NoReturn]()

	if _, err := pond.AddWork("printer", 1); !errors.Is(err, koi.ErrWorkerNotFound) {
		t.Error("expects not found error")
	}
}

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

	pond := koi.NewPond[int, int]()

	square := func(i int) int {
		return i * i
	}

	printWorker := koi.MustNewWoker(square, queueSize, concurrentCount)

	pond.MustRegisterWorker("square", printWorker)

	for i := 0; i < concurrentCount; i++ {
		if _, err := pond.AddWork("square", i); err != nil {
			t.Errorf("error while adding job: %s", err)
		}
	}

	ch := pond.ResultChan("square")
	results := make(map[int]bool)

	for i := 0; i < concurrentCount; i++ {
		r := <-ch
		results[r] = true
	}

	for i := 0; i < concurrentCount; i++ {
		if _, ok := results[i*i]; !ok {
			t.Errorf("cannot find result for %d", i)
		}
	}
}