[#154] netmap: allow only alphabet calls in updateState

This reverts commit a9503b5c
This commit is contained in:
Leonard Lyubich 2021-11-15 09:29:08 +03:00 committed by Alex Vanin
parent 8553320e1c
commit 68eac87e64
2 changed files with 49 additions and 18 deletions

View file

@ -90,3 +90,36 @@ func TestAddPeer(t *testing.T) {
require.Equal(t, stackitem.NewArray([]stackitem.Item{stackitem.NewByteArray(dummyInfo)}),
aer.Events[0].Item)
}
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))
}