msgmux: fix arg expectation for mock in unit test

This commit is contained in:
Olaoluwa Osuntokun committed 2025-03-18 11:44:59 -05:00
1 parent 8d9ed0ca73
commit 720d98cc15
3 files changed
+4 -2

No files matched your search

+1 -1
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)
}