101 lines
2.8 KiB
Go
101 lines
2.8 KiB
Go
package types
|
|
|
|
import (
|
|
"math/big"
|
|
"testing"
|
|
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
|
"github.com/cosmos/cosmos-sdk/store"
|
|
sdkstore "github.com/cosmos/cosmos-sdk/store/types"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/auth"
|
|
"github.com/cosmos/cosmos-sdk/x/params"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
ethcmn "github.com/ethereum/go-ethereum/common"
|
|
ethtypes "github.com/ethereum/go-ethereum/core/types"
|
|
|
|
"github.com/cosmos/ethermint/types"
|
|
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
tmlog "github.com/tendermint/tendermint/libs/log"
|
|
dbm "github.com/tendermint/tm-db"
|
|
)
|
|
|
|
func newTestCodec() *codec.Codec {
|
|
cdc := codec.New()
|
|
|
|
RegisterCodec(cdc)
|
|
types.RegisterCodec(cdc)
|
|
auth.RegisterCodec(cdc)
|
|
sdk.RegisterCodec(cdc)
|
|
codec.RegisterCrypto(cdc)
|
|
|
|
return cdc
|
|
}
|
|
|
|
func setupStateDB() (*CommitStateDB, error) {
|
|
accKey := sdk.NewKVStoreKey("acc")
|
|
storageKey := sdk.NewKVStoreKey(EvmStoreKey)
|
|
codeKey := sdk.NewKVStoreKey(EvmCodeKey)
|
|
logger := tmlog.NewNopLogger()
|
|
|
|
db := dbm.NewMemDB()
|
|
|
|
// create logger, codec and root multi-store
|
|
cdc := newTestCodec()
|
|
cms := store.NewCommitMultiStore(db)
|
|
|
|
// The ParamsKeeper handles parameter storage for the application
|
|
keyParams := sdk.NewKVStoreKey(params.StoreKey)
|
|
tkeyParams := sdk.NewTransientStoreKey(params.TStoreKey)
|
|
paramsKeeper := params.NewKeeper(cdc, keyParams, tkeyParams, params.DefaultCodespace)
|
|
// Set specific supspaces
|
|
authSubspace := paramsKeeper.Subspace(auth.DefaultParamspace)
|
|
ak := auth.NewAccountKeeper(cdc, accKey, authSubspace, types.ProtoBaseAccount)
|
|
|
|
// mount stores
|
|
keys := []*sdk.KVStoreKey{accKey, storageKey, codeKey}
|
|
for _, key := range keys {
|
|
cms.MountStoreWithDB(key, sdk.StoreTypeIAVL, nil)
|
|
}
|
|
|
|
cms.SetPruning(sdkstore.PruneNothing)
|
|
|
|
// load latest version (root)
|
|
if err := cms.LoadLatestVersion(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
ms := cms.CacheMultiStore()
|
|
ctx := sdk.NewContext(ms, abci.Header{}, false, logger)
|
|
return NewCommitStateDB(ctx, ak, storageKey, codeKey), nil
|
|
}
|
|
|
|
func TestBloomFilter(t *testing.T) {
|
|
stateDB, err := setupStateDB()
|
|
require.NoError(t, err)
|
|
|
|
// Prepare db for logs
|
|
tHash := ethcmn.BytesToHash([]byte{0x1})
|
|
stateDB.Prepare(tHash, ethcmn.Hash{}, 0)
|
|
|
|
contractAddress := ethcmn.BigToAddress(big.NewInt(1))
|
|
|
|
// Generate and add a log to test
|
|
log := ethtypes.Log{Address: contractAddress}
|
|
stateDB.AddLog(&log)
|
|
|
|
// Get log from db
|
|
logs := stateDB.GetLogs(tHash)
|
|
require.Equal(t, len(logs), 1)
|
|
|
|
// get logs bloom from the log
|
|
bloomInt := ethtypes.LogsBloom(logs)
|
|
bloomFilter := ethtypes.BytesToBloom(bloomInt.Bytes())
|
|
|
|
// Check to make sure bloom filter will succeed on
|
|
require.True(t, ethtypes.BloomLookup(bloomFilter, contractAddress))
|
|
require.False(t, ethtypes.BloomLookup(bloomFilter, ethcmn.BigToAddress(big.NewInt(2))))
|
|
}
|