msgmux: fix arg expectation for mock in unit test

lightningnetwork · Mar 6, 2025 · 22cad99 · 22cad99
1 parent 7e7339d
commit 22cad99
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/msgmux/msg_router_test.go b/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)
 }