104 lines
2.4 KiB
Go
104 lines
2.4 KiB
Go
|
package pool
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"code.google.com/p/go.net/context"
|
||
|
|
"github.com/drone/drone/server/worker"
|
||
|
|
"github.com/franela/goblin"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestPool(t *testing.T) {
|
||
|
|
|
||
|
|
g := goblin.Goblin(t)
|
||
|
|
g.Describe("Pool", func() {
|
||
|
|
|
||
|
|
g.It("Should allocate workers", func() {
|
||
|
|
w := mockWorker{}
|
||
|
|
pool := New()
|
||
|
|
pool.Allocate(&w)
|
||
|
|
g.Assert(len(pool.workers)).Equal(1)
|
||
|
|
g.Assert(len(pool.workerc)).Equal(1)
|
||
|
|
g.Assert(pool.workers[&w]).Equal(true)
|
||
|
|
})
|
||
|
|
|
||
|
|
g.It("Should not re-allocate an allocated worker", func() {
|
||
|
|
w := mockWorker{}
|
||
|
|
pool := New()
|
||
|
|
g.Assert(pool.Allocate(&w)).Equal(true)
|
||
|
|
g.Assert(pool.Allocate(&w)).Equal(false)
|
||
|
|
})
|
||
|
|
|
||
|
|
g.It("Should reserve a worker", func() {
|
||
|
|
w := mockWorker{}
|
||
|
|
pool := New()
|
||
|
|
pool.Allocate(&w)
|
||
|
|
g.Assert(<-pool.Reserve()).Equal(&w)
|
||
|
|
})
|
||
|
|
|
||
|
|
g.It("Should release a worker", func() {
|
||
|
|
w := mockWorker{}
|
||
|
|
pool := New()
|
||
|
|
pool.Allocate(&w)
|
||
|
|
g.Assert(len(pool.workerc)).Equal(1)
|
||
|
|
g.Assert(<-pool.Reserve()).Equal(&w)
|
||
|
|
g.Assert(len(pool.workerc)).Equal(0)
|
||
|
|
pool.Release(&w)
|
||
|
|
g.Assert(len(pool.workerc)).Equal(1)
|
||
|
|
g.Assert(<-pool.Reserve()).Equal(&w)
|
||
|
|
g.Assert(len(pool.workerc)).Equal(0)
|
||
|
|
})
|
||
|
|
|
||
|
|
g.It("Should not release an unallocated worker", func() {
|
||
|
|
w := mockWorker{}
|
||
|
|
pool := New()
|
||
|
|
g.Assert(len(pool.workers)).Equal(0)
|
||
|
|
g.Assert(len(pool.workerc)).Equal(0)
|
||
|
|
pool.Release(&w)
|
||
|
|
g.Assert(len(pool.workers)).Equal(0)
|
||
|
|
g.Assert(len(pool.workerc)).Equal(0)
|
||
|
|
pool.Release(nil)
|
||
|
|
g.Assert(len(pool.workers)).Equal(0)
|
||
|
|
g.Assert(len(pool.workerc)).Equal(0)
|
||
|
|
})
|
||
|
|
|
||
|
|
g.It("Should list all allocated workers", func() {
|
||
|
|
w1 := mockWorker{}
|
||
|
|
w2 := mockWorker{}
|
||
|
|
pool := New()
|
||
|
|
pool.Allocate(&w1)
|
||
|
|
pool.Allocate(&w2)
|
||
|
|
g.Assert(len(pool.workers)).Equal(2)
|
||
|
|
g.Assert(len(pool.workerc)).Equal(2)
|
||
|
|
g.Assert(len(pool.List())).Equal(2)
|
||
|
|
})
|
||
|
|
|
||
|
|
g.It("Should remove a worker", func() {
|
||
|
|
w1 := mockWorker{}
|
||
|
|
w2 := mockWorker{}
|
||
|
|
pool := New()
|
||
|
|
pool.Allocate(&w1)
|
||
|
|
pool.Allocate(&w2)
|
||
|
|
g.Assert(len(pool.workers)).Equal(2)
|
||
|
|
pool.Deallocate(&w1)
|
||
|
|
pool.Deallocate(&w2)
|
||
|
|
g.Assert(len(pool.workers)).Equal(0)
|
||
|
|
g.Assert(len(pool.List())).Equal(0)
|
||
|
|
})
|
||
|
|
|
||
|
|
g.It("Should add / retrieve from context", func() {
|
||
|
|
c := context.Background()
|
||
|
|
p := New()
|
||
|
|
c = NewContext(c, p)
|
||
|
|
g.Assert(FromContext(c)).Equal(p)
|
||
|
|
})
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
// fake worker for testing purpose only
|
||
|
|
type mockWorker struct {
|
||
|
|
name string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (*mockWorker) Do(c context.Context, w *worker.Work) {}
|