mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-23 11:57:18 +00:00
d17996f8b0
* Update V3 from V4 * Fix build v3 -> v4 * Update ssz * Update beacon_chain.pb.go * Fix formatter import * Update update-mockgen.sh comment to v4 * Fix conflicts. Pass build and tests * Fix test
316 lines
8.6 KiB
Go
316 lines
8.6 KiB
Go
package monitor
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
|
|
mock "github.com/prysmaticlabs/prysm/v4/beacon-chain/blockchain/testing"
|
|
"github.com/prysmaticlabs/prysm/v4/beacon-chain/core/altair"
|
|
"github.com/prysmaticlabs/prysm/v4/beacon-chain/core/feed"
|
|
statefeed "github.com/prysmaticlabs/prysm/v4/beacon-chain/core/feed/state"
|
|
testDB "github.com/prysmaticlabs/prysm/v4/beacon-chain/db/testing"
|
|
doublylinkedtree "github.com/prysmaticlabs/prysm/v4/beacon-chain/forkchoice/doubly-linked-tree"
|
|
"github.com/prysmaticlabs/prysm/v4/beacon-chain/state/stategen"
|
|
"github.com/prysmaticlabs/prysm/v4/consensus-types/blocks"
|
|
"github.com/prysmaticlabs/prysm/v4/consensus-types/primitives"
|
|
"github.com/prysmaticlabs/prysm/v4/encoding/bytesutil"
|
|
"github.com/prysmaticlabs/prysm/v4/testing/require"
|
|
"github.com/prysmaticlabs/prysm/v4/testing/util"
|
|
"github.com/prysmaticlabs/prysm/v4/time/slots"
|
|
logTest "github.com/sirupsen/logrus/hooks/test"
|
|
)
|
|
|
|
func setupService(t *testing.T) *Service {
|
|
beaconDB := testDB.SetupDB(t)
|
|
state, _ := util.DeterministicGenesisStateAltair(t, 256)
|
|
|
|
pubKeys := make([][]byte, 3)
|
|
pubKeys[0] = state.Validators()[0].PublicKey
|
|
pubKeys[1] = state.Validators()[1].PublicKey
|
|
pubKeys[2] = state.Validators()[2].PublicKey
|
|
|
|
currentSyncCommittee := util.ConvertToCommittee([][]byte{
|
|
pubKeys[0], pubKeys[1], pubKeys[2], pubKeys[1], pubKeys[1],
|
|
})
|
|
require.NoError(t, state.SetCurrentSyncCommittee(currentSyncCommittee))
|
|
|
|
chainService := &mock.ChainService{
|
|
Genesis: time.Now(),
|
|
DB: beaconDB,
|
|
State: state,
|
|
Root: []byte("hello-world"),
|
|
ValidatorsRoot: [32]byte{},
|
|
}
|
|
|
|
trackedVals := map[primitives.ValidatorIndex]bool{
|
|
1: true,
|
|
2: true,
|
|
12: true,
|
|
15: true,
|
|
}
|
|
latestPerformance := map[primitives.ValidatorIndex]ValidatorLatestPerformance{
|
|
1: {
|
|
balance: 32000000000,
|
|
},
|
|
2: {
|
|
balance: 32000000000,
|
|
},
|
|
12: {
|
|
balance: 31900000000,
|
|
},
|
|
15: {
|
|
balance: 31900000000,
|
|
},
|
|
}
|
|
aggregatedPerformance := map[primitives.ValidatorIndex]ValidatorAggregatedPerformance{
|
|
1: {
|
|
startEpoch: 0,
|
|
startBalance: 31700000000,
|
|
totalAttestedCount: 12,
|
|
totalRequestedCount: 15,
|
|
totalDistance: 14,
|
|
totalCorrectHead: 8,
|
|
totalCorrectSource: 11,
|
|
totalCorrectTarget: 12,
|
|
totalProposedCount: 1,
|
|
totalSyncCommitteeContributions: 0,
|
|
totalSyncCommitteeAggregations: 0,
|
|
},
|
|
2: {},
|
|
12: {},
|
|
15: {},
|
|
}
|
|
trackedSyncCommitteeIndices := map[primitives.ValidatorIndex][]primitives.CommitteeIndex{
|
|
1: {0, 1, 2, 3},
|
|
12: {4, 5},
|
|
}
|
|
return &Service{
|
|
config: &ValidatorMonitorConfig{
|
|
StateGen: stategen.New(beaconDB, doublylinkedtree.New()),
|
|
StateNotifier: chainService.StateNotifier(),
|
|
HeadFetcher: chainService,
|
|
AttestationNotifier: chainService.OperationNotifier(),
|
|
},
|
|
|
|
ctx: context.Background(),
|
|
TrackedValidators: trackedVals,
|
|
latestPerformance: latestPerformance,
|
|
aggregatedPerformance: aggregatedPerformance,
|
|
trackedSyncCommitteeIndices: trackedSyncCommitteeIndices,
|
|
lastSyncedEpoch: 0,
|
|
}
|
|
}
|
|
|
|
func TestTrackedIndex(t *testing.T) {
|
|
s := &Service{
|
|
TrackedValidators: map[primitives.ValidatorIndex]bool{
|
|
1: true,
|
|
2: true,
|
|
},
|
|
}
|
|
require.Equal(t, s.trackedIndex(primitives.ValidatorIndex(1)), true)
|
|
require.Equal(t, s.trackedIndex(primitives.ValidatorIndex(3)), false)
|
|
}
|
|
|
|
func TestUpdateSyncCommitteeTrackedVals(t *testing.T) {
|
|
hook := logTest.NewGlobal()
|
|
s := setupService(t)
|
|
state, _ := util.DeterministicGenesisStateAltair(t, 1024)
|
|
|
|
s.updateSyncCommitteeTrackedVals(state)
|
|
require.LogsDoNotContain(t, hook, "Sync committee assignments will not be reported")
|
|
newTrackedSyncIndices := map[primitives.ValidatorIndex][]primitives.CommitteeIndex{
|
|
1: {1, 3, 4},
|
|
2: {2},
|
|
}
|
|
require.DeepEqual(t, s.trackedSyncCommitteeIndices, newTrackedSyncIndices)
|
|
}
|
|
|
|
func TestNewService(t *testing.T) {
|
|
config := &ValidatorMonitorConfig{}
|
|
var tracked []primitives.ValidatorIndex
|
|
ctx := context.Background()
|
|
_, err := NewService(ctx, config, tracked)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestStart(t *testing.T) {
|
|
hook := logTest.NewGlobal()
|
|
s := setupService(t)
|
|
stateChannel := make(chan *feed.Event, 1)
|
|
stateSub := s.config.StateNotifier.StateFeed().Subscribe(stateChannel)
|
|
defer stateSub.Unsubscribe()
|
|
|
|
wg := &sync.WaitGroup{}
|
|
wg.Add(1)
|
|
s.Start()
|
|
|
|
go func() {
|
|
select {
|
|
case stateEvent := <-stateChannel:
|
|
if stateEvent.Type == statefeed.Synced {
|
|
_, ok := stateEvent.Data.(*statefeed.SyncedData)
|
|
require.Equal(t, true, ok, "Event feed data is not type *statefeed.SyncedData")
|
|
}
|
|
case <-s.ctx.Done():
|
|
}
|
|
wg.Done()
|
|
}()
|
|
|
|
for sent := 0; sent == 0; {
|
|
sent = s.config.StateNotifier.StateFeed().Send(&feed.Event{
|
|
Type: statefeed.Synced,
|
|
Data: &statefeed.SyncedData{
|
|
StartTime: time.Now(),
|
|
},
|
|
})
|
|
}
|
|
|
|
// wait for Logrus
|
|
time.Sleep(1000 * time.Millisecond)
|
|
require.LogsContain(t, hook, "Synced to head epoch, starting reporting performance")
|
|
require.LogsContain(t, hook, "\"Starting service\" ValidatorIndices=\"[1 2 12 15]\"")
|
|
s.Lock()
|
|
require.Equal(t, s.isLogging, true, "monitor is not running")
|
|
s.Unlock()
|
|
}
|
|
|
|
func TestInitializePerformanceStructures(t *testing.T) {
|
|
hook := logTest.NewGlobal()
|
|
ctx := context.Background()
|
|
s := setupService(t)
|
|
state, err := s.config.HeadFetcher.HeadState(ctx)
|
|
require.NoError(t, err)
|
|
epoch := slots.ToEpoch(state.Slot())
|
|
s.initializePerformanceStructures(state, epoch)
|
|
require.LogsDoNotContain(t, hook, "Could not fetch starting balance")
|
|
latestPerformance := map[primitives.ValidatorIndex]ValidatorLatestPerformance{
|
|
1: {
|
|
balance: 32000000000,
|
|
},
|
|
2: {
|
|
balance: 32000000000,
|
|
},
|
|
12: {
|
|
balance: 32000000000,
|
|
},
|
|
15: {
|
|
balance: 32000000000,
|
|
},
|
|
}
|
|
aggregatedPerformance := map[primitives.ValidatorIndex]ValidatorAggregatedPerformance{
|
|
1: {
|
|
startBalance: 32000000000,
|
|
},
|
|
2: {
|
|
startBalance: 32000000000,
|
|
},
|
|
12: {
|
|
startBalance: 32000000000,
|
|
},
|
|
15: {
|
|
startBalance: 32000000000,
|
|
},
|
|
}
|
|
|
|
require.DeepEqual(t, s.latestPerformance, latestPerformance)
|
|
require.DeepEqual(t, s.aggregatedPerformance, aggregatedPerformance)
|
|
}
|
|
|
|
func TestMonitorRoutine(t *testing.T) {
|
|
ctx := context.Background()
|
|
hook := logTest.NewGlobal()
|
|
s := setupService(t)
|
|
stateChannel := make(chan *feed.Event, 1)
|
|
stateSub := s.config.StateNotifier.StateFeed().Subscribe(stateChannel)
|
|
|
|
wg := &sync.WaitGroup{}
|
|
wg.Add(1)
|
|
|
|
go func() {
|
|
s.monitorRoutine(stateChannel, stateSub)
|
|
wg.Done()
|
|
}()
|
|
|
|
genesis, keys := util.DeterministicGenesisStateAltair(t, 64)
|
|
c, err := altair.NextSyncCommittee(ctx, genesis)
|
|
require.NoError(t, err)
|
|
require.NoError(t, genesis.SetCurrentSyncCommittee(c))
|
|
|
|
genConfig := util.DefaultBlockGenConfig()
|
|
block, err := util.GenerateFullBlockAltair(genesis, keys, genConfig, 1)
|
|
require.NoError(t, err)
|
|
root, err := block.GetBlock().HashTreeRoot()
|
|
require.NoError(t, err)
|
|
require.NoError(t, s.config.StateGen.SaveState(ctx, root, genesis))
|
|
|
|
wrapped, err := blocks.NewSignedBeaconBlock(block)
|
|
require.NoError(t, err)
|
|
|
|
stateChannel <- &feed.Event{
|
|
Type: statefeed.BlockProcessed,
|
|
Data: &statefeed.BlockProcessedData{
|
|
Slot: 1,
|
|
Verified: true,
|
|
SignedBlock: wrapped,
|
|
},
|
|
}
|
|
|
|
// Wait for Logrus
|
|
time.Sleep(1000 * time.Millisecond)
|
|
wanted1 := fmt.Sprintf("\"Proposed beacon block was included\" BalanceChange=100000000 BlockRoot=%#x NewBalance=32000000000 ParentRoot=0xf732eaeb7fae ProposerIndex=15 Slot=1 Version=1 prefix=monitor", bytesutil.Trunc(root[:]))
|
|
require.LogsContain(t, hook, wanted1)
|
|
|
|
}
|
|
|
|
func TestWaitForSync(t *testing.T) {
|
|
s := setupService(t)
|
|
stateChannel := make(chan *feed.Event, 1)
|
|
stateSub := s.config.StateNotifier.StateFeed().Subscribe(stateChannel)
|
|
defer stateSub.Unsubscribe()
|
|
|
|
wg := &sync.WaitGroup{}
|
|
wg.Add(1)
|
|
|
|
go func() {
|
|
err := s.waitForSync(stateChannel, stateSub)
|
|
require.NoError(t, err)
|
|
wg.Done()
|
|
}()
|
|
|
|
stateChannel <- &feed.Event{
|
|
Type: statefeed.Synced,
|
|
Data: &statefeed.SyncedData{
|
|
StartTime: time.Now(),
|
|
},
|
|
}
|
|
}
|
|
|
|
func TestRun(t *testing.T) {
|
|
hook := logTest.NewGlobal()
|
|
s := setupService(t)
|
|
stateChannel := make(chan *feed.Event, 1)
|
|
stateSub := s.config.StateNotifier.StateFeed().Subscribe(stateChannel)
|
|
|
|
wg := &sync.WaitGroup{}
|
|
wg.Add(1)
|
|
|
|
go func() {
|
|
s.run(stateChannel, stateSub)
|
|
wg.Done()
|
|
}()
|
|
|
|
stateChannel <- &feed.Event{
|
|
Type: statefeed.Synced,
|
|
Data: &statefeed.SyncedData{
|
|
StartTime: time.Now(),
|
|
},
|
|
}
|
|
//wait for Logrus
|
|
time.Sleep(1000 * time.Millisecond)
|
|
require.LogsContain(t, hook, "Synced to head epoch, starting reporting performance")
|
|
}
|