46 lines
840 B
Go
46 lines
840 B
Go
package pool
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/deluan/navidrome/log"
|
|
"github.com/deluan/navidrome/tests"
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
func TestCore(t *testing.T) {
|
|
tests.Init(t, false)
|
|
log.SetLevel(log.LevelCritical)
|
|
RegisterFailHandler(Fail)
|
|
RunSpecs(t, "Core Suite")
|
|
}
|
|
|
|
type testItem struct {
|
|
ID int
|
|
}
|
|
|
|
var processed []int
|
|
|
|
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 { return processed }, "10s").Should(HaveLen(5))
|
|
Expect(processed).To(ContainElements(0, 1, 2, 3, 4))
|
|
})
|
|
})
|
|
|
|
func execute(workload interface{}) {
|
|
item := workload.(*testItem)
|
|
processed = append(processed, item.ID)
|
|
}
|