msgmux: fix arg expectation for mock in unit test

This commit is contained in:
Olaoluwa Osuntokun 2025-03-04 18:18:02 -08:00
parent 8d9ed0ca73
commit 720d98cc15
3 changed files with 4 additions and 2 deletions

2
log.go
View File

@ -44,6 +44,7 @@ import (
"github.com/lightningnetwork/lnd/lnwallet/chanfunding"
"github.com/lightningnetwork/lnd/lnwallet/rpcwallet"
"github.com/lightningnetwork/lnd/monitoring"
"github.com/lightningnetwork/lnd/msgmux"
"github.com/lightningnetwork/lnd/netann"
"github.com/lightningnetwork/lnd/peer"
"github.com/lightningnetwork/lnd/peernotifier"
@ -202,6 +203,7 @@ func SetupLoggers(root *build.SubLoggerManager, interceptor signal.Interceptor)
)
AddV1SubLogger(root, graphdb.Subsystem, interceptor, graphdb.UseLogger)
AddSubLogger(root, chainio.Subsystem, interceptor, chainio.UseLogger)
AddSubLogger(root, msgmux.Subsystem, interceptor, msgmux.UseLogger)
}
// AddSubLogger is a helper method to conveniently create and register the

View File

@ -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
}

View File

@ -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)
}