addcfbf5cb
* Update docs/sdk/clients.md * organize ADR directory like tendermint * docs: move spec-proposals into spec/ * remove lotion, moved to website repo * move getting-started to cosmos-hub, and voyager to website * docs: move lite/ into clients/lite/ * move introduction/ content to website repo * move resources/ content to website repo * mv sdk/clients.md to clients/clients.md * mv validators to cosmos-hub/validators * move deprecated sdk/ content to _attic * sdk/modules.md is duplicate with modules/README.md * consolidate remianing sdk/ files into a single sdk.md * move examples/ to docs/examples/ * mv docs/cosmos-hub to docs/gaia * Add keys/accounts section to localnet docs
92 lines
2.7 KiB
Go
92 lines
2.7 KiB
Go
package simplestake
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
"github.com/tendermint/tendermint/crypto/ed25519"
|
|
dbm "github.com/tendermint/tendermint/libs/db"
|
|
"github.com/tendermint/tendermint/libs/log"
|
|
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
|
"github.com/cosmos/cosmos-sdk/store"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/auth"
|
|
"github.com/cosmos/cosmos-sdk/x/bank"
|
|
)
|
|
|
|
func setupMultiStore() (sdk.MultiStore, *sdk.KVStoreKey, *sdk.KVStoreKey) {
|
|
db := dbm.NewMemDB()
|
|
authKey := sdk.NewKVStoreKey("authkey")
|
|
capKey := sdk.NewKVStoreKey("capkey")
|
|
ms := store.NewCommitMultiStore(db)
|
|
ms.MountStoreWithDB(capKey, sdk.StoreTypeIAVL, db)
|
|
ms.MountStoreWithDB(authKey, sdk.StoreTypeIAVL, db)
|
|
ms.LoadLatestVersion()
|
|
return ms, authKey, capKey
|
|
}
|
|
|
|
func TestKeeperGetSet(t *testing.T) {
|
|
ms, authKey, capKey := setupMultiStore()
|
|
cdc := codec.New()
|
|
auth.RegisterBaseAccount(cdc)
|
|
|
|
accountKeeper := auth.NewAccountKeeper(cdc, authKey, auth.ProtoBaseAccount)
|
|
stakeKeeper := NewKeeper(capKey, bank.NewBaseKeeper(accountKeeper), DefaultCodespace)
|
|
ctx := sdk.NewContext(ms, abci.Header{}, false, log.NewNopLogger())
|
|
addr := sdk.AccAddress([]byte("some-address"))
|
|
|
|
bi := stakeKeeper.getBondInfo(ctx, addr)
|
|
require.Equal(t, bi, bondInfo{})
|
|
|
|
privKey := ed25519.GenPrivKey()
|
|
|
|
bi = bondInfo{
|
|
PubKey: privKey.PubKey(),
|
|
Power: int64(10),
|
|
}
|
|
fmt.Printf("Pubkey: %v\n", privKey.PubKey())
|
|
stakeKeeper.setBondInfo(ctx, addr, bi)
|
|
|
|
savedBi := stakeKeeper.getBondInfo(ctx, addr)
|
|
require.NotNil(t, savedBi)
|
|
fmt.Printf("Bond Info: %v\n", savedBi)
|
|
require.Equal(t, int64(10), savedBi.Power)
|
|
}
|
|
|
|
func TestBonding(t *testing.T) {
|
|
ms, authKey, capKey := setupMultiStore()
|
|
cdc := codec.New()
|
|
auth.RegisterBaseAccount(cdc)
|
|
|
|
ctx := sdk.NewContext(ms, abci.Header{}, false, log.NewNopLogger())
|
|
|
|
accountKeeper := auth.NewAccountKeeper(cdc, authKey, auth.ProtoBaseAccount)
|
|
bankKeeper := bank.NewBaseKeeper(accountKeeper)
|
|
stakeKeeper := NewKeeper(capKey, bankKeeper, DefaultCodespace)
|
|
addr := sdk.AccAddress([]byte("some-address"))
|
|
privKey := ed25519.GenPrivKey()
|
|
pubKey := privKey.PubKey()
|
|
|
|
_, _, err := stakeKeeper.unbondWithoutCoins(ctx, addr)
|
|
require.Equal(t, err, ErrInvalidUnbond(DefaultCodespace))
|
|
|
|
_, err = stakeKeeper.bondWithoutCoins(ctx, addr, pubKey, sdk.NewInt64Coin("stake", 10))
|
|
require.Nil(t, err)
|
|
|
|
power, err := stakeKeeper.bondWithoutCoins(ctx, addr, pubKey, sdk.NewInt64Coin("stake", 10))
|
|
require.Nil(t, err)
|
|
require.Equal(t, int64(20), power)
|
|
|
|
pk, _, err := stakeKeeper.unbondWithoutCoins(ctx, addr)
|
|
require.Nil(t, err)
|
|
require.Equal(t, pubKey, pk)
|
|
|
|
_, _, err = stakeKeeper.unbondWithoutCoins(ctx, addr)
|
|
require.Equal(t, err, ErrInvalidUnbond(DefaultCodespace))
|
|
}
|