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

cosmos · Dec 5, 2024 · 227854e · 227854e
1 parent ec5225c
commit 227854e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git 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)