types

type MockConsumer ¶

type MockConsumer struct {
	// add a Mock object instance
	mock.Mock
	// other fields go here as normal
	TotalCount  int
	DoneChannel chan bool
}

A mock consumer consumes all incoming messages and sends a "done" signal on its DoneChannel once the expected number of messages has been consumed

func (test *MockConsumer) ConsumeMessage(msg *message.DataMessage)