fix minus voting power panic

CHANGELOG_PENDING.md

@@ -13,6 +13,7 @@
 
 - P2P Protocol
   - [abci] [\#100](https://github.com/line/tendermint/pull/100) Add `voters_hash` field, which is needed for verification of a block header
+  - [abci] [\#102](https://github.com/line/tendermint/pull/102) Add voting power in `VoterInfo` of abci 
 
 - Go API
 

abci/types/types.proto

@@ -314,7 +314,8 @@ message ValidatorUpdate {
 // VoteInfo
 message VoteInfo {
   Validator validator         = 1 [(gogoproto.nullable) = false];
-  bool      signed_last_block = 2;
+  int64     voting_power      = 2;
+  bool      signed_last_block = 3;
 }
 
 message PubKey {

libs/rand/sampling.go

@@ -142,14 +142,20 @@ func RandomSamplingWithoutReplacement(
 		compensationProportions[i].Add(&compensationProportions[i+1], additionalCompensation)
 	}
 	winners = candidates[len(candidates)-winnerNum:]
+	winPoints := make([]*big.Int, len(winners))
+	totalWinPoint := new(big.Int)
+	for i, winner := range winners {
+		winPoints[i] = new(big.Int).SetUint64(winner.Priority())
+		winPoints[i].Mul(winPoints[i], &compensationProportions[i])
+		winPoints[i].Add(winPoints[i], correction)
+		totalWinPoint.Add(totalWinPoint, winPoints[i])
+	}
 	recalibration := new(big.Int).Div(correction, new(big.Int).SetUint64(precisionCorrectionForSelection))
 	for i, winner := range winners {
-		// winPoint = correction + winner.Priority() * compensationProportions[i]
-		winPoint := new(big.Int).SetUint64(winner.Priority())
-		winPoint.Mul(winPoint, &compensationProportions[i])
-		winPoint.Add(winPoint, correction)
-
-		winner.SetWinPoint(winPoint.Div(winPoint, recalibration).Int64())
+		winPoint := new(big.Int)
+		winPoint.Mul(recalibration, winPoints[i])
+		winPoint.Div(winPoint, totalWinPoint)
+		winner.SetWinPoint(winPoint.Int64())
 	}
 	return winners
 }

libs/rand/sampling_test.go

@@ -187,7 +187,6 @@ func TestRandomSamplingWithoutReplacementOverflow(t *testing.T) {
 		assert.True(t, element.winPoint <= lastWinPoint)
 		lastWinPoint = element.winPoint
 	}
-	assert.Equal(t, lastWinPoint, int64(precisionForSelection))
 }
 
 func accumulateAndResetReward(candidate []Candidate, acc []uint64) uint64 {

state/execution.go

@@ -346,7 +346,9 @@ func getBeginBlockValidatorInfo(block *types.Block, stateDB dbm.DB, voterParams
 		for i, val := range lastVoterSet.Voters {
 			commitSig := block.LastCommit.Signatures[i]
 			voteInfos[i] = abci.VoteInfo{
-				Validator:       types.TM2PB.Validator(val),
+				Validator: types.TM2PB.Validator(val),
+				// TODO We need to change distribution of cosmos-sdk in order to reference this power for reward later
+				VotingPower:     val.VotingPower,
 				SignedLastBlock: !commitSig.Absent(),
 			}
 		}

types/voter_set_test.go

@@ -4,6 +4,8 @@ import (
 	"math"
 	"testing"
 
+	"github.com/tendermint/tendermint/libs/rand"
+
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 
@@ -393,3 +395,40 @@ func TestVoterSetProtoBuf(t *testing.T) {
 		}
 	}
 }
+
+func testVotingPower(t *testing.T, valSet *ValidatorSet) {
+	voterParams := &VoterParams{
+		VoterElectionThreshold:          100,
+		MaxTolerableByzantinePercentage: 20,
+		ElectionPrecision:               2,
+	}
+
+	voterSetNoSampling := SelectVoter(valSet, []byte{0}, voterParams)
+	for _, v := range voterSetNoSampling.Voters {
+		assert.True(t, v.StakingPower == v.VotingPower)
+	}
+
+	for i := 90; i > 50; i-- {
+		voterParams.VoterElectionThreshold = int32(i)
+		voterSetSampling := SelectVoter(valSet, []byte{0}, voterParams)
+		allSame := true
+		for _, v := range voterSetSampling.Voters {
+			if v.StakingPower != v.VotingPower {
+				allSame = false
+				break
+			}
+		}
+		assert.False(t, allSame)
+		assert.True(t, valSet.TotalStakingPower() > voterSetSampling.TotalVotingPower())
+		assert.True(t, valSet.TotalStakingPower()-voterSetSampling.TotalVotingPower() < 1000)
+	}
+}
+
+func TestVotingPower(t *testing.T) {
+	testVotingPower(t, randValidatorSet(100))
+	vals := make([]*Validator, 100)
+	for i := 0; i < len(vals); i++ {
+		vals[i] = newValidator(rand.Bytes(32), 100)
+	}
+	testVotingPower(t, NewValidatorSet(vals))
+}