// If no workers are available (because they're all busy), an event should simply be dropped.
// This tests that functionality
func TestEventDropping(t *testing.T) {
eventsReceived := make(chan *Event)
stopWaiting := make(chan bool)
testHandler := func(event *Event, apiClient *client.RancherClient) error {
eventsReceived <- event
<-stopWaiting
return nil
}
eventHandlers := map[string]EventHandler{"physicalhost.create": testHandler}
// 2 workers, not 3, means the last event should be droppped
router := newRouter(eventHandlers, 2, t)
ready := make(chan bool, 1)
go router.Start(ready)
defer router.Stop()
defer tu.ResetTestServer()
// Wait for start to be ready
<-ready
preCount := 0
pre := func(event *Event) {
event.ID = strconv.Itoa(preCount)
event.ResourceID = strconv.FormatInt(time.Now().UnixNano(), 10)
preCount++
event.Name = "physicalhost.create;handler=testRouter"
}
// Push 3 events
for i := 0; i < 3; i++ {
err := prepAndPostEvent("../testUtils/resources/create_virtualbox.json", pre)
if err != nil {
t.Fatal(err)
}
}
receivedEvents := map[string]*Event{}
for i := 0; i < 3; i++ {
receivedEvent := awaitEvent(eventsReceived, 20, t)
if receivedEvent != nil {
receivedEvents[receivedEvent.ID] = receivedEvent
}
}
if len(receivedEvents) != 2 {
t.Errorf("Unexpected length %v", len(receivedEvents))
}
}
// Tests the simplest case of successfully receiving, routing, and handling
// three events.
func TestSimpleRouting(t *testing.T) {
eventsReceived := make(chan *Event)
testHandler := func(event *Event, apiClient *client.RancherClient) error {
eventsReceived <- event
return nil
}
eventHandlers := map[string]EventHandler{"physicalhost.create": testHandler}
router := newRouter(eventHandlers, 3, t)
ready := make(chan bool, 1)
go router.Start(ready)
defer router.Stop()
defer tu.ResetTestServer()
// Wait for start to be ready
<-ready
preCount := 0
pre := func(event *Event) {
event.ID = strconv.Itoa(preCount)
event.ResourceID = strconv.FormatInt(time.Now().UnixNano(), 10)
preCount++
event.Name = "physicalhost.create;handler=testRouter"
}
// Push 3 events
for i := 0; i < 3; i++ {
err := prepAndPostEvent("../testUtils/resources/create_virtualbox.json", pre)
if err != nil {
t.Fatal(err)
}
}
receivedEvents := map[string]*Event{}
for i := 0; i < 3; i++ {
receivedEvent := awaitEvent(eventsReceived, 100, t)
if receivedEvent != nil {
receivedEvents[receivedEvent.ID] = receivedEvent
}
}
for i := 0; i < 3; i++ {
if _, ok := receivedEvents[strconv.Itoa(i)]; !ok {
t.Errorf("Didn't get event %v", i)
}
}
}
// Tests that when we have more events than workers, workers are added back to the pool
// when they are done doing their work and capable of handling more work.
func TestWorkerReuse(t *testing.T) {
eventsReceived := make(chan *Event)
testHandler := func(event *Event, apiClient *client.RancherClient) error {
time.Sleep(10 * time.Millisecond)
eventsReceived <- event
return nil
}
eventHandlers := map[string]EventHandler{"physicalhost.create": testHandler}
router := newRouter(eventHandlers, 1, t)
ready := make(chan bool, 1)
go router.Start(ready)
defer router.Stop()
defer tu.ResetTestServer()
// Wait for start to be ready
<-ready
preCount := 1
pre := func(event *Event) {
event.ID = strconv.Itoa(preCount)
event.ResourceID = strconv.FormatInt(time.Now().UnixNano(), 10)
preCount++
event.Name = "physicalhost.create;handler=testRouter"
}
// Push 3 events
receivedEvents := map[string]*Event{}
for i := 0; i < 2; i++ {
err := prepAndPostEvent("../testUtils/resources/create_virtualbox.json", pre)
if err != nil {
t.Fatal(err)
}
receivedEvent := awaitEvent(eventsReceived, 500, t)
if receivedEvent != nil {
receivedEvents[receivedEvent.ID] = receivedEvent
}
}
if len(receivedEvents) != 2 {
t.Errorf("Unexpected length %v", len(receivedEvents))
}
}