mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2024-12-23 04:03:49 +00:00
11f4978ed4
Following our previous discussion on erigon's discord, this PR requests to upstream all Otterscan modifications to erigon's repo. That decision comes after getting feedback from lots of users at events this year, and although it may introduce some friction for development, it will make integrators life easier by having all our modifications available out of box, e.g., dappnode users will get our RPCs since their official packages are built from erigon repo. I'm submitting the source-code as-is, please let me know if you think there is a better code organization. The current set of modifications comprises only new RPCs. There are some proposals for extra-stages that would add new tables, but they are still WIP and will be submitted separately in future after more testing.
144 lines
4.0 KiB
Go
144 lines
4.0 KiB
Go
package commands
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/RoaringBitmap/roaring/roaring64"
|
|
)
|
|
|
|
func newMockForwardChunkLocator(chunks [][]byte) ChunkLocator {
|
|
return func(block uint64) (ChunkProvider, bool, error) {
|
|
for i, v := range chunks {
|
|
bm := roaring64.NewBitmap()
|
|
if _, err := bm.ReadFrom(bytes.NewReader(v)); err != nil {
|
|
return nil, false, err
|
|
}
|
|
if block > bm.Maximum() {
|
|
continue
|
|
}
|
|
|
|
return newMockForwardChunkProvider(chunks[i:]), true, nil
|
|
}
|
|
|
|
// Not found; return the last to simulate the behavior of returning
|
|
// the 0xffff... chunk
|
|
if len(chunks) > 0 {
|
|
return newMockForwardChunkProvider(chunks[len(chunks)-1:]), true, nil
|
|
}
|
|
|
|
return nil, true, nil
|
|
}
|
|
}
|
|
|
|
func newMockForwardChunkProvider(chunks [][]byte) ChunkProvider {
|
|
i := 0
|
|
return func() ([]byte, bool, error) {
|
|
if i >= len(chunks) {
|
|
return nil, false, nil
|
|
}
|
|
|
|
chunk := chunks[i]
|
|
i++
|
|
return chunk, true, nil
|
|
}
|
|
}
|
|
|
|
func TestForwardBlockProviderWith1Chunk(t *testing.T) {
|
|
// Mocks 1 chunk
|
|
chunk1 := createBitmap(t, []uint64{1000, 1005, 1010})
|
|
|
|
chunkLocator := newMockForwardChunkLocator([][]byte{chunk1})
|
|
blockProvider := NewForwardBlockProvider(chunkLocator, 0)
|
|
|
|
checkNext(t, blockProvider, 1000, true)
|
|
checkNext(t, blockProvider, 1005, true)
|
|
checkNext(t, blockProvider, 1010, false)
|
|
}
|
|
|
|
func TestForwardBlockProviderWith1ChunkMiddleBlock(t *testing.T) {
|
|
// Mocks 1 chunk
|
|
chunk1 := createBitmap(t, []uint64{1000, 1005, 1010})
|
|
|
|
chunkLocator := newMockForwardChunkLocator([][]byte{chunk1})
|
|
blockProvider := NewForwardBlockProvider(chunkLocator, 1005)
|
|
|
|
checkNext(t, blockProvider, 1005, true)
|
|
checkNext(t, blockProvider, 1010, false)
|
|
}
|
|
|
|
func TestForwardBlockProviderWith1ChunkNotExactBlock(t *testing.T) {
|
|
// Mocks 1 chunk
|
|
chunk1 := createBitmap(t, []uint64{1000, 1005, 1010})
|
|
|
|
chunkLocator := newMockForwardChunkLocator([][]byte{chunk1})
|
|
blockProvider := NewForwardBlockProvider(chunkLocator, 1007)
|
|
|
|
checkNext(t, blockProvider, 1010, false)
|
|
}
|
|
|
|
func TestForwardBlockProviderWith1ChunkLastBlock(t *testing.T) {
|
|
// Mocks 1 chunk
|
|
chunk1 := createBitmap(t, []uint64{1000, 1005, 1010})
|
|
|
|
chunkLocator := newMockForwardChunkLocator([][]byte{chunk1})
|
|
blockProvider := NewForwardBlockProvider(chunkLocator, 1010)
|
|
|
|
checkNext(t, blockProvider, 1010, false)
|
|
}
|
|
|
|
func TestForwardBlockProviderWith1ChunkBlockNotFound(t *testing.T) {
|
|
// Mocks 1 chunk
|
|
chunk1 := createBitmap(t, []uint64{1000, 1005, 1010})
|
|
|
|
chunkLocator := newMockForwardChunkLocator([][]byte{chunk1})
|
|
blockProvider := NewForwardBlockProvider(chunkLocator, 1100)
|
|
|
|
checkNext(t, blockProvider, 0, false)
|
|
}
|
|
|
|
func TestForwardBlockProviderWithNoChunks(t *testing.T) {
|
|
chunkLocator := newMockForwardChunkLocator([][]byte{})
|
|
blockProvider := NewForwardBlockProvider(chunkLocator, 0)
|
|
|
|
checkNext(t, blockProvider, 0, false)
|
|
}
|
|
|
|
func TestForwardBlockProviderWithMultipleChunks(t *testing.T) {
|
|
// Mocks 2 chunks
|
|
chunk1 := createBitmap(t, []uint64{1000, 1005, 1010})
|
|
chunk2 := createBitmap(t, []uint64{1501, 1600})
|
|
|
|
chunkLocator := newMockForwardChunkLocator([][]byte{chunk1, chunk2})
|
|
blockProvider := NewForwardBlockProvider(chunkLocator, 0)
|
|
|
|
checkNext(t, blockProvider, 1000, true)
|
|
checkNext(t, blockProvider, 1005, true)
|
|
checkNext(t, blockProvider, 1010, true)
|
|
checkNext(t, blockProvider, 1501, true)
|
|
checkNext(t, blockProvider, 1600, false)
|
|
}
|
|
|
|
func TestForwardBlockProviderWithMultipleChunksBlockBetweenChunks(t *testing.T) {
|
|
// Mocks 2 chunks
|
|
chunk1 := createBitmap(t, []uint64{1000, 1005, 1010})
|
|
chunk2 := createBitmap(t, []uint64{1501, 1600})
|
|
|
|
chunkLocator := newMockForwardChunkLocator([][]byte{chunk1, chunk2})
|
|
blockProvider := NewForwardBlockProvider(chunkLocator, 1300)
|
|
|
|
checkNext(t, blockProvider, 1501, true)
|
|
checkNext(t, blockProvider, 1600, false)
|
|
}
|
|
|
|
func TestForwardBlockProviderWithMultipleChunksBlockNotFound(t *testing.T) {
|
|
// Mocks 2 chunks
|
|
chunk1 := createBitmap(t, []uint64{1000, 1005, 1010})
|
|
chunk2 := createBitmap(t, []uint64{1501, 1600})
|
|
|
|
chunkLocator := newMockForwardChunkLocator([][]byte{chunk1, chunk2})
|
|
blockProvider := NewForwardBlockProvider(chunkLocator, 1700)
|
|
|
|
checkNext(t, blockProvider, 0, false)
|
|
}
|