From 43e7131c6d9ea15297b35121cb337e6a15e3edf8 Mon Sep 17 00:00:00 2001 From: Alex | Skip Date: Wed, 4 Dec 2024 16:19:17 -0500 Subject: [PATCH] fix test --- runtime/v2/services_test.go | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/runtime/v2/services_test.go b/runtime/v2/services_test.go index 83e16d4e7141..9974d7176da7 100644 --- a/runtime/v2/services_test.go +++ b/runtime/v2/services_test.go @@ -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)