add test

jsonrpc/backend/filters_test.go

@@ -1,14 +1,18 @@
 package backend_test
 
 import (
+	"context"
 	"math/big"
 	"testing"
+	"time"
 
 	"github.com/ethereum/go-ethereum/common"
 	"github.com/ethereum/go-ethereum/common/hexutil"
+	"github.com/ethereum/go-ethereum/core/bloombits"
 	coretypes "github.com/ethereum/go-ethereum/core/types"
 	"github.com/ethereum/go-ethereum/rpc"
 	"github.com/initia-labs/minievm/tests"
+	evmconfig "github.com/initia-labs/minievm/x/evm/config"
 	evmtypes "github.com/initia-labs/minievm/x/evm/types"
 	"github.com/stretchr/testify/require"
 )
@@ -51,3 +55,87 @@ func Test_GetLogsByHeight(t *testing.T) {
 	blockLogs = append(blockLogs, logs1...)
 	require.Equal(t, blockLogs, logs)
 }
+
+func Test_BloomStatus(t *testing.T) {
+	input := setupBackend(t)
+	app, indexer, backend, addrs, privKeys := input.app, input.indexer, input.backend, input.addrs, input.privKeys
+
+	tx1, _ := tests.GenerateCreateERC20Tx(t, app, privKeys[0])
+	_, finalizeRes1 := tests.ExecuteTxs(t, app, tx1)
+	tests.CheckTxResult(t, finalizeRes1.TxResults[0], true)
+	height1 := app.LastBlockHeight()
+
+	for i := uint64(0); i < evmconfig.SectionSize; i++ {
+		tests.IncreaseBlockHeight(t, app)
+	}
+
+	// wait for bloom indexing
+	for {
+		if indexer.IsBloomIndexingRunning() {
+			time.Sleep(100 * time.Millisecond)
+		} else {
+			break
+		}
+	}
+
+	tx2, _ := tests.GenerateCreateERC20Tx(t, app, privKeys[1])
+	_, finalizeRes2 := tests.ExecuteTxs(t, app, tx2)
+	tests.CheckTxResult(t, finalizeRes2.TxResults[0], true)
+	height2 := app.LastBlockHeight()
+
+	size, section, err := backend.BloomStatus()
+	require.NoError(t, err)
+	require.Equal(t, evmconfig.SectionSize, size)
+	require.Equal(t, uint64(1), section)
+
+	filters := make([][][]byte, 1)
+	filters[0] = [][]byte{addrs[0].Bytes()}
+	matcher := bloombits.NewMatcher(evmconfig.SectionSize, filters)
+
+	// start find matches
+	matches := make(chan uint64, 64)
+	session1, err := matcher.Start(context.Background(), uint64(1), uint64(height2), matches)
+	require.NoError(t, err)
+	defer session1.Close()
+
+	backend.ServiceFilter(session1)
+
+LOOP1:
+	for {
+		number, ok := <-matches
+		if !ok {
+			err := session1.Error()
+			require.NoError(t, err)
+			break LOOP1
+		}
+
+		require.True(t, ok)
+		require.Equal(t, height1, number)
+	}
+
+	filters = make([][][]byte, 1)
+	filters[0] = [][]byte{addrs[1].Bytes()}
+	matcher2 := bloombits.NewMatcher(evmconfig.SectionSize, filters)
+
+	// start find matches
+	matches = make(chan uint64, 64)
+	session2, err := matcher2.Start(context.Background(), uint64(1), uint64(height2), matches)
+	require.NoError(t, err)
+	defer session2.Close()
+
+	backend.ServiceFilter(session2)
+
+LOOP2:
+	for {
+		number, ok := <-matches
+		if !ok {
+			err := session2.Error()
+			require.NoError(t, err)
+
+			break LOOP2
+		}
+
+		require.True(t, ok)
+		require.Equal(t, height2, number)
+	}
+}