package tests import ( "bytes" "path" "sort" "testing" "git.frostfs.info/TrueCloudLab/frostfs-contract/frostfs" "github.com/nspcc-dev/neo-go/pkg/core/native/nativenames" "github.com/nspcc-dev/neo-go/pkg/crypto/keys" "github.com/nspcc-dev/neo-go/pkg/neotest" "github.com/nspcc-dev/neo-go/pkg/smartcontract" "github.com/nspcc-dev/neo-go/pkg/util" "github.com/nspcc-dev/neo-go/pkg/vm" "github.com/nspcc-dev/neo-go/pkg/vm/stackitem" "github.com/nspcc-dev/neo-go/pkg/wallet" "github.com/stretchr/testify/require" ) const frostfsPath = "../frostfs" func deployFrostFSContract(t *testing.T, e *neotest.Executor, addrProc util.Uint160, pubs keys.PublicKeys, config ...interface{}) util.Uint160 { args := make([]interface{}, 5) args[0] = false args[1] = addrProc arr := make([]interface{}, len(pubs)) for i := range pubs { arr[i] = pubs[i].Bytes() } args[2] = arr args[3] = append([]interface{}{}, config...) c := neotest.CompileFile(t, e.CommitteeHash, frostfsPath, path.Join(frostfsPath, "config.yml")) e.DeployContract(t, c, args) return c.Hash } func newFrostFSInvoker(t *testing.T, n int, config ...interface{}) (*neotest.ContractInvoker, neotest.Signer, keys.PublicKeys) { e := newExecutor(t) accounts := make([]*wallet.Account, n) for i := 0; i < n; i++ { acc, err := wallet.NewAccount() require.NoError(t, err) accounts[i] = acc } sort.Slice(accounts, func(i, j int) bool { p1 := accounts[i].PrivateKey().PublicKey() p2 := accounts[j].PrivateKey().PublicKey() return p1.Cmp(p2) == -1 }) pubs := make(keys.PublicKeys, n) for i := range accounts { pubs[i] = accounts[i].PrivateKey().PublicKey() } m := smartcontract.GetMajorityHonestNodeCount(len(accounts)) for i := range accounts { require.NoError(t, accounts[i].ConvertMultisig(m, pubs.Copy())) } alphabet := neotest.NewMultiSigner(accounts...) h := deployFrostFSContract(t, e, util.Uint160{}, pubs, config...) gasHash, err := e.Chain.GetNativeContractScriptHash(nativenames.Gas) require.NoError(t, err) vc := e.CommitteeInvoker(gasHash).WithSigners(e.Validator) vc.Invoke(t, true, "transfer", e.Validator.ScriptHash(), alphabet.ScriptHash(), int64(10_0000_0000), nil) return e.CommitteeInvoker(h).WithSigners(alphabet), alphabet, pubs } func TestFrostFS_AlphabetList(t *testing.T) { const alphabetSize = 4 e, _, pubs := newFrostFSInvoker(t, alphabetSize) arr := make([]stackitem.Item, len(pubs)) for i := range arr { arr[i] = stackitem.NewStruct([]stackitem.Item{ stackitem.NewByteArray(pubs[i].Bytes()), }) } e.Invoke(t, stackitem.NewArray(arr), "alphabetList") } func TestFrostFS_InnerRingCandidate(t *testing.T) { e, _, _ := newFrostFSInvoker(t, 4, frostfs.CandidateFeeConfigKey, int64(10)) const candidateCount = 3 accs := make([]neotest.Signer, candidateCount) for i := range accs { accs[i] = e.NewAccount(t) } arr := make([]stackitem.Item, candidateCount) pubs := make([][]byte, candidateCount) sort.Slice(accs, func(i, j int) bool { s1 := accs[i].Script() s2 := accs[j].Script() return bytes.Compare(s1, s2) == -1 }) for i, acc := range accs { cAcc := e.WithSigners(acc) pub, ok := vm.ParseSignatureContract(acc.Script()) require.True(t, ok) cAcc.Invoke(t, stackitem.Null{}, "innerRingCandidateAdd", pub) cAcc.InvokeFail(t, "candidate already in the list", "innerRingCandidateAdd", pub) pubs[i] = pub arr[i] = stackitem.NewStruct([]stackitem.Item{stackitem.NewBuffer(pub)}) } e.Invoke(t, stackitem.NewArray(arr), "innerRingCandidates") cAcc := e.WithSigners(accs[1]) cAcc.Invoke(t, stackitem.Null{}, "innerRingCandidateRemove", pubs[1]) e.Invoke(t, stackitem.NewArray([]stackitem.Item{arr[0], arr[2]}), "innerRingCandidates") cAcc = e.WithSigners(accs[2]) cAcc.Invoke(t, stackitem.Null{}, "innerRingCandidateRemove", pubs[2]) e.Invoke(t, stackitem.NewArray([]stackitem.Item{arr[0]}), "innerRingCandidates") cAcc = e.WithSigners(accs[0]) cAcc.Invoke(t, stackitem.Null{}, "innerRingCandidateRemove", pubs[0]) e.Invoke(t, stackitem.NewArray([]stackitem.Item{}), "innerRingCandidates") }