woodpecker/pkg/bus/builtin/bus_test.go

51 lines
962 B
Go
Raw Normal View History

2015-04-28 22:30:51 +00:00
package builtin
import (
"testing"
2015-05-22 18:37:40 +00:00
. "github.com/drone/drone/Godeps/_workspace/src/github.com/franela/goblin"
"github.com/drone/drone/pkg/bus"
2015-04-28 22:30:51 +00:00
)
func TestBuild(t *testing.T) {
g := Goblin(t)
g.Describe("Bus", func() {
g.It("Should unsubscribe", func() {
c1 := make(chan *bus.Event)
c2 := make(chan *bus.Event)
2015-04-28 22:30:51 +00:00
b := New()
b.Subscribe(c1)
b.Subscribe(c2)
g.Assert(len(b.subs)).Equal(2)
})
g.It("Should subscribe", func() {
c1 := make(chan *bus.Event)
c2 := make(chan *bus.Event)
2015-04-28 22:30:51 +00:00
b := New()
b.Subscribe(c1)
b.Subscribe(c2)
g.Assert(len(b.subs)).Equal(2)
b.Unsubscribe(c1)
b.Unsubscribe(c2)
g.Assert(len(b.subs)).Equal(0)
})
g.It("Should send", func() {
e1 := &bus.Event{Name: "foo"}
e2 := &bus.Event{Name: "bar"}
c := make(chan *bus.Event)
2015-04-28 22:30:51 +00:00
b := New()
b.Subscribe(c)
b.Send(e1)
b.Send(e2)
r1 := <-c
r2 := <-c
g.Assert(e1).Equal(r1)
g.Assert(e2).Equal(r2)
})
})
}