fix benchmarks with updated endblocker

terra-money · Nov 16, 2022 · d544985 · d544985
1 parent cb8a4e7
commit d544985
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/x/alliance/benchmark/benchmark_test.go b/x/alliance/benchmark/benchmark_test.go
@@ -93,16 +93,17 @@ func TestRunBenchmarks(t *testing.T) {
 		}
 
 		// Endblock
+		assets := app.AllianceKeeper.GetAllAssets(ctx)
 		app.AllianceKeeper.CompleteRedelegations(ctx)
 		err = app.AllianceKeeper.CompleteUndelegations(ctx)
 		if err != nil {
 			panic(err)
 		}
-		_, err = app.AllianceKeeper.DeductAssetsHook(ctx)
+		_, err = app.AllianceKeeper.DeductAssetsHook(ctx, assets)
 		if err != nil {
 			panic(err)
 		}
-		err = app.AllianceKeeper.RebalanceHook(ctx)
+		err = app.AllianceKeeper.RebalanceHook(ctx, assets)
 		if err != nil {
 			panic(err)
 		}