certusone
/
cosmos-sdk
mirror of https://github.com/certusone/cosmos-sdk.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
cosmos-sdk/x/bank/keeper_test.go

210 lines
10 KiB
Go
Raw Normal View History

hard reset
2018-04-09 15:10:58 -07:00
package bank
import (
"testing"
"github.com/stretchr/testify/assert"
Switch asserts to require (#1483) * meta: Switch the majority of asserts to require Switch most assert statements to require, to ease debugging. Closes #1418 * Fix imports
2018-06-29 18:10:15 -07:00
"github.com/stretchr/testify/require"
hard reset
2018-04-09 15:10:58 -07:00
Ledger integration (#931) Merges the keybase and Ledger code from go-crypto (which is no more) into the SDK Adds support for Ledger into gaiacli Cherry-picks updated error handling from #1158
2018-06-28 17:54:47 -07:00
abci "github.com/tendermint/tendermint/abci/types"
update for tm v0.22.0. tmlibs->tendermint/libs
2018-07-02 13:34:06 -07:00
dbm "github.com/tendermint/tendermint/libs/db"
"github.com/tendermint/tendermint/libs/log"
hard reset
2018-04-09 15:10:58 -07:00
"github.com/cosmos/cosmos-sdk/store"
sdk "github.com/cosmos/cosmos-sdk/types"
rebase and switch test to amino
2018-04-17 08:07:55 -07:00
wire "github.com/cosmos/cosmos-sdk/wire"
hard reset
2018-04-09 15:10:58 -07:00
"github.com/cosmos/cosmos-sdk/x/auth"
)
func setupMultiStore() (sdk.MultiStore, *sdk.KVStoreKey) {
db := dbm.NewMemDB()
authKey := sdk.NewKVStoreKey("authkey")
ms := store.NewCommitMultiStore(db)
ms.MountStoreWithDB(authKey, sdk.StoreTypeIAVL, db)
ms.LoadLatestVersion()
return ms, authKey
}
many renames / golint compliance
2018-04-18 21:49:24 -07:00
func TestKeeper(t *testing.T) {
hard reset
2018-04-09 15:10:58 -07:00
ms, authKey := setupMultiStore()
rebase and switch test to amino
2018-04-17 08:07:55 -07:00
cdc := wire.NewCodec()
auth.RegisterBaseAccount(cdc)
hard reset
2018-04-09 15:10:58 -07:00
Merge PR #1331: Remove TxBytes from NewContext
2018-06-21 18:03:05 -07:00
ctx := sdk.NewContext(ms, abci.Header{}, false, log.NewNopLogger())
Merge pull request #1379: Account constructor in AccountMaper fix fix democoin fix tests pass lint last fix apply requests fix build failing fix docs
2018-06-25 19:05:47 -07:00
accountMapper := auth.NewAccountMapper(cdc, authKey, auth.ProtoBaseAccount)
Merge PR #2146: x/bank: Keeper Interfaces * Update banking module to implement and use keeper interfaces * Cleanup/consolidate x/bank keeper interface
2018-09-10 11:25:34 -07:00
bankKeeper := NewBaseKeeper(accountMapper)
hard reset
2018-04-09 15:10:58 -07:00
asdf
2018-07-06 00:06:53 -07:00
addr := sdk.AccAddress([]byte("addr1"))
addr2 := sdk.AccAddress([]byte("addr2"))
addr3 := sdk.AccAddress([]byte("addr3"))
hard reset
2018-04-09 15:10:58 -07:00
acc := accountMapper.NewAccountWithAddress(ctx, addr)
// Test GetCoins/SetCoins
accountMapper.SetAccount(ctx, acc)
Rename coinKeeper -> bankKeeper Closes #2267
2018-09-07 22:34:32 -07:00
require.True(t, bankKeeper.GetCoins(ctx, addr).IsEqual(sdk.Coins{}))
hard reset
2018-04-09 15:10:58 -07:00
Rename coinKeeper -> bankKeeper Closes #2267
2018-09-07 22:34:32 -07:00
bankKeeper.SetCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("foocoin", 10)})
require.True(t, bankKeeper.GetCoins(ctx, addr).IsEqual(sdk.Coins{sdk.NewInt64Coin("foocoin", 10)}))
hard reset
2018-04-09 15:10:58 -07:00
// Test HasCoins
Rename coinKeeper -> bankKeeper Closes #2267
2018-09-07 22:34:32 -07:00
require.True(t, bankKeeper.HasCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("foocoin", 10)}))
require.True(t, bankKeeper.HasCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("foocoin", 5)}))
require.False(t, bankKeeper.HasCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("foocoin", 15)}))
require.False(t, bankKeeper.HasCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("barcoin", 5)}))
hard reset
2018-04-09 15:10:58 -07:00
// Test AddCoins
Rename coinKeeper -> bankKeeper Closes #2267
2018-09-07 22:34:32 -07:00
bankKeeper.AddCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("foocoin", 15)})
require.True(t, bankKeeper.GetCoins(ctx, addr).IsEqual(sdk.Coins{sdk.NewInt64Coin("foocoin", 25)}))
hard reset
2018-04-09 15:10:58 -07:00
Rename coinKeeper -> bankKeeper Closes #2267
2018-09-07 22:34:32 -07:00
bankKeeper.AddCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("barcoin", 15)})
require.True(t, bankKeeper.GetCoins(ctx, addr).IsEqual(sdk.Coins{sdk.NewInt64Coin("barcoin", 15), sdk.NewInt64Coin("foocoin", 25)}))
hard reset
2018-04-09 15:10:58 -07:00
// Test SubtractCoins
Rename coinKeeper -> bankKeeper Closes #2267
2018-09-07 22:34:32 -07:00
bankKeeper.SubtractCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("foocoin", 10)})
bankKeeper.SubtractCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("barcoin", 5)})
require.True(t, bankKeeper.GetCoins(ctx, addr).IsEqual(sdk.Coins{sdk.NewInt64Coin("barcoin", 10), sdk.NewInt64Coin("foocoin", 15)}))
hard reset
2018-04-09 15:10:58 -07:00
Rename coinKeeper -> bankKeeper Closes #2267
2018-09-07 22:34:32 -07:00
bankKeeper.SubtractCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("barcoin", 11)})
require.True(t, bankKeeper.GetCoins(ctx, addr).IsEqual(sdk.Coins{sdk.NewInt64Coin("barcoin", 10), sdk.NewInt64Coin("foocoin", 15)}))
hard reset
2018-04-09 15:10:58 -07:00
Rename coinKeeper -> bankKeeper Closes #2267
2018-09-07 22:34:32 -07:00
bankKeeper.SubtractCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("barcoin", 10)})
require.True(t, bankKeeper.GetCoins(ctx, addr).IsEqual(sdk.Coins{sdk.NewInt64Coin("foocoin", 15)}))
require.False(t, bankKeeper.HasCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("barcoin", 1)}))
hard reset
2018-04-09 15:10:58 -07:00
// Test SendCoins
Rename coinKeeper -> bankKeeper Closes #2267
2018-09-07 22:34:32 -07:00
bankKeeper.SendCoins(ctx, addr, addr2, sdk.Coins{sdk.NewInt64Coin("foocoin", 5)})
require.True(t, bankKeeper.GetCoins(ctx, addr).IsEqual(sdk.Coins{sdk.NewInt64Coin("foocoin", 10)}))
require.True(t, bankKeeper.GetCoins(ctx, addr2).IsEqual(sdk.Coins{sdk.NewInt64Coin("foocoin", 5)}))
hard reset
2018-04-09 15:10:58 -07:00
Rename coinKeeper -> bankKeeper Closes #2267
2018-09-07 22:34:32 -07:00
_, err2 := bankKeeper.SendCoins(ctx, addr, addr2, sdk.Coins{sdk.NewInt64Coin("foocoin", 50)})
hard reset
2018-04-09 15:10:58 -07:00
assert.Implements(t, (*sdk.Error)(nil), err2)
Rename coinKeeper -> bankKeeper Closes #2267
2018-09-07 22:34:32 -07:00
require.True(t, bankKeeper.GetCoins(ctx, addr).IsEqual(sdk.Coins{sdk.NewInt64Coin("foocoin", 10)}))
require.True(t, bankKeeper.GetCoins(ctx, addr2).IsEqual(sdk.Coins{sdk.NewInt64Coin("foocoin", 5)}))
hard reset
2018-04-09 15:10:58 -07:00
Rename coinKeeper -> bankKeeper Closes #2267
2018-09-07 22:34:32 -07:00
bankKeeper.AddCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("barcoin", 30)})
bankKeeper.SendCoins(ctx, addr, addr2, sdk.Coins{sdk.NewInt64Coin("barcoin", 10), sdk.NewInt64Coin("foocoin", 5)})
require.True(t, bankKeeper.GetCoins(ctx, addr).IsEqual(sdk.Coins{sdk.NewInt64Coin("barcoin", 20), sdk.NewInt64Coin("foocoin", 5)}))
require.True(t, bankKeeper.GetCoins(ctx, addr2).IsEqual(sdk.Coins{sdk.NewInt64Coin("barcoin", 10), sdk.NewInt64Coin("foocoin", 10)}))
hard reset
2018-04-09 15:10:58 -07:00
// Test InputOutputCoins
Merge PR #1811: NewCoin takes sdk.Int
2018-07-30 17:09:50 -07:00
input1 := NewInput(addr2, sdk.Coins{sdk.NewInt64Coin("foocoin", 2)})
output1 := NewOutput(addr, sdk.Coins{sdk.NewInt64Coin("foocoin", 2)})
Rename coinKeeper -> bankKeeper Closes #2267
2018-09-07 22:34:32 -07:00
bankKeeper.InputOutputCoins(ctx, []Input{input1}, []Output{output1})
require.True(t, bankKeeper.GetCoins(ctx, addr).IsEqual(sdk.Coins{sdk.NewInt64Coin("barcoin", 20), sdk.NewInt64Coin("foocoin", 7)}))
require.True(t, bankKeeper.GetCoins(ctx, addr2).IsEqual(sdk.Coins{sdk.NewInt64Coin("barcoin", 10), sdk.NewInt64Coin("foocoin", 8)}))
hard reset
2018-04-09 15:10:58 -07:00
inputs := []Input{
Merge PR #1811: NewCoin takes sdk.Int
2018-07-30 17:09:50 -07:00
NewInput(addr, sdk.Coins{sdk.NewInt64Coin("foocoin", 3)}),
NewInput(addr2, sdk.Coins{sdk.NewInt64Coin("barcoin", 3), sdk.NewInt64Coin("foocoin", 2)}),
hard reset
2018-04-09 15:10:58 -07:00
}
outputs := []Output{
Merge PR #1811: NewCoin takes sdk.Int
2018-07-30 17:09:50 -07:00
NewOutput(addr, sdk.Coins{sdk.NewInt64Coin("barcoin", 1)}),
NewOutput(addr3, sdk.Coins{sdk.NewInt64Coin("barcoin", 2), sdk.NewInt64Coin("foocoin", 5)}),
hard reset
2018-04-09 15:10:58 -07:00
}
Rename coinKeeper -> bankKeeper Closes #2267
2018-09-07 22:34:32 -07:00
bankKeeper.InputOutputCoins(ctx, inputs, outputs)
require.True(t, bankKeeper.GetCoins(ctx, addr).IsEqual(sdk.Coins{sdk.NewInt64Coin("barcoin", 21), sdk.NewInt64Coin("foocoin", 4)}))
require.True(t, bankKeeper.GetCoins(ctx, addr2).IsEqual(sdk.Coins{sdk.NewInt64Coin("barcoin", 7), sdk.NewInt64Coin("foocoin", 6)}))
require.True(t, bankKeeper.GetCoins(ctx, addr3).IsEqual(sdk.Coins{sdk.NewInt64Coin("barcoin", 2), sdk.NewInt64Coin("foocoin", 5)}))
hard reset
2018-04-09 15:10:58 -07:00
}
func TestSendKeeper(t *testing.T) {
ms, authKey := setupMultiStore()
rebase and switch test to amino
2018-04-17 08:07:55 -07:00
cdc := wire.NewCodec()
auth.RegisterBaseAccount(cdc)
hard reset
2018-04-09 15:10:58 -07:00
Merge PR #1331: Remove TxBytes from NewContext
2018-06-21 18:03:05 -07:00
ctx := sdk.NewContext(ms, abci.Header{}, false, log.NewNopLogger())
Merge pull request #1379: Account constructor in AccountMaper fix fix democoin fix tests pass lint last fix apply requests fix build failing fix docs
2018-06-25 19:05:47 -07:00
accountMapper := auth.NewAccountMapper(cdc, authKey, auth.ProtoBaseAccount)
Merge PR #2146: x/bank: Keeper Interfaces * Update banking module to implement and use keeper interfaces * Cleanup/consolidate x/bank keeper interface
2018-09-10 11:25:34 -07:00
bankKeeper := NewBaseKeeper(accountMapper)
sendKeeper := NewBaseSendKeeper(accountMapper)
hard reset
2018-04-09 15:10:58 -07:00
asdf
2018-07-06 00:06:53 -07:00
addr := sdk.AccAddress([]byte("addr1"))
addr2 := sdk.AccAddress([]byte("addr2"))
addr3 := sdk.AccAddress([]byte("addr3"))
hard reset
2018-04-09 15:10:58 -07:00
acc := accountMapper.NewAccountWithAddress(ctx, addr)
// Test GetCoins/SetCoins
accountMapper.SetAccount(ctx, acc)
Switch asserts to require (#1483) * meta: Switch the majority of asserts to require Switch most assert statements to require, to ease debugging. Closes #1418 * Fix imports
2018-06-29 18:10:15 -07:00
require.True(t, sendKeeper.GetCoins(ctx, addr).IsEqual(sdk.Coins{}))
hard reset
2018-04-09 15:10:58 -07:00
Rename coinKeeper -> bankKeeper Closes #2267
2018-09-07 22:34:32 -07:00
bankKeeper.SetCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("foocoin", 10)})
Merge PR #1811: NewCoin takes sdk.Int
2018-07-30 17:09:50 -07:00
require.True(t, sendKeeper.GetCoins(ctx, addr).IsEqual(sdk.Coins{sdk.NewInt64Coin("foocoin", 10)}))
hard reset
2018-04-09 15:10:58 -07:00
// Test HasCoins
Merge PR #1811: NewCoin takes sdk.Int
2018-07-30 17:09:50 -07:00
require.True(t, sendKeeper.HasCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("foocoin", 10)}))
require.True(t, sendKeeper.HasCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("foocoin", 5)}))
require.False(t, sendKeeper.HasCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("foocoin", 15)}))
require.False(t, sendKeeper.HasCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("barcoin", 5)}))
hard reset
2018-04-09 15:10:58 -07:00
Rename coinKeeper -> bankKeeper Closes #2267
2018-09-07 22:34:32 -07:00
bankKeeper.SetCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("foocoin", 15)})
hard reset
2018-04-09 15:10:58 -07:00
// Test SendCoins
Merge PR #1811: NewCoin takes sdk.Int
2018-07-30 17:09:50 -07:00
sendKeeper.SendCoins(ctx, addr, addr2, sdk.Coins{sdk.NewInt64Coin("foocoin", 5)})
require.True(t, sendKeeper.GetCoins(ctx, addr).IsEqual(sdk.Coins{sdk.NewInt64Coin("foocoin", 10)}))
require.True(t, sendKeeper.GetCoins(ctx, addr2).IsEqual(sdk.Coins{sdk.NewInt64Coin("foocoin", 5)}))
hard reset
2018-04-09 15:10:58 -07:00
Merge PR #1811: NewCoin takes sdk.Int
2018-07-30 17:09:50 -07:00
_, err2 := sendKeeper.SendCoins(ctx, addr, addr2, sdk.Coins{sdk.NewInt64Coin("foocoin", 50)})
hard reset
2018-04-09 15:10:58 -07:00
assert.Implements(t, (*sdk.Error)(nil), err2)
Merge PR #1811: NewCoin takes sdk.Int
2018-07-30 17:09:50 -07:00
require.True(t, sendKeeper.GetCoins(ctx, addr).IsEqual(sdk.Coins{sdk.NewInt64Coin("foocoin", 10)}))
require.True(t, sendKeeper.GetCoins(ctx, addr2).IsEqual(sdk.Coins{sdk.NewInt64Coin("foocoin", 5)}))
hard reset
2018-04-09 15:10:58 -07:00
Rename coinKeeper -> bankKeeper Closes #2267
2018-09-07 22:34:32 -07:00
bankKeeper.AddCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("barcoin", 30)})
Merge PR #1811: NewCoin takes sdk.Int
2018-07-30 17:09:50 -07:00
sendKeeper.SendCoins(ctx, addr, addr2, sdk.Coins{sdk.NewInt64Coin("barcoin", 10), sdk.NewInt64Coin("foocoin", 5)})
require.True(t, sendKeeper.GetCoins(ctx, addr).IsEqual(sdk.Coins{sdk.NewInt64Coin("barcoin", 20), sdk.NewInt64Coin("foocoin", 5)}))
require.True(t, sendKeeper.GetCoins(ctx, addr2).IsEqual(sdk.Coins{sdk.NewInt64Coin("barcoin", 10), sdk.NewInt64Coin("foocoin", 10)}))
hard reset
2018-04-09 15:10:58 -07:00
// Test InputOutputCoins
Merge PR #1811: NewCoin takes sdk.Int
2018-07-30 17:09:50 -07:00
input1 := NewInput(addr2, sdk.Coins{sdk.NewInt64Coin("foocoin", 2)})
output1 := NewOutput(addr, sdk.Coins{sdk.NewInt64Coin("foocoin", 2)})
hard reset
2018-04-09 15:10:58 -07:00
sendKeeper.InputOutputCoins(ctx, []Input{input1}, []Output{output1})
Merge PR #1811: NewCoin takes sdk.Int
2018-07-30 17:09:50 -07:00
require.True(t, sendKeeper.GetCoins(ctx, addr).IsEqual(sdk.Coins{sdk.NewInt64Coin("barcoin", 20), sdk.NewInt64Coin("foocoin", 7)}))
require.True(t, sendKeeper.GetCoins(ctx, addr2).IsEqual(sdk.Coins{sdk.NewInt64Coin("barcoin", 10), sdk.NewInt64Coin("foocoin", 8)}))
hard reset
2018-04-09 15:10:58 -07:00
inputs := []Input{
Merge PR #1811: NewCoin takes sdk.Int
2018-07-30 17:09:50 -07:00
NewInput(addr, sdk.Coins{sdk.NewInt64Coin("foocoin", 3)}),
NewInput(addr2, sdk.Coins{sdk.NewInt64Coin("barcoin", 3), sdk.NewInt64Coin("foocoin", 2)}),
hard reset
2018-04-09 15:10:58 -07:00
}
outputs := []Output{
Merge PR #1811: NewCoin takes sdk.Int
2018-07-30 17:09:50 -07:00
NewOutput(addr, sdk.Coins{sdk.NewInt64Coin("barcoin", 1)}),
NewOutput(addr3, sdk.Coins{sdk.NewInt64Coin("barcoin", 2), sdk.NewInt64Coin("foocoin", 5)}),
hard reset
2018-04-09 15:10:58 -07:00
}
sendKeeper.InputOutputCoins(ctx, inputs, outputs)
Merge PR #1811: NewCoin takes sdk.Int
2018-07-30 17:09:50 -07:00
require.True(t, sendKeeper.GetCoins(ctx, addr).IsEqual(sdk.Coins{sdk.NewInt64Coin("barcoin", 21), sdk.NewInt64Coin("foocoin", 4)}))
require.True(t, sendKeeper.GetCoins(ctx, addr2).IsEqual(sdk.Coins{sdk.NewInt64Coin("barcoin", 7), sdk.NewInt64Coin("foocoin", 6)}))
require.True(t, sendKeeper.GetCoins(ctx, addr3).IsEqual(sdk.Coins{sdk.NewInt64Coin("barcoin", 2), sdk.NewInt64Coin("foocoin", 5)}))
hard reset
2018-04-09 15:10:58 -07:00
}
func TestViewKeeper(t *testing.T) {
ms, authKey := setupMultiStore()
rebase and switch test to amino
2018-04-17 08:07:55 -07:00
cdc := wire.NewCodec()
auth.RegisterBaseAccount(cdc)
hard reset
2018-04-09 15:10:58 -07:00
Merge PR #1331: Remove TxBytes from NewContext
2018-06-21 18:03:05 -07:00
ctx := sdk.NewContext(ms, abci.Header{}, false, log.NewNopLogger())
Merge pull request #1379: Account constructor in AccountMaper fix fix democoin fix tests pass lint last fix apply requests fix build failing fix docs
2018-06-25 19:05:47 -07:00
accountMapper := auth.NewAccountMapper(cdc, authKey, auth.ProtoBaseAccount)
Merge PR #2146: x/bank: Keeper Interfaces * Update banking module to implement and use keeper interfaces * Cleanup/consolidate x/bank keeper interface
2018-09-10 11:25:34 -07:00
bankKeeper := NewBaseKeeper(accountMapper)
viewKeeper := NewBaseViewKeeper(accountMapper)
hard reset
2018-04-09 15:10:58 -07:00
asdf
2018-07-06 00:06:53 -07:00
addr := sdk.AccAddress([]byte("addr1"))
hard reset
2018-04-09 15:10:58 -07:00
acc := accountMapper.NewAccountWithAddress(ctx, addr)
// Test GetCoins/SetCoins
accountMapper.SetAccount(ctx, acc)
Switch asserts to require (#1483) * meta: Switch the majority of asserts to require Switch most assert statements to require, to ease debugging. Closes #1418 * Fix imports
2018-06-29 18:10:15 -07:00
require.True(t, viewKeeper.GetCoins(ctx, addr).IsEqual(sdk.Coins{}))
hard reset
2018-04-09 15:10:58 -07:00
Rename coinKeeper -> bankKeeper Closes #2267
2018-09-07 22:34:32 -07:00
bankKeeper.SetCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("foocoin", 10)})
Merge PR #1811: NewCoin takes sdk.Int
2018-07-30 17:09:50 -07:00
require.True(t, viewKeeper.GetCoins(ctx, addr).IsEqual(sdk.Coins{sdk.NewInt64Coin("foocoin", 10)}))
hard reset
2018-04-09 15:10:58 -07:00
// Test HasCoins
Merge PR #1811: NewCoin takes sdk.Int
2018-07-30 17:09:50 -07:00
require.True(t, viewKeeper.HasCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("foocoin", 10)}))
require.True(t, viewKeeper.HasCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("foocoin", 5)}))
require.False(t, viewKeeper.HasCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("foocoin", 15)}))
require.False(t, viewKeeper.HasCoins(ctx, addr, sdk.Coins{sdk.NewInt64Coin("barcoin", 5)}))
hard reset
2018-04-09 15:10:58 -07:00
}