mirror of
https://gitlab.com/pulsechaincom/erigon-pulse.git
synced 2024-12-22 03:30:37 +00:00
c5b75d00ca
It is possible that a span update happens during a milestone. A headers slice might cross to the new span. Also if 2 forks evolve simulaneously, a shorter fork can still be in the previous span. In these cases we need access to the previous span to calculate difficulty and validate header times. SpansCache will keep recent spans. The cache will be updated on new span events from the heimdall. The cache is pruned on new milestone events and in practice no more than 2 spans are kept. The header difficulty calculation and time validation depends on having a span for that header in the cache.
139 lines
3.5 KiB
Go
139 lines
3.5 KiB
Go
package sync
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
libcommon "github.com/ledgerwatch/erigon-lib/common"
|
|
"github.com/ledgerwatch/erigon/core/types"
|
|
"github.com/ledgerwatch/erigon/polygon/bor/borcfg"
|
|
)
|
|
|
|
type testValidatorSetInterface struct {
|
|
signers []libcommon.Address
|
|
sprintNum int
|
|
}
|
|
|
|
func (v *testValidatorSetInterface) IncrementProposerPriority(times int) {
|
|
v.sprintNum = times
|
|
}
|
|
|
|
func (v *testValidatorSetInterface) GetSignerSuccessionNumber(signer libcommon.Address, number uint64) (int, error) {
|
|
var i int
|
|
for (i < len(v.signers)) && (v.signers[i] != signer) {
|
|
i++
|
|
}
|
|
|
|
sprintOffset := v.sprintNum % len(v.signers)
|
|
var delta int
|
|
if i >= sprintOffset {
|
|
delta = i - sprintOffset
|
|
} else {
|
|
delta = i + len(v.signers) - sprintOffset
|
|
}
|
|
|
|
return delta, nil
|
|
}
|
|
|
|
func (v *testValidatorSetInterface) Difficulty(signer libcommon.Address) (uint64, error) {
|
|
delta, err := v.GetSignerSuccessionNumber(signer, 0)
|
|
if err != nil {
|
|
return 0, nil
|
|
}
|
|
return uint64(len(v.signers) - delta), nil
|
|
}
|
|
|
|
func TestSignerDifficulty(t *testing.T) {
|
|
borConfig := borcfg.BorConfig{
|
|
Sprint: map[string]uint64{"0": 16},
|
|
}
|
|
signers := []libcommon.Address{
|
|
libcommon.HexToAddress("00"),
|
|
libcommon.HexToAddress("01"),
|
|
libcommon.HexToAddress("02"),
|
|
}
|
|
validatorSetFactory := func(uint64) validatorSetInterface { return &testValidatorSetInterface{signers: signers} }
|
|
calc := NewDifficultyCalculator(&borConfig, nil, validatorSetFactory, nil).(*difficultyCalculatorImpl)
|
|
|
|
var d uint64
|
|
|
|
// sprint 0
|
|
d, _ = calc.signerDifficulty(signers[0], 0)
|
|
assert.Equal(t, uint64(3), d)
|
|
|
|
d, _ = calc.signerDifficulty(signers[0], 1)
|
|
assert.Equal(t, uint64(3), d)
|
|
|
|
d, _ = calc.signerDifficulty(signers[0], 15)
|
|
assert.Equal(t, uint64(3), d)
|
|
|
|
d, _ = calc.signerDifficulty(signers[1], 0)
|
|
assert.Equal(t, uint64(2), d)
|
|
|
|
d, _ = calc.signerDifficulty(signers[1], 1)
|
|
assert.Equal(t, uint64(2), d)
|
|
|
|
d, _ = calc.signerDifficulty(signers[1], 15)
|
|
assert.Equal(t, uint64(2), d)
|
|
|
|
d, _ = calc.signerDifficulty(signers[2], 0)
|
|
assert.Equal(t, uint64(1), d)
|
|
|
|
d, _ = calc.signerDifficulty(signers[2], 1)
|
|
assert.Equal(t, uint64(1), d)
|
|
|
|
d, _ = calc.signerDifficulty(signers[2], 15)
|
|
assert.Equal(t, uint64(1), d)
|
|
|
|
// sprint 1
|
|
d, _ = calc.signerDifficulty(signers[1], 16)
|
|
assert.Equal(t, uint64(3), d)
|
|
|
|
d, _ = calc.signerDifficulty(signers[2], 16)
|
|
assert.Equal(t, uint64(2), d)
|
|
|
|
d, _ = calc.signerDifficulty(signers[0], 16)
|
|
assert.Equal(t, uint64(1), d)
|
|
|
|
// sprint 2
|
|
d, _ = calc.signerDifficulty(signers[2], 32)
|
|
assert.Equal(t, uint64(3), d)
|
|
|
|
d, _ = calc.signerDifficulty(signers[0], 32)
|
|
assert.Equal(t, uint64(2), d)
|
|
|
|
d, _ = calc.signerDifficulty(signers[1], 32)
|
|
assert.Equal(t, uint64(1), d)
|
|
|
|
// sprint 3
|
|
d, _ = calc.signerDifficulty(signers[0], 48)
|
|
assert.Equal(t, uint64(3), d)
|
|
|
|
d, _ = calc.signerDifficulty(signers[1], 48)
|
|
assert.Equal(t, uint64(2), d)
|
|
|
|
d, _ = calc.signerDifficulty(signers[2], 48)
|
|
assert.Equal(t, uint64(1), d)
|
|
}
|
|
|
|
func TestHeaderDifficultyNoSignature(t *testing.T) {
|
|
borConfig := borcfg.BorConfig{}
|
|
spans := NewSpansCache()
|
|
calc := NewDifficultyCalculator(&borConfig, spans, nil, nil)
|
|
|
|
_, err := calc.HeaderDifficulty(new(types.Header))
|
|
require.ErrorContains(t, err, "signature suffix missing")
|
|
}
|
|
|
|
func TestSignerDifficultyNoSpan(t *testing.T) {
|
|
borConfig := borcfg.BorConfig{}
|
|
spans := NewSpansCache()
|
|
calc := NewDifficultyCalculator(&borConfig, spans, nil, nil).(*difficultyCalculatorImpl)
|
|
|
|
_, err := calc.signerDifficulty(libcommon.HexToAddress("00"), 0)
|
|
require.ErrorContains(t, err, "no span")
|
|
}
|