mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-11-22 19:29:39 +00:00
266 lines
8.8 KiB
Go
266 lines
8.8 KiB
Go
package wallet_test
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"math/big"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/nspcc-dev/neo-go/internal/testcli"
|
|
"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/smartcontract/context"
|
|
"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 := testcli.NewExecutor(t, true)
|
|
|
|
privs, pubs := testcli.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",
|
|
pubs[0].StringCompressed(),
|
|
pubs[1].StringCompressed(),
|
|
pubs[2].StringCompressed())
|
|
}
|
|
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.Addresses()[0],
|
|
"--wallet", testcli.ValidatorWallet,
|
|
"--from", testcli.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.Addresses()[0],
|
|
"--wallet", wallet1Path, "--from", multisigAddr,
|
|
"--to", priv.Address(), "--token", "NEO", "--amount", "1",
|
|
"--out", txPath)
|
|
|
|
// missing wallet
|
|
e.RunWithErrorCheck(t, `Required flag "in" not set`, "neo-go", "wallet", "sign")
|
|
|
|
// missing in
|
|
e.RunWithErrorCheck(t, `Required flag "in" not set`, "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.Addresses()[0],
|
|
"--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.Addresses()[0],
|
|
"--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.Addresses()[0],
|
|
txPath, "garbage")
|
|
})
|
|
e.Run(t, "neo-go", "util", "txdump",
|
|
"--rpc-endpoint", "http://"+e.RPC.Addresses()[0],
|
|
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)
|
|
})
|
|
|
|
t.Run("console output", func(t *testing.T) {
|
|
oldIn, err := os.ReadFile(txPath)
|
|
require.NoError(t, err)
|
|
e.In.WriteString("pass\r")
|
|
e.Run(t, "neo-go", "wallet", "sign",
|
|
"--wallet", wallet2Path, "--address", multisigAddr,
|
|
"--in", txPath)
|
|
newIn, err := os.ReadFile(txPath)
|
|
require.NoError(t, err)
|
|
require.Equal(t, oldIn, newIn)
|
|
|
|
pcOld := new(context.ParameterContext)
|
|
require.NoError(t, json.Unmarshal(oldIn, pcOld))
|
|
|
|
jOut := e.Out.Bytes()
|
|
pcNew := new(context.ParameterContext)
|
|
require.NoError(t, json.Unmarshal(jOut, pcNew))
|
|
|
|
require.Equal(t, pcOld.Type, pcNew.Type)
|
|
require.Equal(t, pcOld.Network, pcNew.Network)
|
|
require.Equal(t, pcOld.Verifiable, pcNew.Verifiable)
|
|
require.Equal(t, pcOld.Items[multisigHash].Script, pcNew.Items[multisigHash].Script)
|
|
// It's completely signed after this, so parameters have signatures now as well.
|
|
require.NotEqual(t, pcOld.Items[multisigHash].Parameters, pcNew.Items[multisigHash].Parameters)
|
|
require.NotEqual(t, pcOld.Items[multisigHash].Signatures, pcNew.Items[multisigHash].Signatures)
|
|
})
|
|
|
|
t.Run("sign, save and send", func(t *testing.T) {
|
|
e.In.WriteString("pass\r")
|
|
e.Run(t, "neo-go", "wallet", "sign",
|
|
"--rpc-endpoint", "http://"+e.RPC.Addresses()[0],
|
|
"--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.Addresses()[0],
|
|
"--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.Addresses()[0],
|
|
"--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.Addresses()[0],
|
|
"--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.Addresses()[0],
|
|
"--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 deployVerifyContract(t *testing.T, e *testcli.Executor) util.Uint160 {
|
|
return testcli.DeployContract(t, e, "../smartcontract/testdata/verify.go", "../smartcontract/testdata/verify.yml", testcli.ValidatorWallet, testcli.ValidatorAddr, testcli.ValidatorPass)
|
|
}
|