mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-12-28 23:27:36 +00:00
b31d39836a
I had some reason to suspect something is wrong here (but it's all OK). Even though this is kinda tested in TestDeployGetUpdateDestroyContract, it uses internal APIs and with neotest we can ensure it work OK for a complete tx/block environment. So, won't hurt having these tests as well. Signed-off-by: Roman Khimov <roman@nspcc.ru>
134 lines
4.1 KiB
Go
134 lines
4.1 KiB
Go
package native_test
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/nspcc-dev/neo-go/internal/basicchain"
|
|
"github.com/nspcc-dev/neo-go/pkg/config"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/native/nativenames"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/state"
|
|
"github.com/nspcc-dev/neo-go/pkg/neotest"
|
|
"github.com/nspcc-dev/neo-go/pkg/neotest/chain"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract/manifest"
|
|
"github.com/nspcc-dev/neo-go/pkg/smartcontract/nef"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/opcode"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestManagement_GetNEP17Contracts(t *testing.T) {
|
|
t.Run("empty chain", func(t *testing.T) {
|
|
bc, validators, committee := chain.NewMulti(t)
|
|
e := neotest.NewExecutor(t, bc, validators, committee)
|
|
|
|
require.ElementsMatch(t, []util.Uint160{e.NativeHash(t, nativenames.Neo),
|
|
e.NativeHash(t, nativenames.Gas)}, bc.GetNEP17Contracts())
|
|
})
|
|
|
|
t.Run("basic chain", func(t *testing.T) {
|
|
bc, validators, committee := chain.NewMultiWithCustomConfig(t, func(c *config.Blockchain) {
|
|
c.P2PSigExtensions = true // `basicchain.Init` requires Notary enabled
|
|
})
|
|
e := neotest.NewExecutor(t, bc, validators, committee)
|
|
basicchain.Init(t, "../../../", e)
|
|
|
|
require.ElementsMatch(t, []util.Uint160{e.NativeHash(t, nativenames.Neo),
|
|
e.NativeHash(t, nativenames.Gas), e.ContractHash(t, 1)}, bc.GetNEP17Contracts())
|
|
})
|
|
}
|
|
|
|
func TestManagement_DeployUpdate_HFBasilisk(t *testing.T) {
|
|
bc, acc := chain.NewSingleWithCustomConfig(t, func(c *config.Blockchain) {
|
|
c.Hardforks = map[string]uint32{
|
|
config.HFBasilisk.String(): 2,
|
|
}
|
|
})
|
|
e := neotest.NewExecutor(t, bc, acc, acc)
|
|
|
|
ne, err := nef.NewFile([]byte{byte(opcode.JMP), 0x05})
|
|
require.NoError(t, err)
|
|
|
|
m := &manifest.Manifest{
|
|
Name: "ctr",
|
|
ABI: manifest.ABI{
|
|
Methods: []manifest.Method{
|
|
{
|
|
Name: "main",
|
|
Offset: 0,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
ctr := &neotest.Contract{
|
|
|
|
Hash: state.CreateContractHash(e.Validator.ScriptHash(), ne.Checksum, m.Name),
|
|
NEF: ne,
|
|
Manifest: m,
|
|
}
|
|
|
|
// Block 1: no script check on deploy.
|
|
e.DeployContract(t, ctr, nil)
|
|
e.AddNewBlock(t)
|
|
|
|
// Block 3: script check on deploy.
|
|
ctr.Manifest.Name = "other name"
|
|
e.DeployContractCheckFAULT(t, ctr, nil, "invalid contract script: invalid offset 5 ip at 0")
|
|
}
|
|
|
|
func TestManagement_CallInTheSameBlock(t *testing.T) {
|
|
bc, acc := chain.NewSingle(t)
|
|
|
|
e := neotest.NewExecutor(t, bc, acc, acc)
|
|
|
|
ne, err := nef.NewFile([]byte{byte(opcode.PUSH1)})
|
|
require.NoError(t, err)
|
|
|
|
m := &manifest.Manifest{
|
|
Name: "ctr",
|
|
ABI: manifest.ABI{
|
|
Methods: []manifest.Method{
|
|
{
|
|
Name: "main",
|
|
Offset: 0,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
t.Run("same tx", func(t *testing.T) {
|
|
h := state.CreateContractHash(e.Validator.ScriptHash(), ne.Checksum, m.Name)
|
|
|
|
neb, err := ne.Bytes()
|
|
require.NoError(t, err)
|
|
|
|
rawManifest, err := json.Marshal(m)
|
|
require.NoError(t, err)
|
|
|
|
b := smartcontract.NewBuilder()
|
|
b.InvokeWithAssert(bc.ManagementContractHash(), "deploy", neb, rawManifest) // We need to drop the resulting contract and ASSERT does that.
|
|
b.InvokeWithAssert(bc.ManagementContractHash(), "hasMethod", h, "main", 0)
|
|
b.InvokeMethod(h, "main")
|
|
|
|
script, err := b.Script()
|
|
require.NoError(t, err)
|
|
txHash := e.InvokeScript(t, script, []neotest.Signer{e.Validator})
|
|
e.CheckHalt(t, txHash, stackitem.Make(1))
|
|
})
|
|
t.Run("next tx", func(t *testing.T) {
|
|
m.Name = "another contract"
|
|
h := state.CreateContractHash(e.Validator.ScriptHash(), ne.Checksum, m.Name)
|
|
|
|
txDeploy := e.NewDeployTx(t, e.Chain, &neotest.Contract{Hash: h, NEF: ne, Manifest: m}, nil)
|
|
txHasMethod := e.NewTx(t, []neotest.Signer{e.Validator}, bc.ManagementContractHash(), "hasMethod", h, "main", 0)
|
|
|
|
txCall := e.NewUnsignedTx(t, h, "main") // Test invocation doesn't give true GAS cost before deployment.
|
|
txCall = e.SignTx(t, txCall, 1_0000_0000, e.Validator)
|
|
|
|
e.AddNewBlock(t, txDeploy, txHasMethod, txCall)
|
|
e.CheckHalt(t, txHasMethod.Hash(), stackitem.Make(true))
|
|
e.CheckHalt(t, txCall.Hash(), stackitem.Make(1))
|
|
})
|
|
}
|