msgmux: fix arg expectation for mock in unit test

msgmux/msg_router.go

@@ -244,7 +244,7 @@ func (p *MultiMsgRouter) msgRouter(ctx context.Context) {
 			var err error
 			if !couldSend {
 				log.Tracef("MsgRouter: unable to route "+
-					"msg %T", msg)
+					"msg %T", msg.Message)
 
 				err = ErrUnableToRouteMsg
 			}

msgmux/msg_router_test.go

@@ -26,7 +26,7 @@ func (m *mockEndpoint) CanHandle(msg PeerMsg) bool {
 }
 
 func (m *mockEndpoint) SendMessage(ctx context.Context, msg PeerMsg) bool {
-	args := m.Called(msg)
+	args := m.Called(ctx, msg)
 
 	return args.Bool(0)
 }