github.com/hyperledger/burrow@v0.34.5-0.20220512172541-77f09336001d/acm/validator/set_test.go (about)

     1  package validator
     2  
     3  import (
     4  	"fmt"
     5  	"math/big"
     6  	"testing"
     7  
     8  	"github.com/hyperledger/burrow/acm"
     9  	"github.com/hyperledger/burrow/crypto"
    10  	"github.com/stretchr/testify/assert"
    11  )
    12  
    13  func TestValidators_AlterPower(t *testing.T) {
    14  	vs := NewSet()
    15  	pow1 := big.NewInt(2312312321)
    16  	pubA := pubKey(1)
    17  	vs.ChangePower(pubA, pow1)
    18  	assert.Equal(t, pow1, vs.TotalPower())
    19  	vs.ChangePower(pubA, big.NewInt(0))
    20  	assertZero(t, vs.TotalPower())
    21  }
    22  
    23  func pubKey(secret interface{}) *crypto.PublicKey {
    24  	return acm.NewAccountFromSecret(fmt.Sprintf("%v", secret)).PublicKey
    25  }