135 lines
2.3 KiB
Go
135 lines
2.3 KiB
Go
package pool
|
|
|
|
import (
|
|
"errors"
|
|
"strconv"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestTaskPool_Stop(t *testing.T) {
|
|
pool := NewTaskPool(5, 10)
|
|
pool.Start()
|
|
|
|
lck := sync.Mutex{}
|
|
counter := 0
|
|
|
|
for i := 1; i <= 10; i++ {
|
|
f := func(i int) func() error {
|
|
return func() error {
|
|
lck.Lock()
|
|
t.Log("task", i, "locked")
|
|
counter += i
|
|
t.Log("task", i, "unlocked")
|
|
lck.Unlock()
|
|
|
|
time.Sleep(time.Duration(i*100) * time.Millisecond)
|
|
t.Log("task", i, "finished")
|
|
|
|
return nil
|
|
}
|
|
}(i)
|
|
pool.AddTask(f)
|
|
}
|
|
|
|
pool.Stop()
|
|
|
|
if counter != 55 {
|
|
t.Error("some tasks were not executed")
|
|
}
|
|
}
|
|
|
|
func TestTaskPool_WaitForTask(t *testing.T) {
|
|
pool := NewTaskPool(10, 10)
|
|
pool.Start()
|
|
|
|
counter := 0
|
|
|
|
for i := 1; i <= 10; i++ {
|
|
f := func(i int) func() error {
|
|
return func() error {
|
|
counter += 1
|
|
t.Log("task", i, "finished")
|
|
time.Sleep(100 * time.Millisecond)
|
|
return errors.New(strconv.Itoa(i))
|
|
}
|
|
}(i)
|
|
id := pool.AddTask(f)
|
|
|
|
ret := pool.WaitForTask(id)
|
|
if counter != 1 {
|
|
t.Errorf("Counter mismatch: expected %d, got %d, task %d", 1, counter, id)
|
|
}
|
|
if ret.Error() != strconv.Itoa(i) {
|
|
t.Errorf("Return value mismatch: expected %s, got %s, task %d", strconv.Itoa(i), ret.Error(), id)
|
|
}
|
|
counter -= 1
|
|
}
|
|
|
|
pool.Stop()
|
|
}
|
|
|
|
func TestTaskPool_DoubleWait(t *testing.T) {
|
|
pool := NewTaskPool(1, 1)
|
|
pool.Start()
|
|
|
|
f := func() error {
|
|
t.Log("task invoked")
|
|
return nil
|
|
}
|
|
id := pool.AddTask(f)
|
|
|
|
ret := pool.WaitForTask(id)
|
|
if ret != nil {
|
|
t.Errorf("task returned error: %v", ret)
|
|
}
|
|
|
|
ret2 := pool.WaitForTask(id)
|
|
if ret2 == nil {
|
|
t.Errorf("2nd wait returned nil")
|
|
} else if !errors.Is(ret2, &ErrTaskNotFound{}) {
|
|
t.Errorf("2nd wait returned wrong error: %v", ret2)
|
|
}
|
|
|
|
pool.Stop()
|
|
}
|
|
|
|
func TestTaskPool_One(t *testing.T) {
|
|
pool := NewTaskPool(1, 1)
|
|
pool.Start()
|
|
|
|
lck := sync.Mutex{}
|
|
counter := 0
|
|
|
|
ids := make([]int, 0)
|
|
for i := 1; i <= 10; i++ {
|
|
f := func(i int) func() error {
|
|
return func() error {
|
|
lck.Lock()
|
|
t.Log("task", i, "locked")
|
|
counter += i
|
|
t.Log("task", i, "unlocked")
|
|
lck.Unlock()
|
|
|
|
time.Sleep(time.Duration(i*10) * time.Millisecond)
|
|
t.Log("task", i, "finished")
|
|
|
|
return nil
|
|
}
|
|
}(i)
|
|
id := pool.AddTask(f)
|
|
ids = append(ids, id)
|
|
}
|
|
|
|
for _, id := range ids {
|
|
_ = pool.WaitForTask(id)
|
|
}
|
|
|
|
if counter != 55 {
|
|
t.Error("some tasks were not executed")
|
|
}
|
|
|
|
pool.Stop()
|
|
}
|