github.com/fibonacci-chain/fbc@v0.0.0-20231124064014-c7636198c1e9/libs/cosmos-sdk/x/auth/ante/setup_test.go (about)

     1  package ante_test
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/fibonacci-chain/fbc/libs/tendermint/crypto"
     7  	"github.com/stretchr/testify/require"
     8  
     9  	sdk "github.com/fibonacci-chain/fbc/libs/cosmos-sdk/types"
    10  	sdkerrors "github.com/fibonacci-chain/fbc/libs/cosmos-sdk/types/errors"
    11  	"github.com/fibonacci-chain/fbc/libs/cosmos-sdk/x/auth/ante"
    12  	"github.com/fibonacci-chain/fbc/libs/cosmos-sdk/x/auth/types"
    13  )
    14  
    15  func TestSetup(t *testing.T) {
    16  	// setup
    17  	_, ctx := createTestApp(true)
    18  
    19  	// keys and addresses
    20  	priv1, _, addr1 := types.KeyTestPubAddr()
    21  
    22  	// msg and signatures
    23  	msg1 := types.NewTestMsg(addr1)
    24  	fee := types.NewTestStdFee()
    25  
    26  	msgs := []sdk.Msg{msg1}
    27  
    28  	privs, accNums, seqs := []crypto.PrivKey{priv1}, []uint64{0}, []uint64{0}
    29  	tx := types.NewTestTx(ctx, msgs, privs, accNums, seqs, fee)
    30  
    31  	sud := ante.NewSetUpContextDecorator()
    32  	antehandler := sdk.ChainAnteDecorators(sud)
    33  
    34  	// Set height to non-zero value for GasMeter to be set
    35  	ctx.SetBlockHeight(1)
    36  
    37  	// Context GasMeter Limit not set
    38  	require.Equal(t, uint64(0), ctx.GasMeter().Limit(), "GasMeter set with limit before setup")
    39  
    40  	newCtx, err := antehandler(ctx, tx, false)
    41  	require.Nil(t, err, "SetUpContextDecorator returned error")
    42  
    43  	// Context GasMeter Limit should be set after SetUpContextDecorator runs
    44  	require.Equal(t, fee.Gas, newCtx.GasMeter().Limit(), "GasMeter not set correctly")
    45  }
    46  
    47  func TestRecoverPanic(t *testing.T) {
    48  	// setup
    49  	_, ctx := createTestApp(true)
    50  
    51  	// keys and addresses
    52  	priv1, _, addr1 := types.KeyTestPubAddr()
    53  
    54  	// msg and signatures
    55  	msg1 := types.NewTestMsg(addr1)
    56  	fee := types.NewTestStdFee()
    57  
    58  	msgs := []sdk.Msg{msg1}
    59  
    60  	privs, accNums, seqs := []crypto.PrivKey{priv1}, []uint64{0}, []uint64{0}
    61  	tx := types.NewTestTx(ctx, msgs, privs, accNums, seqs, fee)
    62  
    63  	sud := ante.NewSetUpContextDecorator()
    64  	antehandler := sdk.ChainAnteDecorators(sud, OutOfGasDecorator{})
    65  
    66  	// Set height to non-zero value for GasMeter to be set
    67  	ctx.SetBlockHeight(1)
    68  
    69  	newCtx, err := antehandler(ctx, tx, false)
    70  
    71  	require.NotNil(t, err, "Did not return error on OutOfGas panic")
    72  
    73  	require.True(t, sdkerrors.ErrOutOfGas.Is(err), "Returned error is not an out of gas error")
    74  	require.Equal(t, fee.Gas, newCtx.GasMeter().Limit())
    75  
    76  	antehandler = sdk.ChainAnteDecorators(sud, PanicDecorator{})
    77  	require.Panics(t, func() { antehandler(ctx, tx, false) }, "Recovered from non-Out-of-Gas panic")
    78  }
    79  
    80  type OutOfGasDecorator struct{}
    81  
    82  // AnteDecorator that will throw OutOfGas panic
    83  func (ogd OutOfGasDecorator) AnteHandle(ctx sdk.Context, tx sdk.Tx, simulate bool, next sdk.AnteHandler) (sdk.Context, error) {
    84  	overLimit := ctx.GasMeter().Limit() + 1
    85  
    86  	// Should panic with outofgas error
    87  	ctx.GasMeter().ConsumeGas(overLimit, "test panic")
    88  
    89  	// not reached
    90  	return next(ctx, tx, simulate)
    91  }
    92  
    93  type PanicDecorator struct{}
    94  
    95  func (pd PanicDecorator) AnteHandle(ctx sdk.Context, tx sdk.Tx, simulate bool, next sdk.AnteHandler) (newCtx sdk.Context, err error) {
    96  	panic("random error")
    97  }