test: fix failing runtime/v2 test (#22760)

This commit is contained in:
Alex | Skip 2024-12-05 03:29:55 -05:00 committed by GitHub
parent ec5225c31c
commit 227854e258
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -39,8 +39,11 @@ func TestRegisterServices(t *testing.T) {
},
}
mockModule.On("RegisterMsgHandlers", app.msgRouterBuilder).Once()
mockModule.On("RegisterQueryHandlers", app.queryRouterBuilder).Once()
msgWrapper := newStfRouterWrapper(app.msgRouterBuilder)
queryWrapper := newStfRouterWrapper(app.queryRouterBuilder)
mockModule.On("RegisterMsgHandlers", &msgWrapper).Once()
mockModule.On("RegisterQueryHandlers", &queryWrapper).Once()
err := mm.RegisterServices(app)