forked from TrueCloudLab/frostfs-contract
113 lines
3.7 KiB
Go
113 lines
3.7 KiB
Go
package tests
|
|
|
|
import (
|
|
"bytes"
|
|
"path"
|
|
"sort"
|
|
"testing"
|
|
|
|
"github.com/TrueCloudLab/frostfs-contract/container"
|
|
"github.com/mr-tron/base58"
|
|
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
|
"github.com/nspcc-dev/neo-go/pkg/encoding/address"
|
|
"github.com/nspcc-dev/neo-go/pkg/neotest"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
const frostfsidPath = "../frostfsid"
|
|
|
|
func deployFrostFSIDContract(t *testing.T, e *neotest.Executor, addrNetmap, addrContainer util.Uint160) util.Uint160 {
|
|
args := make([]interface{}, 5)
|
|
args[0] = false
|
|
args[1] = addrNetmap
|
|
args[2] = addrContainer
|
|
|
|
c := neotest.CompileFile(t, e.CommitteeHash, frostfsidPath, path.Join(frostfsidPath, "config.yml"))
|
|
e.DeployContract(t, c, args)
|
|
return c.Hash
|
|
}
|
|
|
|
func newFrostFSIDInvoker(t *testing.T) *neotest.ContractInvoker {
|
|
e := newExecutor(t)
|
|
|
|
ctrNNS := neotest.CompileFile(t, e.CommitteeHash, nnsPath, path.Join(nnsPath, "config.yml"))
|
|
ctrNetmap := neotest.CompileFile(t, e.CommitteeHash, netmapPath, path.Join(netmapPath, "config.yml"))
|
|
ctrBalance := neotest.CompileFile(t, e.CommitteeHash, balancePath, path.Join(balancePath, "config.yml"))
|
|
ctrContainer := neotest.CompileFile(t, e.CommitteeHash, containerPath, path.Join(containerPath, "config.yml"))
|
|
|
|
e.DeployContract(t, ctrNNS, nil)
|
|
deployNetmapContract(t, e, ctrBalance.Hash, ctrContainer.Hash,
|
|
container.RegistrationFeeKey, int64(containerFee),
|
|
container.AliasFeeKey, int64(containerAliasFee))
|
|
deployBalanceContract(t, e, ctrNetmap.Hash, ctrContainer.Hash)
|
|
deployContainerContract(t, e, ctrNetmap.Hash, ctrBalance.Hash, ctrNNS.Hash)
|
|
h := deployFrostFSIDContract(t, e, ctrNetmap.Hash, ctrContainer.Hash)
|
|
return e.CommitteeInvoker(h)
|
|
}
|
|
|
|
func TestFrostFSID_AddKey(t *testing.T) {
|
|
e := newFrostFSIDInvoker(t)
|
|
|
|
pubs := make([][]byte, 6)
|
|
for i := range pubs {
|
|
p, err := keys.NewPrivateKey()
|
|
require.NoError(t, err)
|
|
pubs[i] = p.PublicKey().Bytes()
|
|
}
|
|
acc := e.NewAccount(t)
|
|
owner, _ := base58.Decode(address.Uint160ToString(acc.ScriptHash()))
|
|
e.Invoke(t, stackitem.Null{}, "addKey", owner,
|
|
[]interface{}{pubs[0], pubs[1]})
|
|
|
|
sort.Slice(pubs[:2], func(i, j int) bool {
|
|
return bytes.Compare(pubs[i], pubs[j]) == -1
|
|
})
|
|
arr := []stackitem.Item{
|
|
stackitem.NewBuffer(pubs[0]),
|
|
stackitem.NewBuffer(pubs[1]),
|
|
}
|
|
e.Invoke(t, stackitem.NewArray(arr), "key", owner)
|
|
|
|
t.Run("multiple addKey per block", func(t *testing.T) {
|
|
tx1 := e.PrepareInvoke(t, "addKey", owner, []interface{}{pubs[2]})
|
|
tx2 := e.PrepareInvoke(t, "addKey", owner, []interface{}{pubs[3], pubs[4]})
|
|
e.AddNewBlock(t, tx1, tx2)
|
|
e.CheckHalt(t, tx1.Hash(), stackitem.Null{})
|
|
e.CheckHalt(t, tx2.Hash(), stackitem.Null{})
|
|
|
|
sort.Slice(pubs[:5], func(i, j int) bool {
|
|
return bytes.Compare(pubs[i], pubs[j]) == -1
|
|
})
|
|
arr = []stackitem.Item{
|
|
stackitem.NewBuffer(pubs[0]),
|
|
stackitem.NewBuffer(pubs[1]),
|
|
stackitem.NewBuffer(pubs[2]),
|
|
stackitem.NewBuffer(pubs[3]),
|
|
stackitem.NewBuffer(pubs[4]),
|
|
}
|
|
e.Invoke(t, stackitem.NewArray(arr), "key", owner)
|
|
})
|
|
|
|
e.Invoke(t, stackitem.Null{}, "removeKey", owner,
|
|
[]interface{}{pubs[1], pubs[5]})
|
|
arr = []stackitem.Item{
|
|
stackitem.NewBuffer(pubs[0]),
|
|
stackitem.NewBuffer(pubs[2]),
|
|
stackitem.NewBuffer(pubs[3]),
|
|
stackitem.NewBuffer(pubs[4]),
|
|
}
|
|
e.Invoke(t, stackitem.NewArray(arr), "key", owner)
|
|
|
|
t.Run("multiple removeKey per block", func(t *testing.T) {
|
|
tx1 := e.PrepareInvoke(t, "removeKey", owner, []interface{}{pubs[2]})
|
|
tx2 := e.PrepareInvoke(t, "removeKey", owner, []interface{}{pubs[0], pubs[4]})
|
|
e.AddNewBlock(t, tx1, tx2)
|
|
e.CheckHalt(t, tx1.Hash(), stackitem.Null{})
|
|
e.CheckHalt(t, tx2.Hash(), stackitem.Null{})
|
|
|
|
arr = []stackitem.Item{stackitem.NewBuffer(pubs[3])}
|
|
e.Invoke(t, stackitem.NewArray(arr), "key", owner)
|
|
})
|
|
}
|