b0e82f9640
* mainly sdk.int to cosmossdk.io/math * staking keys * fumpt * var-naming linter errors and a fumpt * Update CHANGELOG.md * Update .golangci.yml * Update CHANGELOG.md * Update test_helpers.go * Update test_helpers.go * fumpt and lint * this lints the db module, and makes it easier to use. It adds breaking name changes * DBConnection -> Connection * previous commit contained a merge error * Update test_helpers.go * Update test_helpers.go * db renamings * merge master * changelog * DBWriter -> Writer * consistent multistore reciever * standard recievers for multistore v2alpha1 * general cleanup of linting issues * more linter fixes * remove prealloc linter * nolint the secp256k1 import * nolint the secp256k1 package * completenolint resulting in a diff that has only nolints
129 lines
3.4 KiB
Go
129 lines
3.4 KiB
Go
package mock
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"path/filepath"
|
|
|
|
"github.com/tendermint/tendermint/types"
|
|
db "github.com/tendermint/tm-db"
|
|
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
"github.com/tendermint/tendermint/libs/log"
|
|
|
|
bam "github.com/cosmos/cosmos-sdk/baseapp"
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
|
storetypes "github.com/cosmos/cosmos-sdk/store/types"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
)
|
|
|
|
// NewApp creates a simple mock kvstore app for testing. It should work
|
|
// similar to a real app. Make sure rootDir is empty before running the test,
|
|
// in order to guarantee consistent results
|
|
func NewApp(rootDir string, logger log.Logger) (abci.Application, error) {
|
|
db, err := db.NewGoLevelDB("mock", filepath.Join(rootDir, "data"))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Capabilities key to access the main KVStore.
|
|
capKeyMainStore := sdk.NewKVStoreKey("main")
|
|
|
|
// Create BaseApp.
|
|
baseApp := bam.NewBaseApp("kvstore", logger, db, decodeTx)
|
|
|
|
// Set mounts for BaseApp's MultiStore.
|
|
baseApp.MountStores(capKeyMainStore)
|
|
|
|
baseApp.SetInitChainer(InitChainer(capKeyMainStore))
|
|
|
|
// Set a Route.
|
|
baseApp.Router().AddRoute(sdk.NewRoute("kvstore", KVStoreHandler(capKeyMainStore)))
|
|
|
|
// Load latest version.
|
|
if err := baseApp.LoadLatestVersion(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return baseApp, nil
|
|
}
|
|
|
|
// KVStoreHandler is a simple handler that takes kvstoreTx and writes
|
|
// them to the db
|
|
func KVStoreHandler(storeKey storetypes.StoreKey) sdk.Handler {
|
|
return func(ctx sdk.Context, msg sdk.Msg) (*sdk.Result, error) {
|
|
dTx, ok := msg.(kvstoreTx)
|
|
if !ok {
|
|
return nil, errors.New("KVStoreHandler should only receive kvstoreTx")
|
|
}
|
|
|
|
// tx is already unmarshalled
|
|
key := dTx.key
|
|
value := dTx.value
|
|
|
|
store := ctx.KVStore(storeKey)
|
|
store.Set(key, value)
|
|
|
|
return &sdk.Result{
|
|
Log: fmt.Sprintf("set %s=%s", key, value),
|
|
}, nil
|
|
}
|
|
}
|
|
|
|
// basic KV structure
|
|
type KV struct {
|
|
Key string `json:"key"`
|
|
Value string `json:"value"`
|
|
}
|
|
|
|
// What Genesis JSON is formatted as
|
|
type GenesisJSON struct {
|
|
Values []KV `json:"values"`
|
|
}
|
|
|
|
// InitChainer returns a function that can initialize the chain
|
|
// with key/value pairs
|
|
func InitChainer(key storetypes.StoreKey) func(sdk.Context, abci.RequestInitChain) abci.ResponseInitChain {
|
|
return func(ctx sdk.Context, req abci.RequestInitChain) abci.ResponseInitChain {
|
|
stateJSON := req.AppStateBytes
|
|
|
|
genesisState := new(GenesisJSON)
|
|
err := json.Unmarshal(stateJSON, genesisState)
|
|
if err != nil {
|
|
panic(err) // TODO https://github.com/cosmos/cosmos-sdk/issues/468
|
|
// return sdk.ErrGenesisParse("").TraceCause(err, "")
|
|
}
|
|
|
|
for _, val := range genesisState.Values {
|
|
store := ctx.KVStore(key)
|
|
store.Set([]byte(val.Key), []byte(val.Value))
|
|
}
|
|
return abci.ResponseInitChain{}
|
|
}
|
|
}
|
|
|
|
// AppGenState can be passed into InitCmd, returns a static string of a few
|
|
// key-values that can be parsed by InitChainer
|
|
func AppGenState(_ *codec.LegacyAmino, _ types.GenesisDoc, _ []json.RawMessage) (appState json.RawMessage, err error) {
|
|
appState = json.RawMessage(`{
|
|
"values": [
|
|
{
|
|
"key": "hello",
|
|
"value": "goodbye"
|
|
},
|
|
{
|
|
"key": "foo",
|
|
"value": "bar"
|
|
}
|
|
]
|
|
}`)
|
|
return
|
|
}
|
|
|
|
// AppGenStateEmpty returns an empty transaction state for mocking.
|
|
func AppGenStateEmpty(_ *codec.LegacyAmino, _ types.GenesisDoc, _ []json.RawMessage) (appState json.RawMessage, err error) {
|
|
appState = json.RawMessage(``)
|
|
return
|
|
}
|