forked from TrueCloudLab/frostfs-contract
e1ee1c203c
If notary is enabled only alphabet calls are expected. Signed-off-by: Evgenii Stratonikov <evgeniy@nspcc.ru>
131 lines
4.1 KiB
Go
131 lines
4.1 KiB
Go
package tests
|
|
|
|
import (
|
|
"math/big"
|
|
"math/rand"
|
|
"testing"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/core"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
|
"github.com/nspcc-dev/neo-go/pkg/wallet"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
const netmapPath = "../netmap"
|
|
|
|
func deployNetmapContract(t *testing.T, bc *core.Blockchain, addrBalance, addrContainer util.Uint160, config ...interface{}) util.Uint160 {
|
|
args := make([]interface{}, 5)
|
|
args[0] = false
|
|
args[1] = addrBalance
|
|
args[2] = addrContainer
|
|
args[3] = []interface{}{CommitteeAcc.PrivateKey().PublicKey().Bytes()}
|
|
args[4] = append([]interface{}{}, config...)
|
|
return DeployContract(t, bc, netmapPath, args)
|
|
}
|
|
|
|
func prepareNetmapContract(t *testing.T, bc *core.Blockchain) util.Uint160 {
|
|
addrNNS := DeployContract(t, bc, nnsPath, nil)
|
|
|
|
ctrNetmap, err := ContractInfo(CommitteeAcc.Contract.ScriptHash(), netmapPath)
|
|
require.NoError(t, err)
|
|
|
|
ctrBalance, err := ContractInfo(CommitteeAcc.Contract.ScriptHash(), balancePath)
|
|
require.NoError(t, err)
|
|
|
|
ctrContainer, err := ContractInfo(CommitteeAcc.Contract.ScriptHash(), containerPath)
|
|
require.NoError(t, err)
|
|
|
|
deployContainerContract(t, bc, ctrNetmap.Hash, ctrBalance.Hash, addrNNS)
|
|
deployBalanceContract(t, bc, ctrNetmap.Hash, ctrContainer.Hash)
|
|
return deployNetmapContract(t, bc, ctrBalance.Hash, ctrContainer.Hash, "ContainerFee", []byte{})
|
|
}
|
|
|
|
func dummyNodeInfo(acc *wallet.Account) []byte {
|
|
ni := make([]byte, 66)
|
|
rand.Read(ni)
|
|
copy(ni[2:], acc.PrivateKey().PublicKey().Bytes())
|
|
return ni
|
|
}
|
|
|
|
func TestAddPeer(t *testing.T) {
|
|
bc := NewChain(t)
|
|
h := prepareNetmapContract(t, bc)
|
|
|
|
acc := NewAccount(t, bc)
|
|
dummyInfo := dummyNodeInfo(acc)
|
|
|
|
tx := PrepareInvoke(t, bc, CommitteeAcc, h, "addPeer", dummyInfo)
|
|
AddBlock(t, bc, tx)
|
|
CheckFault(t, bc, tx.Hash(), "addPeer: witness check failed")
|
|
|
|
tx = PrepareInvoke(t, bc, acc, h, "addPeer", dummyInfo)
|
|
AddBlock(t, bc, tx)
|
|
CheckFault(t, bc, tx.Hash(), "addPeer: alphabet witness check failed")
|
|
|
|
tx = PrepareInvoke(t, bc, []*wallet.Account{CommitteeAcc, acc}, h, "addPeer", dummyInfo)
|
|
AddBlockCheckHalt(t, bc, tx)
|
|
|
|
tx = PrepareInvoke(t, bc, acc, h, "netmapCandidates")
|
|
AddBlock(t, bc, tx)
|
|
|
|
checkNode := func(t *testing.T, info []byte, state int64, actual stackitem.Item) {
|
|
// Actual is netmap.netmapNode.
|
|
require.Equal(t, stackitem.NewStruct([]stackitem.Item{
|
|
stackitem.NewStruct([]stackitem.Item{
|
|
stackitem.NewByteArray(dummyInfo),
|
|
}),
|
|
stackitem.NewBigInteger(big.NewInt(state)),
|
|
}), actual)
|
|
}
|
|
|
|
aer := CheckHalt(t, bc, tx.Hash())
|
|
nodes, _ := aer.Stack[0].Value().([]stackitem.Item)
|
|
require.Equal(t, 1, len(nodes))
|
|
checkNode(t, dummyInfo, 1, nodes[0])
|
|
|
|
dummyInfo[0] ^= 0xFF
|
|
tx = PrepareInvoke(t, bc, []*wallet.Account{CommitteeAcc, acc}, h, "addPeer", dummyInfo)
|
|
AddBlockCheckHalt(t, bc, tx)
|
|
|
|
tx = PrepareInvoke(t, bc, acc, h, "netmapCandidates")
|
|
AddBlock(t, bc, tx)
|
|
|
|
aer = CheckHalt(t, bc, tx.Hash())
|
|
nodes, _ = aer.Stack[0].Value().([]stackitem.Item)
|
|
require.Equal(t, 1, len(nodes))
|
|
checkNode(t, dummyInfo, 1, nodes[0])
|
|
}
|
|
|
|
func TestUpdateState(t *testing.T) {
|
|
bc := NewChain(t)
|
|
h := prepareNetmapContract(t, bc)
|
|
|
|
acc := NewAccount(t, bc)
|
|
dummyInfo := dummyNodeInfo(acc)
|
|
|
|
tx := PrepareInvoke(t, bc, []*wallet.Account{CommitteeAcc, acc}, h, "addPeer", dummyInfo)
|
|
AddBlockCheckHalt(t, bc, tx)
|
|
|
|
t.Run("missing witness", func(t *testing.T) {
|
|
tx = PrepareInvoke(t, bc, acc, h, "updateState", int64(2), acc.PrivateKey().PublicKey().Bytes())
|
|
AddBlock(t, bc, tx)
|
|
CheckFault(t, bc, tx.Hash(), "updateState: alphabet witness check failed")
|
|
|
|
tx = PrepareInvoke(t, bc, CommitteeAcc, h, "updateState", int64(2), acc.PrivateKey().PublicKey().Bytes())
|
|
AddBlock(t, bc, tx)
|
|
CheckFault(t, bc, tx.Hash(), "updateState: witness check failed")
|
|
})
|
|
|
|
tx = PrepareInvoke(t, bc, []*wallet.Account{CommitteeAcc, acc}, h,
|
|
"updateState", int64(2), acc.PrivateKey().PublicKey().Bytes())
|
|
AddBlockCheckHalt(t, bc, tx)
|
|
|
|
tx = PrepareInvoke(t, bc, acc, h, "netmapCandidates")
|
|
AddBlock(t, bc, tx)
|
|
|
|
aer := CheckHalt(t, bc, tx.Hash())
|
|
nodes, ok := aer.Stack[0].Value().([]stackitem.Item)
|
|
require.True(t, ok)
|
|
require.Equal(t, 0, len(nodes))
|
|
}
|