frostfs-contract/tests/netmap_test.go

package tests

import (
	"math/big"
	"math/rand"
	"testing"

	"github.com/nspcc-dev/neo-go/pkg/core"
	"github.com/nspcc-dev/neo-go/pkg/encoding/bigint"
	"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/nspcc-dev/neofs-contract/container"
	"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, config ...interface{}) 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, config...)
}

func TestDeploySetConfig(t *testing.T) {
	bc := NewChain(t)
	h := prepareNetmapContract(t, bc, "SomeKey", "TheValue", container.AliasFeeKey, int64(123))

	tx := PrepareInvoke(t, bc, CommitteeAcc, h, "config", "SomeKey")
	CheckTestInvoke(t, bc, tx, "TheValue")

	tx = PrepareInvoke(t, bc, CommitteeAcc, h, "config", container.AliasFeeKey)
	CheckTestInvoke(t, bc, tx, bigint.ToBytes(big.NewInt(123)))
}

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))
}