forked from TrueCloudLab/neoneo-go
299 lines
14 KiB
Go
299 lines
14 KiB
Go
|
package core
|
||
|
|
|
||
|
|
import (
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/nspcc-dev/neo-go/internal/testchain"
|
||
|
|
"github.com/nspcc-dev/neo-go/pkg/core/native"
|
||
|
|
"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/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"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestNotaryContractPipeline(t *testing.T) {
|
||
|
|
chain := newTestChain(t)
|
||
|
|
defer chain.Close()
|
||
|
|
|
||
|
|
notaryHash := chain.contracts.Notary.Hash
|
||
|
|
gasHash := chain.contracts.GAS.Hash
|
||
|
|
depositLock := 30
|
||
|
|
|
||
|
|
// check Notary contract has no GAS on the account
|
||
|
|
checkBalanceOf(t, chain, notaryHash, 0)
|
||
|
|
|
||
|
|
// `balanceOf`: check multisig account has no GAS on deposit
|
||
|
|
balance, err := invokeContractMethod(chain, 100000000, notaryHash, "balanceOf", testchain.MultisigScriptHash())
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, balance, stackitem.Make(0))
|
||
|
|
|
||
|
|
// `expirationOf`: should fail to get deposit which does not exist
|
||
|
|
till, err := invokeContractMethod(chain, 100000000, notaryHash, "expirationOf", testchain.MultisigScriptHash())
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, till, stackitem.Make(0))
|
||
|
|
|
||
|
|
// `lockDepositUntil`: should fail because there's no deposit
|
||
|
|
lockDepositUntilRes, err := invokeContractMethod(chain, 100000000, notaryHash, "lockDepositUntil", testchain.MultisigScriptHash(), int64(depositLock+1))
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, lockDepositUntilRes, stackitem.NewBool(false))
|
||
|
|
|
||
|
|
// `onPayment`: bad token
|
||
|
|
transferTx := transferTokenFromMultisigAccount(t, chain, notaryHash, chain.contracts.NEO.Hash, 1, nil, int64(depositLock))
|
||
|
|
res, err := chain.GetAppExecResults(transferTx.Hash(), trigger.Application)
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkFAULTState(t, &res[0])
|
||
|
|
|
||
|
|
// `onPayment`: insufficient first deposit
|
||
|
|
transferTx = transferTokenFromMultisigAccount(t, chain, notaryHash, gasHash, 2*transaction.NotaryServiceFeePerKey-1, nil, int64(depositLock))
|
||
|
|
res, err = chain.GetAppExecResults(transferTx.Hash(), trigger.Application)
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkFAULTState(t, &res[0])
|
||
|
|
|
||
|
|
// `onPayment`: invalid `data` (missing `till` parameter)
|
||
|
|
transferTx = transferTokenFromMultisigAccount(t, chain, notaryHash, gasHash, 2*transaction.NotaryServiceFeePerKey-1, nil)
|
||
|
|
res, err = chain.GetAppExecResults(transferTx.Hash(), trigger.Application)
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkFAULTState(t, &res[0])
|
||
|
|
|
||
|
|
// `onPayment`: good
|
||
|
|
transferTx = transferTokenFromMultisigAccount(t, chain, notaryHash, gasHash, 2*transaction.NotaryServiceFeePerKey, nil, int64(depositLock))
|
||
|
|
res, err = chain.GetAppExecResults(transferTx.Hash(), trigger.Application)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, vm.HaltState, res[0].VMState)
|
||
|
|
require.Equal(t, 0, len(res[0].Stack))
|
||
|
|
checkBalanceOf(t, chain, notaryHash, 2*transaction.NotaryServiceFeePerKey)
|
||
|
|
|
||
|
|
// `expirationOf`: check `till` was set
|
||
|
|
till, err = invokeContractMethod(chain, 100000000, notaryHash, "expirationOf", testchain.MultisigScriptHash())
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, till, stackitem.Make(depositLock))
|
||
|
|
|
||
|
|
// `balanceOf`: check deposited amount for the multisig account
|
||
|
|
balance, err = invokeContractMethod(chain, 100000000, notaryHash, "balanceOf", testchain.MultisigScriptHash())
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, balance, stackitem.Make(2*transaction.NotaryServiceFeePerKey))
|
||
|
|
|
||
|
|
// `onPayment`: good second deposit and explicit `to` paramenter
|
||
|
|
transferTx = transferTokenFromMultisigAccount(t, chain, notaryHash, gasHash, transaction.NotaryServiceFeePerKey, testchain.MultisigScriptHash(), int64(depositLock+1))
|
||
|
|
res, err = chain.GetAppExecResults(transferTx.Hash(), trigger.Application)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, vm.HaltState, res[0].VMState)
|
||
|
|
require.Equal(t, 0, len(res[0].Stack))
|
||
|
|
checkBalanceOf(t, chain, notaryHash, 3*transaction.NotaryServiceFeePerKey)
|
||
|
|
|
||
|
|
// `balanceOf`: check deposited amount for the multisig account
|
||
|
|
balance, err = invokeContractMethod(chain, 100000000, notaryHash, "balanceOf", testchain.MultisigScriptHash())
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, balance, stackitem.Make(3*transaction.NotaryServiceFeePerKey))
|
||
|
|
|
||
|
|
// `expirationOf`: check `till` is updated.
|
||
|
|
till, err = invokeContractMethod(chain, 100000000, notaryHash, "expirationOf", testchain.MultisigScriptHash())
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, till, stackitem.Make(depositLock+1))
|
||
|
|
|
||
|
|
// `onPayment`: empty payment, should fail because `till` less then the previous one
|
||
|
|
transferTx = transferTokenFromMultisigAccount(t, chain, notaryHash, gasHash, 0, testchain.MultisigScriptHash(), int64(depositLock))
|
||
|
|
res, err = chain.GetAppExecResults(transferTx.Hash(), trigger.Application)
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkFAULTState(t, &res[0])
|
||
|
|
checkBalanceOf(t, chain, notaryHash, 3*transaction.NotaryServiceFeePerKey)
|
||
|
|
till, err = invokeContractMethod(chain, 100000000, notaryHash, "expirationOf", testchain.MultisigScriptHash())
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, till, stackitem.Make(depositLock+1))
|
||
|
|
|
||
|
|
// `onPayment`: empty payment, should fail because `till` less then the chain height
|
||
|
|
transferTx = transferTokenFromMultisigAccount(t, chain, notaryHash, gasHash, 0, testchain.MultisigScriptHash(), int64(1))
|
||
|
|
res, err = chain.GetAppExecResults(transferTx.Hash(), trigger.Application)
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkFAULTState(t, &res[0])
|
||
|
|
checkBalanceOf(t, chain, notaryHash, 3*transaction.NotaryServiceFeePerKey)
|
||
|
|
till, err = invokeContractMethod(chain, 100000000, notaryHash, "expirationOf", testchain.MultisigScriptHash())
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, till, stackitem.Make(depositLock+1))
|
||
|
|
|
||
|
|
// `onPayment`: empty payment, should successfully update `till`
|
||
|
|
transferTx = transferTokenFromMultisigAccount(t, chain, notaryHash, gasHash, 0, testchain.MultisigScriptHash(), int64(depositLock+2))
|
||
|
|
res, err = chain.GetAppExecResults(transferTx.Hash(), trigger.Application)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, vm.HaltState, res[0].VMState)
|
||
|
|
require.Equal(t, 0, len(res[0].Stack))
|
||
|
|
checkBalanceOf(t, chain, notaryHash, 3*transaction.NotaryServiceFeePerKey)
|
||
|
|
till, err = invokeContractMethod(chain, 100000000, notaryHash, "expirationOf", testchain.MultisigScriptHash())
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, till, stackitem.Make(depositLock+2))
|
||
|
|
|
||
|
|
// `lockDepositUntil`: bad witness
|
||
|
|
lockDepositUntilRes, err = invokeContractMethod(chain, 100000000, notaryHash, "lockDepositUntil", util.Uint160{1, 2, 3}, int64(depositLock+5))
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, lockDepositUntilRes, stackitem.NewBool(false))
|
||
|
|
till, err = invokeContractMethod(chain, 100000000, notaryHash, "expirationOf", testchain.MultisigScriptHash())
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, till, stackitem.Make(depositLock+2))
|
||
|
|
|
||
|
|
// `lockDepositUntil`: bad `till` (less then the previous one)
|
||
|
|
lockDepositUntilRes, err = invokeContractMethod(chain, 100000000, notaryHash, "lockDepositUntil", testchain.MultisigScriptHash(), int64(depositLock+1))
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, lockDepositUntilRes, stackitem.NewBool(false))
|
||
|
|
till, err = invokeContractMethod(chain, 100000000, notaryHash, "expirationOf", testchain.MultisigScriptHash())
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, till, stackitem.Make(depositLock+2))
|
||
|
|
|
||
|
|
// `lockDepositUntil`: bad `till` (less then the chain's height)
|
||
|
|
lockDepositUntilRes, err = invokeContractMethod(chain, 100000000, notaryHash, "lockDepositUntil", testchain.MultisigScriptHash(), int64(1))
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, lockDepositUntilRes, stackitem.NewBool(false))
|
||
|
|
till, err = invokeContractMethod(chain, 100000000, notaryHash, "expirationOf", testchain.MultisigScriptHash())
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, till, stackitem.Make(depositLock+2))
|
||
|
|
|
||
|
|
// `lockDepositUntil`: good `till`
|
||
|
|
lockDepositUntilRes, err = invokeContractMethod(chain, 100000000, notaryHash, "lockDepositUntil", testchain.MultisigScriptHash(), int64(depositLock+3))
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, lockDepositUntilRes, stackitem.NewBool(true))
|
||
|
|
till, err = invokeContractMethod(chain, 100000000, notaryHash, "expirationOf", testchain.MultisigScriptHash())
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, till, stackitem.Make(depositLock+3))
|
||
|
|
|
||
|
|
// transfer 1 GAS to the new account for the next test
|
||
|
|
acc, _ := wallet.NewAccount()
|
||
|
|
transferTx = transferTokenFromMultisigAccount(t, chain, acc.PrivateKey().PublicKey().GetScriptHash(), gasHash, 100000000)
|
||
|
|
res, err = chain.GetAppExecResults(transferTx.Hash(), trigger.Application)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, vm.HaltState, res[0].VMState)
|
||
|
|
require.Equal(t, 0, len(res[0].Stack))
|
||
|
|
|
||
|
|
// `withdraw`: bad witness
|
||
|
|
w := io.NewBufBinWriter()
|
||
|
|
emit.AppCallWithOperationAndArgs(w.BinWriter, notaryHash, "withdraw", testchain.MultisigScriptHash(), acc.PrivateKey().PublicKey().GetScriptHash())
|
||
|
|
require.NoError(t, w.Err)
|
||
|
|
script := w.Bytes()
|
||
|
|
withdrawTx := transaction.New(chain.GetConfig().Magic, script, 10000000)
|
||
|
|
withdrawTx.ValidUntilBlock = chain.blockHeight + 1
|
||
|
|
withdrawTx.NetworkFee = 10000000
|
||
|
|
withdrawTx.Signers = []transaction.Signer{
|
||
|
|
{
|
||
|
|
Account: acc.PrivateKey().PublicKey().GetScriptHash(),
|
||
|
|
Scopes: transaction.None,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
err = acc.SignTx(withdrawTx)
|
||
|
|
require.NoError(t, err)
|
||
|
|
b := chain.newBlock(withdrawTx)
|
||
|
|
err = chain.AddBlock(b)
|
||
|
|
require.NoError(t, err)
|
||
|
|
appExecRes, err := chain.GetAppExecResults(withdrawTx.Hash(), trigger.Application)
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, &appExecRes[0], stackitem.NewBool(false))
|
||
|
|
balance, err = invokeContractMethod(chain, 100000000, notaryHash, "balanceOf", testchain.MultisigScriptHash())
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, balance, stackitem.Make(3*transaction.NotaryServiceFeePerKey))
|
||
|
|
|
||
|
|
// `withdraw`: locked deposit
|
||
|
|
withdrawRes, err := invokeContractMethod(chain, 100000000, notaryHash, "withdraw", testchain.MultisigScriptHash(), acc.PrivateKey().PublicKey().GetScriptHash())
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, withdrawRes, stackitem.NewBool(false))
|
||
|
|
balance, err = invokeContractMethod(chain, 100000000, notaryHash, "balanceOf", testchain.MultisigScriptHash())
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, balance, stackitem.Make(3*transaction.NotaryServiceFeePerKey))
|
||
|
|
|
||
|
|
// `withdraw`: unlock deposit and transfer GAS back to owner
|
||
|
|
chain.genBlocks(depositLock)
|
||
|
|
withdrawRes, err = invokeContractMethod(chain, 100000000, notaryHash, "withdraw", testchain.MultisigScriptHash(), testchain.MultisigScriptHash())
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, withdrawRes, stackitem.NewBool(true))
|
||
|
|
balance, err = invokeContractMethod(chain, 100000000, notaryHash, "balanceOf", testchain.MultisigScriptHash())
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, balance, stackitem.Make(0))
|
||
|
|
checkBalanceOf(t, chain, notaryHash, 0)
|
||
|
|
|
||
|
|
// `withdraw`: the second time it should fail, because there's no deposit left
|
||
|
|
withdrawRes, err = invokeContractMethod(chain, 100000000, notaryHash, "withdraw", testchain.MultisigScriptHash(), testchain.MultisigScriptHash())
|
||
|
|
require.NoError(t, err)
|
||
|
|
checkResult(t, withdrawRes, stackitem.NewBool(false))
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestNotaryNodesReward(t *testing.T) {
|
||
|
|
checkReward := func(nKeys int, nNotaryNodes int, spendFullDeposit bool) {
|
||
|
|
chain := newTestChain(t)
|
||
|
|
defer chain.Close()
|
||
|
|
notaryHash := chain.contracts.Notary.Hash
|
||
|
|
gasHash := chain.contracts.GAS.Hash
|
||
|
|
signer := testchain.MultisigScriptHash()
|
||
|
|
var err error
|
||
|
|
|
||
|
|
// set Notary nodes and check their balance
|
||
|
|
notaryNodes := make([]*keys.PrivateKey, nNotaryNodes)
|
||
|
|
notaryNodesPublicKeys := make(keys.PublicKeys, nNotaryNodes)
|
||
|
|
for i := range notaryNodes {
|
||
|
|
notaryNodes[i], err = keys.NewPrivateKey()
|
||
|
|
require.NoError(t, err)
|
||
|
|
notaryNodesPublicKeys[i] = notaryNodes[i].PublicKey()
|
||
|
|
}
|
||
|
|
chain.setNodesByRole(t, true, native.RoleP2PNotary, notaryNodesPublicKeys)
|
||
|
|
for _, notaryNode := range notaryNodesPublicKeys {
|
||
|
|
checkBalanceOf(t, chain, notaryNode.GetScriptHash(), 0)
|
||
|
|
}
|
||
|
|
|
||
|
|
// deposit GAS for `signer` with lock until the next block
|
||
|
|
depositAmount := 100_0000 + (2+int64(nKeys))*transaction.NotaryServiceFeePerKey // sysfee + netfee of the next transaction
|
||
|
|
if !spendFullDeposit {
|
||
|
|
depositAmount += 1_0000
|
||
|
|
}
|
||
|
|
transferTx := transferTokenFromMultisigAccount(t, chain, notaryHash, gasHash, depositAmount, signer, int64(chain.BlockHeight()+1))
|
||
|
|
res, err := chain.GetAppExecResults(transferTx.Hash(), trigger.Application)
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.Equal(t, vm.HaltState, res[0].VMState)
|
||
|
|
require.Equal(t, 0, len(res[0].Stack))
|
||
|
|
|
||
|
|
// send transaction with Notary contract as a sender
|
||
|
|
tx := chain.newTestTx(util.Uint160{}, []byte{byte(opcode.PUSH1)})
|
||
|
|
tx.Attributes = append(tx.Attributes, transaction.Attribute{Type: transaction.NotaryAssistedT, Value: &transaction.NotaryAssisted{NKeys: uint8(nKeys)}})
|
||
|
|
tx.NetworkFee = (2 + int64(nKeys)) * transaction.NotaryServiceFeePerKey
|
||
|
|
tx.Signers = []transaction.Signer{
|
||
|
|
{
|
||
|
|
Account: notaryHash,
|
||
|
|
Scopes: transaction.None,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Account: signer,
|
||
|
|
Scopes: transaction.None,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
data := tx.GetSignedPart()
|
||
|
|
tx.Scripts = []transaction.Witness{
|
||
|
|
{
|
||
|
|
InvocationScript: append([]byte{byte(opcode.PUSHDATA1), 64}, notaryNodes[0].Sign(data)...),
|
||
|
|
},
|
||
|
|
{
|
||
|
|
InvocationScript: testchain.Sign(data),
|
||
|
|
VerificationScript: testchain.MultisigVerificationScript(),
|
||
|
|
},
|
||
|
|
}
|
||
|
|
b := chain.newBlock(tx)
|
||
|
|
require.NoError(t, chain.AddBlock(b))
|
||
|
|
checkBalanceOf(t, chain, notaryHash, int(depositAmount-tx.SystemFee-tx.NetworkFee))
|
||
|
|
for _, notaryNode := range notaryNodesPublicKeys {
|
||
|
|
checkBalanceOf(t, chain, notaryNode.GetScriptHash(), transaction.NotaryServiceFeePerKey*(nKeys+1)/nNotaryNodes)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, spendDeposit := range []bool{true, false} {
|
||
|
|
checkReward(0, 1, spendDeposit)
|
||
|
|
checkReward(0, 2, spendDeposit)
|
||
|
|
checkReward(1, 1, spendDeposit)
|
||
|
|
checkReward(1, 2, spendDeposit)
|
||
|
|
checkReward(1, 3, spendDeposit)
|
||
|
|
checkReward(5, 1, spendDeposit)
|
||
|
|
checkReward(5, 2, spendDeposit)
|
||
|
|
checkReward(5, 6, spendDeposit)
|
||
|
|
checkReward(5, 7, spendDeposit)
|
||
|
|
}
|
||
|
|
}
|