code: func (m *MockAgent) HandleMessage(arg0 context.Context, arg1 *choria.Message, arg2 protocol.Request, arg3 choria.ConnectorInfo, arg4 chan *AgentReply) {
  	m.ctrl.Call(m, "HandleMessage", arg0, arg1, arg2, arg3, arg4)
  }
docs: // HandleMessage mocks base method