Add statedb bloom filter test (#160)

x/evm/types/statedb_test.go

@@ -0,0 +1,101 @@
+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"
+
+	"github.com/ethereum/go-ethereum/common"
+	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, common.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))))
+}