forked from TrueCloudLab/neoneo-go
66 lines
1.9 KiB
Go
66 lines
1.9 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/hex"
|
||
|
|
"math/big"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Register standby validator and vote for it.
|
||
|
|
// We don't create a new account here, because chain will
|
||
|
|
// stop working after validator will change.
|
||
|
|
func TestRegisterCandidate(t *testing.T) {
|
||
|
|
e := newExecutor(t, true)
|
||
|
|
defer e.Close(t)
|
||
|
|
|
||
|
|
e.In.WriteString("one\r")
|
||
|
|
e.Run(t, "neo-go", "wallet", "nep5", "multitransfer",
|
||
|
|
"--unittest", "--rpc-endpoint", "http://"+e.RPC.Addr,
|
||
|
|
"--wallet", validatorWallet,
|
||
|
|
"--from", validatorAddr,
|
||
|
|
"neo:"+validatorPriv.Address()+":10",
|
||
|
|
"gas:"+validatorPriv.Address()+":100")
|
||
|
|
e.checkTxPersisted(t)
|
||
|
|
|
||
|
|
e.In.WriteString("one\r")
|
||
|
|
e.Run(t, "neo-go", "wallet", "candidate", "register",
|
||
|
|
"--unittest", "--rpc-endpoint", "http://"+e.RPC.Addr,
|
||
|
|
"--wallet", validatorWallet,
|
||
|
|
"--address", validatorPriv.Address())
|
||
|
|
e.checkTxPersisted(t)
|
||
|
|
|
||
|
|
vs, err := e.Chain.GetEnrollments()
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, 1, len(vs))
|
||
|
|
require.Equal(t, validatorPriv.PublicKey(), vs[0].Key)
|
||
|
|
require.Equal(t, big.NewInt(0), vs[0].Votes)
|
||
|
|
|
||
|
|
t.Run("Vote", func(t *testing.T) {
|
||
|
|
e.In.WriteString("one\r")
|
||
|
|
e.Run(t, "neo-go", "wallet", "candidate", "vote",
|
||
|
|
"--unittest", "--rpc-endpoint", "http://"+e.RPC.Addr,
|
||
|
|
"--wallet", validatorWallet,
|
||
|
|
"--address", validatorPriv.Address(),
|
||
|
|
"--candidate", hex.EncodeToString(validatorPriv.PublicKey().Bytes()))
|
||
|
|
e.checkTxPersisted(t)
|
||
|
|
|
||
|
|
vs, err = e.Chain.GetEnrollments()
|
||
|
|
require.Equal(t, 1, len(vs))
|
||
|
|
require.Equal(t, validatorPriv.PublicKey(), vs[0].Key)
|
||
|
|
b, _ := e.Chain.GetGoverningTokenBalance(validatorPriv.GetScriptHash())
|
||
|
|
require.Equal(t, b, vs[0].Votes)
|
||
|
|
})
|
||
|
|
|
||
|
|
e.In.WriteString("one\r")
|
||
|
|
e.Run(t, "neo-go", "wallet", "candidate", "unregister",
|
||
|
|
"--unittest", "--rpc-endpoint", "http://"+e.RPC.Addr,
|
||
|
|
"--wallet", validatorWallet,
|
||
|
|
"--address", validatorPriv.Address())
|
||
|
|
e.checkTxPersisted(t)
|
||
|
|
|
||
|
|
vs, err = e.Chain.GetEnrollments()
|
||
|
|
require.Equal(t, 0, len(vs))
|
||
|
|
}
|