172 lines
2.7 KiB
Go
172 lines
2.7 KiB
Go
package pool
|
|
|
|
import (
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
. "gopkg.in/go-playground/assert.v1"
|
|
)
|
|
|
|
// NOTES:
|
|
// - Run "go test" to run tests
|
|
// - Run "gocov test | gocov report" to report on test converage by file
|
|
// - Run "gocov test | gocov annotate -" to report on all code and functions, those ,marked with "MISS" were never called
|
|
//
|
|
// or
|
|
//
|
|
// -- may be a good idea to change to output path to somewherelike /tmp
|
|
// go test -coverprofile cover.out && go tool cover -html=cover.out -o cover.html
|
|
//
|
|
|
|
func TestUnlimitedBatch(t *testing.T) {
|
|
|
|
newFunc := func(i int) func(WorkUnit) (interface{}, error) {
|
|
return func(WorkUnit) (interface{}, error) {
|
|
time.Sleep(time.Second * 1)
|
|
return i, nil
|
|
}
|
|
}
|
|
|
|
pool := New()
|
|
defer pool.Close()
|
|
|
|
batch := pool.Batch()
|
|
|
|
for i := 0; i < 4; i++ {
|
|
batch.Queue(newFunc(i))
|
|
}
|
|
|
|
batch.QueueComplete()
|
|
|
|
var count int
|
|
|
|
for range batch.Results() {
|
|
count++
|
|
}
|
|
|
|
Equal(t, count, 4)
|
|
}
|
|
|
|
func TestUnlimitedBatchGlobalPool(t *testing.T) {
|
|
|
|
newFunc := func(i int) func(WorkUnit) (interface{}, error) {
|
|
return func(WorkUnit) (interface{}, error) {
|
|
time.Sleep(time.Second * 1)
|
|
return i, nil
|
|
}
|
|
}
|
|
|
|
batch := unlimitedGpool.Batch()
|
|
|
|
for i := 0; i < 4; i++ {
|
|
batch.Queue(newFunc(i))
|
|
}
|
|
|
|
batch.QueueComplete()
|
|
|
|
var count int
|
|
|
|
for range batch.Results() {
|
|
count++
|
|
}
|
|
|
|
Equal(t, count, 4)
|
|
}
|
|
|
|
func TestUnlimitedBatchCancelItemsThrownAway(t *testing.T) {
|
|
|
|
newFunc := func(i int) func(WorkUnit) (interface{}, error) {
|
|
return func(WorkUnit) (interface{}, error) {
|
|
time.Sleep(time.Second * 1)
|
|
return i, nil
|
|
}
|
|
}
|
|
|
|
pool := New()
|
|
defer pool.Close()
|
|
|
|
batch := pool.Batch()
|
|
|
|
go func() {
|
|
for i := 0; i < 40; i++ {
|
|
batch.Queue(newFunc(i))
|
|
}
|
|
}()
|
|
|
|
batch.Cancel()
|
|
|
|
var count int
|
|
|
|
for range batch.Results() {
|
|
count++
|
|
}
|
|
|
|
NotEqual(t, count, 40)
|
|
}
|
|
|
|
func TestUnlimitedBatchCancelItemsCancelledAfterward(t *testing.T) {
|
|
|
|
newFunc := func(i int) func(WorkUnit) (interface{}, error) {
|
|
return func(WorkUnit) (interface{}, error) {
|
|
time.Sleep(time.Second * 1)
|
|
return i, nil
|
|
}
|
|
}
|
|
|
|
pool := New()
|
|
defer pool.Close()
|
|
|
|
batch := pool.Batch()
|
|
|
|
go func() {
|
|
for i := 0; i < 40; i++ {
|
|
batch.Queue(newFunc(i))
|
|
}
|
|
}()
|
|
|
|
time.Sleep(time.Second * 2)
|
|
batch.Cancel()
|
|
|
|
var count int
|
|
|
|
for range batch.Results() {
|
|
count++
|
|
}
|
|
|
|
Equal(t, count, 40)
|
|
}
|
|
|
|
func TestUnlimitedBatchWaitAll(t *testing.T) {
|
|
|
|
var count int
|
|
var m sync.Mutex
|
|
|
|
newFunc := func(i int) func(WorkUnit) (interface{}, error) {
|
|
return func(WorkUnit) (interface{}, error) {
|
|
time.Sleep(time.Second * 1)
|
|
m.Lock()
|
|
count++
|
|
m.Unlock()
|
|
return i, nil
|
|
}
|
|
}
|
|
|
|
pool := New()
|
|
defer pool.Close()
|
|
|
|
batch := pool.Batch()
|
|
|
|
go func() {
|
|
|
|
for i := 0; i < 10; i++ {
|
|
batch.Queue(newFunc(i))
|
|
}
|
|
|
|
batch.QueueComplete()
|
|
}()
|
|
|
|
batch.WaitAll()
|
|
|
|
Equal(t, count, 10)
|
|
}
|