gwp/gwp_test.go

package gwp

import (
	"errors"
	"sync/atomic"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestBasic(t *testing.T) {
	for i := 0; i < 10; i++ {
		wp := New(i)

		count := new(int64)

		for j := 0; j < 100; j++ {
			wp.Add(func() error {
				atomic.AddInt64(count, 1)
				return nil
			})
		}

		wp.CloseAndWait()

		assert.EqualValues(t, 100, *count)
		assert.EqualValues(t, 100, wp.processedCount)
	}
}

func TestErrorCounter(t *testing.T) {
	for i := 0; i < 10; i++ {
		wp := New(i)

		for j := 0; j < 100; j++ {
			n := j
			wp.Add(func() error {
				if n%2 == 0 {
					return errors.New("error")
				}

				return nil
			})
		}

		wp.CloseAndWait()

		assert.EqualValues(t, 100, wp.processedCount)
		assert.EqualValues(t, 50, wp.errorCount)
	}
}