67 lines
1.6 KiB
Go
67 lines
1.6 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/cometbft/cometbft/crypto/tmhash"
|
|
"github.com/cometbft/cometbft/mempool"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/cosmos/cosmos-sdk/client/flags"
|
|
sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
|
|
)
|
|
|
|
func CreateContextWithErrorAndMode(err error, mode string) Context {
|
|
return Context{
|
|
Client: MockClient{err: err},
|
|
BroadcastMode: mode,
|
|
}
|
|
}
|
|
|
|
// Test the correct code is returned when
|
|
func TestBroadcastError(t *testing.T) {
|
|
errors := map[error]uint32{
|
|
mempool.ErrTxInCache: sdkerrors.ErrTxInMempoolCache.ABCICode(),
|
|
mempool.ErrTxTooLarge{}: sdkerrors.ErrTxTooLarge.ABCICode(),
|
|
mempool.ErrMempoolIsFull{}: sdkerrors.ErrMempoolIsFull.ABCICode(),
|
|
}
|
|
|
|
modes := []string{
|
|
flags.BroadcastAsync,
|
|
flags.BroadcastSync,
|
|
}
|
|
|
|
txBytes := []byte{0xA, 0xB}
|
|
txHash := fmt.Sprintf("%X", tmhash.Sum(txBytes))
|
|
|
|
for _, mode := range modes {
|
|
for err, code := range errors {
|
|
ctx := CreateContextWithErrorAndMode(err, mode)
|
|
resp, returnedErr := ctx.BroadcastTx(txBytes)
|
|
require.NoError(t, returnedErr)
|
|
require.Equal(t, code, resp.Code)
|
|
require.NotEmpty(t, resp.Codespace)
|
|
require.Equal(t, txHash, resp.TxHash)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestBroadcastCancellation(t *testing.T) {
|
|
modes := []string{
|
|
flags.BroadcastAsync,
|
|
flags.BroadcastSync,
|
|
}
|
|
|
|
txBytes := []byte{0xA, 0xB}
|
|
cmdCtx, cancel := context.WithCancel(context.Background())
|
|
cancel()
|
|
|
|
for _, mode := range modes {
|
|
ctx := CreateContextWithErrorAndMode(nil, mode).WithCmdContext(cmdCtx)
|
|
_, err := ctx.BroadcastTx(txBytes)
|
|
require.ErrorIs(t, err, context.Canceled)
|
|
}
|
|
}
|