fix : lint

This commit is contained in:
Shivam Sharma 2022-07-19 14:39:26 +05:30
parent 8a6c16dcde
commit 58ef6fdbb7
3 changed files with 32 additions and 8 deletions

View file

@ -122,6 +122,7 @@ func benchmarkBloomBits(b *testing.B, sectionSize uint64) {
b.Log("Running filter benchmarks...")
start = time.Now()
var backend *TestBackend
for i := 0; i < benchFilterCnt; i++ {

View file

@ -10,6 +10,7 @@ import (
types "github.com/ethereum/go-ethereum/core/types"
"github.com/ethereum/go-ethereum/crypto"
"github.com/ethereum/go-ethereum/params"
gomock "github.com/golang/mock/gomock"
)
@ -27,6 +28,7 @@ func newTestHeader(blockNumber uint) *types.Header {
head := types.Header{
Number: big.NewInt(int64(blockNumber)),
}
return &head
}
@ -40,6 +42,7 @@ func newTestReceipt(contractAddr common.Address, topicAddress common.Hash) *type
}
receipt.Bloom = types.CreateBloom(types.Receipts{receipt})
return receipt
}
@ -49,6 +52,7 @@ func (backend *MockBackend) expectBorReceiptsFromMock(hashes []*common.Hash) {
backend.EXPECT().GetBorBlockReceipt(gomock.Any(), gomock.Any()).Return(nil, nil)
continue
}
backend.EXPECT().GetBorBlockReceipt(gomock.Any(), gomock.Any()).Return(newTestReceipt(addr, *hashes[i]), nil)
}
}
@ -78,9 +82,11 @@ func TestBorFilters(t *testing.T) {
filter := NewBorBlockLogsRangeFilter(backend, sprint, 0, 18, []common.Address{addr}, [][]common.Hash{{hash1, hash2, hash3, hash4}})
logs, err := filter.Logs(context.Background())
if err != nil {
t.Error(err)
}
if len(logs) != 4 {
t.Error("expected 4 log, got", len(logs))
}
@ -119,6 +125,7 @@ func TestBorFilters(t *testing.T) {
filter = NewBorBlockLogsRangeFilter(backend, sprint, 1, 16, []common.Address{addr}, [][]common.Hash{{hash1, hash2}})
logs, _ = filter.Logs(context.Background())
if len(logs) != 2 {
t.Error("expected 2 log, got", len(logs))
}

View file

@ -2,7 +2,8 @@ package filters
import (
context "context"
"math/rand"
"crypto/rand"
"math/big"
common "github.com/ethereum/go-ethereum/common"
core "github.com/ethereum/go-ethereum/core"
@ -16,7 +17,6 @@ import (
)
type TestBackend struct {
mux *event.TypeMux
DB ethdb.Database
sections uint64
txFeed event.Feed
@ -35,21 +35,23 @@ func (b *TestBackend) BloomStatus() (uint64, uint64) {
func (b *TestBackend) GetBorBlockReceipt(ctx context.Context, hash common.Hash) (*types.Receipt, error) {
number := rawdb.ReadHeaderNumber(b.DB, hash)
if number == nil {
return nil, nil
return &types.Receipt{}, nil
}
receipt := rawdb.ReadBorReceipt(b.DB, hash, *number)
if receipt == nil {
return nil, nil
return &types.Receipt{}, nil
}
return receipt, nil
}
func (b *TestBackend) GetBorBlockLogs(ctx context.Context, hash common.Hash) ([]*types.Log, error) {
receipt, err := b.GetBorBlockReceipt(ctx, hash)
if receipt == nil || err != nil {
return nil, nil
return []*types.Log{}, nil
}
return receipt.Logs, nil
}
@ -62,25 +64,30 @@ func (b *TestBackend) HeaderByNumber(ctx context.Context, blockNr rpc.BlockNumbe
hash common.Hash
num uint64
)
if blockNr == rpc.LatestBlockNumber {
hash = rawdb.ReadHeadBlockHash(b.DB)
number := rawdb.ReadHeaderNumber(b.DB, hash)
if number == nil {
return nil, nil
return &types.Header{}, nil
}
num = *number
} else {
num = uint64(blockNr)
hash = rawdb.ReadCanonicalHash(b.DB, num)
}
return rawdb.ReadHeader(b.DB, hash, num), nil
}
func (b *TestBackend) HeaderByHash(ctx context.Context, hash common.Hash) (*types.Header, error) {
number := rawdb.ReadHeaderNumber(b.DB, hash)
if number == nil {
return nil, nil
return &types.Header{}, nil
}
return rawdb.ReadHeader(b.DB, hash, *number), nil
}
@ -88,6 +95,7 @@ func (b *TestBackend) GetReceipts(ctx context.Context, hash common.Hash) (types.
if number := rawdb.ReadHeaderNumber(b.DB, hash); number != nil {
return rawdb.ReadReceipts(b.DB, hash, *number, params.TestChainConfig), nil
}
return nil, nil
}
@ -96,12 +104,14 @@ func (b *TestBackend) GetLogs(ctx context.Context, hash common.Hash) ([][]*types
if number == nil {
return nil, nil
}
receipts := rawdb.ReadReceipts(b.DB, hash, *number, params.TestChainConfig)
logs := make([][]*types.Log, len(receipts))
for i, receipt := range receipts {
logs[i] = receipt.Logs
}
return logs, nil
}
@ -141,7 +151,13 @@ func (b *TestBackend) ServiceFilter(ctx context.Context, session *bloombits.Matc
task.Bitsets = make([][]byte, len(task.Sections))
for i, section := range task.Sections {
if rand.Int()%4 != 0 { // Handle occasional missing deliveries
nBig, err := rand.Int(rand.Reader, big.NewInt(100))
if err != nil {
panic(err)
}
if nBig.Int64()%4 != 0 { // Handle occasional missing deliveries
head := rawdb.ReadCanonicalHash(b.DB, (section+1)*params.BloomBitsBlocks-1)
task.Bitsets[i], _ = rawdb.ReadBloomBits(b.DB, task.Bit, section, head)
}