forked from TrueCloudLab/frostfs-contract
273 lines
8.7 KiB
Go
273 lines
8.7 KiB
Go
package tests
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"encoding/json"
|
|
"math/rand"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/config"
|
|
"github.com/nspcc-dev/neo-go/pkg/config/netmode"
|
|
"github.com/nspcc-dev/neo-go/pkg/core"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/block"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/fee"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/native"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/storage"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/transaction"
|
|
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
|
"github.com/nspcc-dev/neo-go/pkg/io"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract/callflag"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract/trigger"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/emit"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/opcode"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
|
"github.com/nspcc-dev/neo-go/pkg/wallet"
|
|
"github.com/stretchr/testify/require"
|
|
"go.uber.org/zap/zaptest"
|
|
)
|
|
|
|
const validatorWIF = "KxyjQ8eUa4FHt3Gvioyt1Wz29cTUrE4eTqX3yFSk1YFCsPL8uNsY"
|
|
|
|
// CommitteeAcc is an account used to sign tx as a committee.
|
|
var CommitteeAcc *wallet.Account
|
|
|
|
func init() {
|
|
CommitteeAcc, _ = wallet.NewAccountFromWIF(validatorWIF)
|
|
pubs := keys.PublicKeys{CommitteeAcc.PrivateKey().PublicKey()}
|
|
err := CommitteeAcc.ConvertMultisig(1, pubs)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
var _nonce uint32
|
|
|
|
func nonce() uint32 {
|
|
_nonce++
|
|
return _nonce
|
|
}
|
|
|
|
// NewChain creates new blockchain instance with a single validator and
|
|
// setups cleanup functions.
|
|
func NewChain(t *testing.T) *core.Blockchain {
|
|
protoCfg := config.ProtocolConfiguration{
|
|
Magic: netmode.UnitTestNet,
|
|
P2PSigExtensions: true,
|
|
SecondsPerBlock: 1,
|
|
StandbyCommittee: []string{hex.EncodeToString(CommitteeAcc.PrivateKey().PublicKey().Bytes())},
|
|
ValidatorsCount: 1,
|
|
VerifyBlocks: true,
|
|
VerifyTransactions: true,
|
|
}
|
|
|
|
st := storage.NewMemoryStore()
|
|
log := zaptest.NewLogger(t)
|
|
bc, err := core.NewBlockchain(st, protoCfg, log)
|
|
require.NoError(t, err)
|
|
go bc.Run()
|
|
t.Cleanup(bc.Close)
|
|
return bc
|
|
}
|
|
|
|
// PrepareInvoke creates new invocation transaction.
|
|
// Signer can be either bool or *wallet.Account.
|
|
// In the first case `true` means sign by committee, `false` means sign by validators.
|
|
func PrepareInvoke(t *testing.T, bc *core.Blockchain, signer interface{},
|
|
hash util.Uint160, method string, args ...interface{}) *transaction.Transaction {
|
|
w := io.NewBufBinWriter()
|
|
emit.AppCall(w.BinWriter, hash, method, callflag.All, args...)
|
|
require.NoError(t, w.Err)
|
|
|
|
script := w.Bytes()
|
|
tx := transaction.New(script, 0)
|
|
tx.Nonce = nonce()
|
|
tx.ValidUntilBlock = bc.BlockHeight() + 1
|
|
|
|
switch s := signer.(type) {
|
|
case *wallet.Account:
|
|
tx.Signers = append(tx.Signers, transaction.Signer{
|
|
Account: s.Contract.ScriptHash(),
|
|
Scopes: transaction.Global,
|
|
})
|
|
require.NoError(t, addNetworkFee(bc, tx, s))
|
|
require.NoError(t, addSystemFee(bc, tx))
|
|
require.NoError(t, s.SignTx(netmode.UnitTestNet, tx))
|
|
case []*wallet.Account:
|
|
for _, acc := range s {
|
|
tx.Signers = append(tx.Signers, transaction.Signer{
|
|
Account: acc.Contract.ScriptHash(),
|
|
Scopes: transaction.Global,
|
|
})
|
|
require.NoError(t, addNetworkFee(bc, tx, acc))
|
|
}
|
|
require.NoError(t, addSystemFee(bc, tx))
|
|
for _, acc := range s {
|
|
require.NoError(t, acc.SignTx(netmode.UnitTestNet, tx))
|
|
}
|
|
default:
|
|
panic("invalid signer")
|
|
}
|
|
|
|
return tx
|
|
}
|
|
|
|
// NewAccount creates new account and transfers 100.0 GAS to it.
|
|
func NewAccount(t *testing.T, bc *core.Blockchain) *wallet.Account {
|
|
acc, err := wallet.NewAccount()
|
|
require.NoError(t, err)
|
|
|
|
tx := PrepareInvoke(t, bc, CommitteeAcc,
|
|
bc.UtilityTokenHash(), "transfer",
|
|
CommitteeAcc.Contract.ScriptHash(), acc.Contract.ScriptHash(), int64(100_0000_0000), nil)
|
|
AddBlock(t, bc, tx)
|
|
CheckHalt(t, bc, tx.Hash())
|
|
return acc
|
|
}
|
|
|
|
// DeployContract compiles and deploys contract to bc.
|
|
// path should contain Go source files.
|
|
// data is an optional argument to `_deploy`.
|
|
func DeployContract(t *testing.T, bc *core.Blockchain, path string, data interface{}) util.Uint160 {
|
|
tx, h := newDeployTx(t, bc, path, data)
|
|
AddBlock(t, bc, tx)
|
|
CheckHalt(t, bc, tx.Hash())
|
|
return h
|
|
}
|
|
|
|
// CheckHalt checks that transaction persisted with HALT state.
|
|
func CheckHalt(t *testing.T, bc *core.Blockchain, h util.Uint256, stack ...stackitem.Item) {
|
|
aer, err := bc.GetAppExecResults(h, trigger.Application)
|
|
require.NoError(t, err)
|
|
require.Equal(t, vm.HaltState, aer[0].VMState, aer[0].FaultException)
|
|
if len(stack) != 0 {
|
|
require.Equal(t, stack, aer[0].Stack)
|
|
}
|
|
}
|
|
|
|
// CheckFault checks that transaction persisted with FAULT state.
|
|
// Raised exception is also checked to contain s as a substring.
|
|
func CheckFault(t *testing.T, bc *core.Blockchain, h util.Uint256, s string) {
|
|
aer, err := bc.GetAppExecResults(h, trigger.Application)
|
|
require.NoError(t, err)
|
|
require.Equal(t, vm.FaultState, aer[0].VMState)
|
|
require.True(t, strings.Contains(aer[0].FaultException, s),
|
|
"expected: %s, got: %s", s, aer[0].FaultException)
|
|
}
|
|
|
|
// newDeployTx returns new deployment tx for contract.
|
|
func newDeployTx(t *testing.T, bc *core.Blockchain, ctrPath string, data interface{}) (*transaction.Transaction, util.Uint160) {
|
|
c, err := ContractInfo(CommitteeAcc.Contract.ScriptHash(), ctrPath)
|
|
require.NoError(t, err)
|
|
|
|
rawManifest, err := json.Marshal(c.Manifest)
|
|
require.NoError(t, err)
|
|
|
|
neb, err := c.NEF.Bytes()
|
|
require.NoError(t, err)
|
|
|
|
buf := io.NewBufBinWriter()
|
|
emit.AppCall(buf.BinWriter, bc.ManagementContractHash(), "deploy", callflag.All, neb, rawManifest, data)
|
|
require.NoError(t, buf.Err)
|
|
|
|
tx := transaction.New(buf.Bytes(), 100*native.GASFactor)
|
|
tx.Nonce = nonce()
|
|
tx.ValidUntilBlock = bc.BlockHeight() + 1
|
|
tx.Signers = []transaction.Signer{{
|
|
Account: CommitteeAcc.Contract.ScriptHash(),
|
|
Scopes: transaction.Global,
|
|
}}
|
|
require.NoError(t, addNetworkFee(bc, tx, CommitteeAcc))
|
|
require.NoError(t, CommitteeAcc.SignTx(netmode.UnitTestNet, tx))
|
|
return tx, c.Hash
|
|
}
|
|
|
|
func addSystemFee(bc *core.Blockchain, tx *transaction.Transaction) error {
|
|
v, _ := TestInvoke(bc, tx) // ignore error to support failing transactions
|
|
tx.SystemFee = v.GasConsumed()
|
|
return nil
|
|
}
|
|
|
|
func addNetworkFee(bc *core.Blockchain, tx *transaction.Transaction, sender *wallet.Account) error {
|
|
size := io.GetVarSize(tx)
|
|
netFee, sizeDelta := fee.Calculate(bc.GetBaseExecFee(), sender.Contract.Script)
|
|
tx.NetworkFee += netFee
|
|
size += sizeDelta
|
|
for _, cosigner := range tx.Signers {
|
|
contract := bc.GetContractState(cosigner.Account)
|
|
if contract != nil {
|
|
netFee, sizeDelta = fee.Calculate(bc.GetBaseExecFee(), contract.NEF.Script)
|
|
tx.NetworkFee += netFee
|
|
size += sizeDelta
|
|
}
|
|
}
|
|
tx.NetworkFee += int64(size) * bc.FeePerByte()
|
|
return nil
|
|
}
|
|
|
|
// AddBlock creates a new block from provided transactions and adds it on bc.
|
|
func AddBlock(t *testing.T, bc *core.Blockchain, txs ...*transaction.Transaction) *block.Block {
|
|
lastBlock, err := bc.GetBlock(bc.GetHeaderHash(int(bc.BlockHeight())))
|
|
require.NoError(t, err)
|
|
b := &block.Block{
|
|
Header: block.Header{
|
|
NextConsensus: CommitteeAcc.Contract.ScriptHash(),
|
|
Script: transaction.Witness{
|
|
VerificationScript: CommitteeAcc.Contract.Script,
|
|
},
|
|
Timestamp: lastBlock.Timestamp + 1,
|
|
},
|
|
Transactions: txs,
|
|
}
|
|
b.PrevHash = lastBlock.Hash()
|
|
b.Index = bc.BlockHeight() + 1
|
|
b.RebuildMerkleRoot()
|
|
|
|
sign := CommitteeAcc.PrivateKey().SignHashable(uint32(netmode.UnitTestNet), b)
|
|
b.Script.InvocationScript = append([]byte{byte(opcode.PUSHDATA1), 64}, sign...)
|
|
require.NoError(t, bc.AddBlock(b))
|
|
|
|
return b
|
|
}
|
|
|
|
// AddBlockCheckHalt is a convenient wrapper over AddBlock and CheckHalt.
|
|
func AddBlockCheckHalt(t *testing.T, bc *core.Blockchain, txs ...*transaction.Transaction) *block.Block {
|
|
b := AddBlock(t, bc, txs...)
|
|
for _, tx := range txs {
|
|
CheckHalt(t, bc, tx.Hash())
|
|
}
|
|
return b
|
|
}
|
|
|
|
// CheckTestInvoke executes transaction without persisting it's state and
|
|
// compares the result with the expected.
|
|
func CheckTestInvoke(t *testing.T, bc *core.Blockchain, tx *transaction.Transaction, expected interface{}) {
|
|
v, err := TestInvoke(bc, tx)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, v.Estack().Len())
|
|
require.Equal(t, stackitem.Make(expected), v.Estack().Pop().Item())
|
|
}
|
|
|
|
// TestInvoke creates a test VM with dummy block and executes transaction in it.
|
|
func TestInvoke(bc *core.Blockchain, tx *transaction.Transaction) (*vm.VM, error) {
|
|
lastBlock, err := bc.GetBlock(bc.GetHeaderHash(int(bc.BlockHeight())))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
b := &block.Block{Header: block.Header{
|
|
Index: bc.BlockHeight() + 1,
|
|
Timestamp: lastBlock.Timestamp + 1,
|
|
}}
|
|
v := bc.GetTestVM(trigger.Application, tx, b)
|
|
v.LoadWithFlags(tx.Script, callflag.All)
|
|
err = v.Run()
|
|
return v, err
|
|
}
|
|
|
|
func randomBytes(n int) []byte {
|
|
a := make([]byte, n)
|
|
rand.Read(a)
|
|
return a
|
|
}
|