mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-12-23 23:25:22 +00:00
1518019be8
Some commands don't accept arguments, but users try giving them and don't notice a mistake. It's a bit more user-friendly to tell the user that there is something wrong with the way he tries to use the command.
251 lines
7.7 KiB
Go
251 lines
7.7 KiB
Go
package main
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"encoding/json"
|
|
"fmt"
|
|
"math/big"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/crypto/hash"
|
|
"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/neorpc/result"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/vmstate"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// Test signing of multisig transactions.
|
|
// 1. Transfer funds to a created multisig address.
|
|
// 2. Transfer from multisig to another account.
|
|
func TestSignMultisigTx(t *testing.T) {
|
|
e := newExecutor(t, true)
|
|
|
|
privs, pubs := generateKeys(t, 3)
|
|
script, err := smartcontract.CreateMultiSigRedeemScript(2, pubs)
|
|
require.NoError(t, err)
|
|
multisigHash := hash.Hash160(script)
|
|
multisigAddr := address.Uint160ToString(multisigHash)
|
|
|
|
// Create 2 wallets participating in multisig.
|
|
tmpDir := t.TempDir()
|
|
wallet1Path := filepath.Join(tmpDir, "multiWallet1.json")
|
|
wallet2Path := filepath.Join(tmpDir, "multiWallet2.json")
|
|
|
|
addAccount := func(w string, wif string) {
|
|
e.Run(t, "neo-go", "wallet", "init", "--wallet", w)
|
|
e.In.WriteString("acc\rpass\rpass\r")
|
|
e.Run(t, "neo-go", "wallet", "import-multisig",
|
|
"--wallet", w,
|
|
"--wif", wif,
|
|
"--min", "2",
|
|
hex.EncodeToString(pubs[0].Bytes()),
|
|
hex.EncodeToString(pubs[1].Bytes()),
|
|
hex.EncodeToString(pubs[2].Bytes()))
|
|
}
|
|
addAccount(wallet1Path, privs[0].WIF())
|
|
addAccount(wallet2Path, privs[1].WIF())
|
|
|
|
// Transfer funds to the multisig.
|
|
e.In.WriteString("one\r")
|
|
e.Run(t, "neo-go", "wallet", "nep17", "multitransfer",
|
|
"--rpc-endpoint", "http://"+e.RPC.Addr,
|
|
"--wallet", validatorWallet,
|
|
"--from", validatorAddr,
|
|
"--force",
|
|
"NEO:"+multisigAddr+":4",
|
|
"GAS:"+multisigAddr+":1")
|
|
e.checkTxPersisted(t)
|
|
|
|
// Sign and transfer funds to another account.
|
|
priv, err := keys.NewPrivateKey()
|
|
require.NoError(t, err)
|
|
|
|
t.Run("bad cases", func(t *testing.T) {
|
|
txPath := filepath.Join(tmpDir, "multisigtx.json")
|
|
t.Cleanup(func() {
|
|
os.Remove(txPath)
|
|
})
|
|
e.In.WriteString("pass\r")
|
|
e.Run(t, "neo-go", "wallet", "nep17", "transfer",
|
|
"--rpc-endpoint", "http://"+e.RPC.Addr,
|
|
"--wallet", wallet1Path, "--from", multisigAddr,
|
|
"--to", priv.Address(), "--token", "NEO", "--amount", "1",
|
|
"--out", txPath)
|
|
|
|
// missing wallet
|
|
e.RunWithError(t, "neo-go", "wallet", "sign")
|
|
|
|
// missing in
|
|
e.RunWithError(t, "neo-go", "wallet", "sign",
|
|
"--wallet", wallet2Path)
|
|
|
|
// missing address
|
|
e.RunWithError(t, "neo-go", "wallet", "sign",
|
|
"--wallet", wallet2Path,
|
|
"--in", txPath)
|
|
|
|
// invalid address
|
|
e.RunWithError(t, "neo-go", "wallet", "sign",
|
|
"--wallet", wallet2Path, "--address", util.Uint160{}.StringLE(),
|
|
"--in", txPath)
|
|
|
|
// invalid out
|
|
e.In.WriteString("pass\r")
|
|
e.RunWithError(t, "neo-go", "wallet", "sign",
|
|
"--rpc-endpoint", "http://"+e.RPC.Addr,
|
|
"--wallet", wallet2Path, "--address", multisigAddr,
|
|
"--in", txPath, "--out", t.TempDir())
|
|
|
|
// invalid RPC endpoint
|
|
e.In.WriteString("pass\r")
|
|
e.RunWithError(t, "neo-go", "wallet", "sign",
|
|
"--rpc-endpoint", "http://not-an-address",
|
|
"--wallet", wallet2Path, "--address", multisigAddr,
|
|
"--in", txPath)
|
|
})
|
|
|
|
// Create transaction and save it for further multisigning.
|
|
txPath := filepath.Join(tmpDir, "multisigtx.json")
|
|
t.Cleanup(func() {
|
|
os.Remove(txPath)
|
|
})
|
|
e.In.WriteString("pass\r")
|
|
e.Run(t, "neo-go", "wallet", "nep17", "transfer",
|
|
"--rpc-endpoint", "http://"+e.RPC.Addr,
|
|
"--wallet", wallet1Path, "--from", multisigAddr,
|
|
"--to", priv.Address(), "--token", "NEO", "--amount", "1",
|
|
"--out", txPath)
|
|
|
|
simplePriv, err := keys.NewPrivateKey()
|
|
require.NoError(t, err)
|
|
{ // Simple signer, not in signers.
|
|
e.In.WriteString("acc\rpass\rpass\r")
|
|
e.Run(t, "neo-go", "wallet", "import",
|
|
"--wallet", wallet1Path,
|
|
"--wif", simplePriv.WIF())
|
|
t.Run("sign with missing signer", func(t *testing.T) {
|
|
e.In.WriteString("pass\r")
|
|
e.RunWithError(t, "neo-go", "wallet", "sign",
|
|
"--wallet", wallet1Path, "--address", simplePriv.Address(),
|
|
"--in", txPath, "--out", txPath)
|
|
})
|
|
}
|
|
|
|
t.Run("test invoke", func(t *testing.T) {
|
|
t.Run("missing file", func(t *testing.T) {
|
|
e.RunWithError(t, "neo-go", "util", "txdump")
|
|
fmt.Println(e.Out.String())
|
|
})
|
|
|
|
t.Run("no invoke", func(t *testing.T) {
|
|
e.Run(t, "neo-go", "util", "txdump", txPath)
|
|
e.checkTxTestInvokeOutput(t, 11)
|
|
e.checkEOF(t)
|
|
})
|
|
|
|
t.Run("excessive parameters", func(t *testing.T) {
|
|
e.RunWithError(t, "neo-go", "util", "txdump",
|
|
"--rpc-endpoint", "http://"+e.RPC.Addr,
|
|
txPath, "garbage")
|
|
})
|
|
e.Run(t, "neo-go", "util", "txdump",
|
|
"--rpc-endpoint", "http://"+e.RPC.Addr,
|
|
txPath)
|
|
e.checkTxTestInvokeOutput(t, 11)
|
|
res := new(result.Invoke)
|
|
require.NoError(t, json.Unmarshal(e.Out.Bytes(), res))
|
|
require.Equal(t, vmstate.Halt.String(), res.State, res.FaultException)
|
|
})
|
|
|
|
e.In.WriteString("pass\r")
|
|
e.Run(t, "neo-go", "wallet", "sign",
|
|
"--rpc-endpoint", "http://"+e.RPC.Addr,
|
|
"--wallet", wallet2Path, "--address", multisigAddr,
|
|
"--in", txPath, "--out", txPath)
|
|
e.checkTxPersisted(t)
|
|
|
|
t.Run("double-sign", func(t *testing.T) {
|
|
e.In.WriteString("pass\r")
|
|
e.RunWithError(t, "neo-go", "wallet", "sign",
|
|
"--rpc-endpoint", "http://"+e.RPC.Addr,
|
|
"--wallet", wallet2Path, "--address", multisigAddr,
|
|
"--in", txPath, "--out", txPath)
|
|
})
|
|
|
|
b, _ := e.Chain.GetGoverningTokenBalance(priv.GetScriptHash())
|
|
require.Equal(t, big.NewInt(1), b)
|
|
b, _ = e.Chain.GetGoverningTokenBalance(multisigHash)
|
|
require.Equal(t, big.NewInt(3), b)
|
|
|
|
t.Run("via invokefunction", func(t *testing.T) {
|
|
h := deployVerifyContract(t, e)
|
|
|
|
e.In.WriteString("acc\rpass\rpass\r")
|
|
e.Run(t, "neo-go", "wallet", "import-deployed",
|
|
"--rpc-endpoint", "http://"+e.RPC.Addr,
|
|
"--wallet", wallet1Path, "--wif", simplePriv.WIF(),
|
|
"--contract", h.StringLE())
|
|
|
|
e.In.WriteString("pass\r")
|
|
e.Run(t, "neo-go", "contract", "invokefunction",
|
|
"--rpc-endpoint", "http://"+e.RPC.Addr,
|
|
"--wallet", wallet1Path, "--address", multisigHash.StringLE(), // test with scripthash instead of address
|
|
"--out", txPath,
|
|
e.Chain.GoverningTokenHash().StringLE(), "transfer",
|
|
"bytes:"+multisigHash.StringBE(),
|
|
"bytes:"+priv.GetScriptHash().StringBE(),
|
|
"int:1", "bytes:",
|
|
"--", multisigHash.StringLE()+":"+"Global",
|
|
h.StringLE(),
|
|
simplePriv.GetScriptHash().StringLE(),
|
|
)
|
|
|
|
e.In.WriteString("pass\r")
|
|
e.Run(t, "neo-go", "wallet", "sign",
|
|
"--wallet", wallet2Path, "--address", multisigAddr,
|
|
"--in", txPath, "--out", txPath)
|
|
|
|
// Simple signer, not in signers.
|
|
e.In.WriteString("pass\r")
|
|
e.Run(t, "neo-go", "wallet", "sign",
|
|
"--wallet", wallet1Path, "--address", simplePriv.Address(),
|
|
"--in", txPath, "--out", txPath)
|
|
|
|
// Contract.
|
|
e.In.WriteString("pass\r")
|
|
e.Run(t, "neo-go", "wallet", "sign",
|
|
"--rpc-endpoint", "http://"+e.RPC.Addr,
|
|
"--wallet", wallet1Path, "--address", address.Uint160ToString(h),
|
|
"--in", txPath, "--out", txPath)
|
|
tx, _ := e.checkTxPersisted(t)
|
|
require.Equal(t, 3, len(tx.Signers))
|
|
|
|
b, _ := e.Chain.GetGoverningTokenBalance(priv.GetScriptHash())
|
|
require.Equal(t, big.NewInt(2), b)
|
|
b, _ = e.Chain.GetGoverningTokenBalance(multisigHash)
|
|
require.Equal(t, big.NewInt(2), b)
|
|
})
|
|
}
|
|
|
|
func (e *executor) checkTxTestInvokeOutput(t *testing.T, scriptSize int) {
|
|
e.checkNextLine(t, `Hash:\s+`)
|
|
e.checkNextLine(t, `OnChain:\s+false`)
|
|
e.checkNextLine(t, `ValidUntil:\s+\d+`)
|
|
e.checkNextLine(t, `Signer:\s+\w+`)
|
|
e.checkNextLine(t, `SystemFee:\s+(\d|\.)+`)
|
|
e.checkNextLine(t, `NetworkFee:\s+(\d|\.)+`)
|
|
e.checkNextLine(t, `Script:\s+\w+`)
|
|
e.checkScriptDump(t, scriptSize)
|
|
}
|
|
|
|
func (e *executor) checkScriptDump(t *testing.T, scriptSize int) {
|
|
e.checkNextLine(t, `INDEX\s+`)
|
|
for i := 0; i < scriptSize; i++ {
|
|
e.checkNextLine(t, `\d+\s+\w+`)
|
|
}
|
|
}
|