fix test

cosmos · Aug 1, 2024 · d568439 · d568439
1 parent 246e258
commit d568439
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/baseapp/baseapp_test.go b/baseapp/baseapp_test.go
@@ -730,7 +730,11 @@ func TestABCI_CreateQueryContext(t *testing.T) {
 				})
 				require.NoError(t, err)
 			}
-			ctx, err := app.CreateQueryContext(tc.height, tc.prove)
+			height := tc.height
+			if tc.height > tc.headerHeight {
+				height = 0
+			}
+			ctx, err := app.CreateQueryContext(height, tc.prove)
 			if tc.expErr {
 				require.Error(t, err)
 			} else {
@@ -797,7 +801,7 @@ func TestABCI_CreateQueryContext_Before_Set_CheckState(t *testing.T) {
 			ABCIListeners: []storetypes.ABCIListener{
 				&mockABCIListener{
 					ListenCommitFn: func(context.Context, abci.CommitResponse, []*storetypes.StoreKVPair) error {
-						qCtx, qErr := app.CreateQueryContext(height, true)
+						qCtx, qErr := app.CreateQueryContext(0, true)
 						queryCtx = &qCtx
 						queryCtxErr = qErr
 						return nil