fix test

cosmos · Dec 4, 2024 · 43e7131 · 43e7131
1 parent 556102c
commit 43e7131
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)