110 lines
2.6 KiB
Go
110 lines
2.6 KiB
Go
package cli
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/tendermint/tendermint/crypto/secp256k1"
|
|
|
|
"github.com/cosmos/cosmos-sdk/client/context"
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/auth"
|
|
authclient "github.com/cosmos/cosmos-sdk/x/auth/client"
|
|
)
|
|
|
|
func createFakeTxBuilder() auth.TxBuilder {
|
|
cdc := codec.New()
|
|
return auth.NewTxBuilder(
|
|
authclient.GetTxEncoder(cdc),
|
|
123,
|
|
9876,
|
|
0,
|
|
1.2,
|
|
false,
|
|
"test_chain",
|
|
"hello",
|
|
sdk.NewCoins(sdk.NewCoin(sdk.DefaultBondDenom, sdk.NewInt(1))),
|
|
sdk.DecCoins{sdk.NewDecCoinFromDec(sdk.DefaultBondDenom, sdk.NewDecWithPrec(10000, sdk.Precision))},
|
|
)
|
|
}
|
|
|
|
func Test_splitAndCall_NoMessages(t *testing.T) {
|
|
ctx := context.CLIContext{}
|
|
txBldr := createFakeTxBuilder()
|
|
|
|
err := splitAndApply(nil, ctx, txBldr, nil, 10)
|
|
assert.NoError(t, err, "")
|
|
}
|
|
|
|
func Test_splitAndCall_Splitting(t *testing.T) {
|
|
ctx := context.CLIContext{}
|
|
txBldr := createFakeTxBuilder()
|
|
|
|
addr := sdk.AccAddress(secp256k1.GenPrivKey().PubKey().Address())
|
|
|
|
// Add five messages
|
|
msgs := []sdk.Msg{
|
|
sdk.NewTestMsg(addr),
|
|
sdk.NewTestMsg(addr),
|
|
sdk.NewTestMsg(addr),
|
|
sdk.NewTestMsg(addr),
|
|
sdk.NewTestMsg(addr),
|
|
}
|
|
|
|
// Keep track of number of calls
|
|
const chunkSize = 2
|
|
|
|
callCount := 0
|
|
err := splitAndApply(
|
|
func(ctx context.CLIContext, txBldr auth.TxBuilder, msgs []sdk.Msg) error {
|
|
callCount++
|
|
|
|
assert.NotNil(t, ctx)
|
|
assert.NotNil(t, txBldr)
|
|
assert.NotNil(t, msgs)
|
|
|
|
if callCount < 3 {
|
|
assert.Equal(t, len(msgs), 2)
|
|
} else {
|
|
assert.Equal(t, len(msgs), 1)
|
|
}
|
|
|
|
return nil
|
|
},
|
|
ctx, txBldr, msgs, chunkSize)
|
|
|
|
assert.NoError(t, err, "")
|
|
assert.Equal(t, 3, callCount)
|
|
}
|
|
|
|
func TestParseProposal(t *testing.T) {
|
|
cdc := codec.New()
|
|
okJSON, err := ioutil.TempFile("", "proposal")
|
|
require.Nil(t, err, "unexpected error")
|
|
okJSON.WriteString(`
|
|
{
|
|
"title": "Community Pool Spend",
|
|
"description": "Pay me some Atoms!",
|
|
"recipient": "cosmos1s5afhd6gxevu37mkqcvvsj8qeylhn0rz46zdlq",
|
|
"amount": "1000stake",
|
|
"deposit": "1000stake"
|
|
}
|
|
`)
|
|
|
|
proposal, err := ParseCommunityPoolSpendProposalJSON(cdc, okJSON.Name())
|
|
require.NoError(t, err)
|
|
|
|
addr, err := sdk.AccAddressFromBech32("cosmos1s5afhd6gxevu37mkqcvvsj8qeylhn0rz46zdlq")
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, "Community Pool Spend", proposal.Title)
|
|
require.Equal(t, "Pay me some Atoms!", proposal.Description)
|
|
require.Equal(t, addr, proposal.Recipient)
|
|
require.Equal(t, "1000stake", proposal.Deposit)
|
|
require.Equal(t, "1000stake", proposal.Amount)
|
|
}
|