mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-10 18:15:19 +00:00
78 lines
2.0 KiB
Go
78 lines
2.0 KiB
Go
package app
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/wire"
|
|
"github.com/cosmos/cosmos-sdk/x/auth"
|
|
"github.com/cosmos/cosmos-sdk/x/stake"
|
|
"github.com/kava-labs/kava/internal/types"
|
|
|
|
abci "github.com/tendermint/abci/types"
|
|
crypto "github.com/tendermint/go-crypto"
|
|
dbm "github.com/tendermint/tmlibs/db"
|
|
"github.com/tendermint/tmlibs/log"
|
|
)
|
|
|
|
func setGenesis(bapp *KavaApp, accs ...auth.BaseAccount) error {
|
|
genaccs := make([]*types.GenesisAccount, len(accs))
|
|
for i, acc := range accs {
|
|
genaccs[i] = types.NewGenesisAccount(&types.AppAccount{acc, "foobart"})
|
|
}
|
|
|
|
genesisState := types.GenesisState{
|
|
Accounts: genaccs,
|
|
StakeData: stake.DefaultGenesisState(),
|
|
}
|
|
|
|
stateBytes, err := wire.MarshalJSONIndent(bapp.cdc, genesisState)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Initialize the chain
|
|
vals := []abci.Validator{}
|
|
bapp.InitChain(abci.RequestInitChain{Validators: vals, AppStateBytes: stateBytes})
|
|
bapp.Commit()
|
|
|
|
return nil
|
|
}
|
|
|
|
//_______________________________________________________________________
|
|
|
|
func TestGenesis(t *testing.T) {
|
|
logger := log.NewTMLogger(log.NewSyncWriter(os.Stdout)).With("module", "sdk/app")
|
|
db := dbm.NewMemDB()
|
|
bapp := NewKavaApp(logger, db)
|
|
|
|
// Construct some genesis bytes to reflect basecoin/types/AppAccount
|
|
pk := crypto.GenPrivKeyEd25519().PubKey()
|
|
addr := pk.Address()
|
|
coins, err := sdk.ParseCoins("77foocoin,99barcoin")
|
|
require.Nil(t, err)
|
|
baseAcc := auth.BaseAccount{
|
|
Address: addr,
|
|
Coins: coins,
|
|
}
|
|
acc := &types.AppAccount{baseAcc, "foobart"}
|
|
|
|
err = setGenesis(bapp, baseAcc)
|
|
require.Nil(t, err)
|
|
|
|
// A checkTx context
|
|
ctx := bapp.BaseApp.NewContext(true, abci.Header{})
|
|
res1 := bapp.accountMapper.GetAccount(ctx, baseAcc.Address)
|
|
assert.Equal(t, acc, res1)
|
|
|
|
// reload app and ensure the account is still there
|
|
bapp = NewKavaApp(logger, db)
|
|
ctx = bapp.BaseApp.NewContext(true, abci.Header{})
|
|
res1 = bapp.accountMapper.GetAccount(ctx, baseAcc.Address)
|
|
assert.Equal(t, acc, res1)
|
|
}
|