// Test unknown user events are dispatched properly func (suite *EventHandlerTestSuite) TestUserEvent_UnknownEvent() { // Create Member Event evt := serf.UserEvent{ LTime: serf.LamportTime(0), Name: "unk", Payload: make([]byte, 0), Coalesce: false, } // Add UserEvent handler m := &MockUserEventHandler{} m.On("HandleUserEvent", evt).Return() suite.Handler.UnknownEventHandler = m // Process event suite.Handler.HandleEvent(evt) m.AssertCalled(suite.T(), "HandleUserEvent", evt) }
// Test user events are dispatched properly func (suite *EventHandlerTestSuite) TestUserEvent() { // Create Member Event evt := serf.UserEvent{ LTime: serf.LamportTime(0), Name: suite.Prefix + ":user-event", Payload: make([]byte, 0), Coalesce: false, } modified := evt modified.Name = "user-event" // Add UserEvent handler m := &MockUserEventHandler{} m.On("HandleUserEvent", modified).Return() suite.Handler.UserEvent = m // Process event suite.Handler.HandleEvent(evt) m.AssertCalled(suite.T(), "HandleUserEvent", modified) }
// Test QueryEvent messages are dispatched properly func (suite *EventHandlerTestSuite) TestQueryEvent() { // Create Query query := serf.Query{ LTime: serf.LamportTime(0), Name: "Event", Payload: make([]byte, 0), } // Add UserEvent handler m := &MockQueryEventHandler{} m.On("HandleQueryEvent", query).Return() suite.Handler.QueryHandler = m // Add Reconciler r := &MockReconciler{} r.On("Reconcile", query).Return() suite.Handler.Reconciler = r // Process event suite.Handler.HandleEvent(&query) m.AssertCalled(suite.T(), "HandleQueryEvent", query) r.AssertNotCalled(suite.T(), "Reconcile", query) }