add test code for rewarding delegations

5 years ago · d51cd526ee
parent 6953a1a7f8
commit d51cd526ee
1 changed files with 109 additions and 0 deletions
--- a/test/chain/reward/main.go
+++ b/test/chain/reward/main.go
@ -0,0 +1,109 @@
+package main
+
+import (
+	"fmt"
+	"math/big"
+	"math/rand"
+	"time"
+
+	common2 "github.com/ethereum/go-ethereum/common"
+	"github.com/ethereum/go-ethereum/ethdb"
+	"github.com/harmony-one/bls/ffi/go/bls"
+	"github.com/harmony-one/harmony/core"
+	"github.com/harmony-one/harmony/core/state"
+	"github.com/harmony-one/harmony/crypto/hash"
+	"github.com/harmony-one/harmony/internal/common"
+	"github.com/harmony-one/harmony/numeric"
+	"github.com/harmony-one/harmony/shard"
+	staking "github.com/harmony-one/harmony/staking/types"
+)
+
+var (
+	validatorAddress = common2.Address(common.MustBech32ToAddress("one1pdv9lrdwl0rg5vglh4xtyrv3wjk3wsqket7zxy"))
+
+	testBLSPubKey    = "30b2c38b1316da91e068ac3bd8751c0901ef6c02a1d58bc712104918302c6ed03d5894671d0c816dad2b4d303320f202"
+	testBLSPrvKey    = "c6d7603520311f7a4e6aac0b26701fc433b75b38df504cd416ef2b900cd66205"
+	postStakingEpoch = big.NewInt(200)
+)
+
+func init() {
+
+	bls.Init(bls.BLS12_381)
+}
+
+func generateBlsKeySigPair() (shard.BlsPublicKey, shard.BLSSignature) {
+	p := &bls.PublicKey{}
+	p.DeserializeHexStr(testBLSPubKey)
+	pub := shard.BlsPublicKey{}
+	pub.FromLibBLSPublicKey(p)
+	messageBytes := []byte(staking.BlsVerificationStr)
+	privateKey := &bls.SecretKey{}
+	privateKey.DeserializeHexStr(testBLSPrvKey)
+	msgHash := hash.Keccak256(messageBytes)
+	signature := privateKey.SignHash(msgHash[:])
+	var sig shard.BLSSignature
+	copy(sig[:], signature.Serialize())
+	return pub, sig
+}
+
+func createValidator() *staking.CreateValidator {
+	desc := staking.Description{
+		Name:            "SuperHero",
+		Identity:        "YouWouldNotKnow",
+		Website:         "Secret Website",
+		SecurityContact: "LicenseToKill",
+		Details:         "blah blah blah",
+	}
+	rate, _ := numeric.NewDecFromStr("0.1")
+	maxRate, _ := numeric.NewDecFromStr("0.5")
+	maxChangeRate, _ := numeric.NewDecFromStr("0.05")
+	commission := staking.CommissionRates{
+		Rate:          rate,
+		MaxRate:       maxRate,
+		MaxChangeRate: maxChangeRate,
+	}
+	minSelfDel := big.NewInt(1e18)
+	maxTotalDel := big.NewInt(9e18)
+	pubKey, pubSig := generateBlsKeySigPair()
+	slotPubKeys := []shard.BlsPublicKey{pubKey}
+	slotKeySigs := []shard.BLSSignature{pubSig}
+	amount := big.NewInt(5e18)
+	v := staking.CreateValidator{
+		ValidatorAddress:   validatorAddress,
+		Description:        desc,
+		CommissionRates:    commission,
+		MinSelfDelegation:  minSelfDel,
+		MaxTotalDelegation: maxTotalDel,
+		SlotPubKeys:        slotPubKeys,
+		SlotKeySigs:        slotKeySigs,
+		Amount:             amount,
+	}
+	return &v
+}
+
+func main() {
+	statedb, _ := state.New(common2.Hash{}, state.NewDatabase(ethdb.NewMemDatabase()))
+	msg := createValidator()
+	statedb.AddBalance(msg.ValidatorAddress, big.NewInt(5e18))
+	validator, _ := core.VerifyAndCreateValidatorFromMsg(
+		statedb, postStakingEpoch, big.NewInt(0), msg,
+	)
+	for i := 0; i < 100000; i++ {
+		validator.Delegations = append(validator.Delegations, staking.Delegation{
+			common2.Address{},
+			big.NewInt(int64(rand.Intn(100))),
+			big.NewInt(0),
+			nil,
+		})
+	}
+
+	statedb.UpdateValidatorWrapper(validator.Address, validator)
+
+	startTime := time.Now()
+	validator, _ = statedb.ValidatorWrapper(msg.ValidatorAddress)
+
+	fmt.Printf("Total num delegations: %d\n", len(validator.Delegations))
+	statedb.AddReward(validator, big.NewInt(1000))
+	endTime := time.Now()
+	fmt.Printf("Time required to reward a validator with %d delegations: %f seconds\n", len(validator.Delegations), endTime.Sub(startTime).Seconds())
+}