57 lines
970 B
Go
57 lines
970 B
Go
package pool
|
|
|
|
import (
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/navidrome/navidrome/log"
|
|
"github.com/navidrome/navidrome/tests"
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
func TestPool(t *testing.T) {
|
|
tests.Init(t, false)
|
|
log.SetLevel(log.LevelCritical)
|
|
RegisterFailHandler(Fail)
|
|
RunSpecs(t, "Pool Suite")
|
|
}
|
|
|
|
type testItem struct {
|
|
ID int
|
|
}
|
|
|
|
var (
|
|
processed []int
|
|
mutex sync.RWMutex
|
|
)
|
|
|
|
var _ = Describe("Pool", func() {
|
|
var pool *Pool
|
|
|
|
BeforeEach(func() {
|
|
processed = nil
|
|
pool, _ = NewPool("test", 2, execute)
|
|
})
|
|
|
|
It("processes items", func() {
|
|
for i := 0; i < 5; i++ {
|
|
pool.Submit(&testItem{ID: i})
|
|
}
|
|
Eventually(func() []int {
|
|
mutex.RLock()
|
|
defer mutex.RUnlock()
|
|
return processed
|
|
}, "10s").Should(HaveLen(5))
|
|
|
|
Expect(processed).To(ContainElements(0, 1, 2, 3, 4))
|
|
})
|
|
})
|
|
|
|
func execute(workload interface{}) {
|
|
mutex.Lock()
|
|
defer mutex.Unlock()
|
|
item := workload.(*testItem)
|
|
processed = append(processed, item.ID)
|
|
}
|