mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-09 11:11:20 +00:00
37 lines
864 B
Go
37 lines
864 B
Go
|
package peers_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io/ioutil"
|
||
|
|
"math"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/flags"
|
||
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/p2p/peers"
|
||
|
|
"github.com/prysmaticlabs/prysm/shared/featureconfig"
|
||
|
|
"github.com/sirupsen/logrus"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestMain(m *testing.M) {
|
||
|
|
logrus.SetLevel(logrus.DebugLevel)
|
||
|
|
logrus.SetOutput(ioutil.Discard)
|
||
|
|
|
||
|
|
resetCfg := featureconfig.InitWithReset(&featureconfig.Flags{})
|
||
|
|
defer resetCfg()
|
||
|
|
|
||
|
|
resetFlags := flags.Get()
|
||
|
|
flags.Init(&flags.GlobalFlags{
|
||
|
|
BlockBatchLimit: 64,
|
||
|
|
BlockBatchLimitBurstFactor: 10,
|
||
|
|
})
|
||
|
|
defer func() {
|
||
|
|
flags.Init(resetFlags)
|
||
|
|
}()
|
||
|
|
os.Exit(m.Run())
|
||
|
|
}
|
||
|
|
|
||
|
|
// roundScore returns score rounded in accordance with the score manager's rounding factor.
|
||
|
|
func roundScore(score float64) float64 {
|
||
|
|
return math.Round(score*peers.ScoreRoundingFactor) / peers.ScoreRoundingFactor
|
||
|
|
}
|