mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-12-12 01:10:36 +00:00
9d2ef775cf
They never return errors, so their interface should reflect that. This allows to remove quite a lot of useless and never tested code. Notice that Get still does return an error. It can be made not to do that, but usually we need to differentiate between successful/unsuccessful accesses anyway, so this doesn't help much.
132 lines
4.3 KiB
Go
132 lines
4.3 KiB
Go
package native
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/core/dao"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/interop"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/state"
|
|
"github.com/nspcc-dev/neo-go/pkg/core/storage"
|
|
"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/stretchr/testify/require"
|
|
)
|
|
|
|
func TestDeployGetUpdateDestroyContract(t *testing.T) {
|
|
mgmt := newManagement()
|
|
d := dao.NewSimple(storage.NewMemoryStore(), false, false)
|
|
err := mgmt.Initialize(&interop.Context{DAO: d})
|
|
require.NoError(t, err)
|
|
script := []byte{byte(opcode.RET)}
|
|
sender := util.Uint160{1, 2, 3}
|
|
ne, err := nef.NewFile(script)
|
|
require.NoError(t, err)
|
|
manif := manifest.NewManifest("Test")
|
|
manif.ABI.Methods = append(manif.ABI.Methods, manifest.Method{
|
|
Name: "dummy",
|
|
ReturnType: smartcontract.VoidType,
|
|
Parameters: []manifest.Parameter{},
|
|
})
|
|
|
|
h := state.CreateContractHash(sender, ne.Checksum, manif.Name)
|
|
|
|
contract, err := mgmt.Deploy(d, sender, ne, manif)
|
|
require.NoError(t, err)
|
|
require.Equal(t, int32(1), contract.ID)
|
|
require.Equal(t, uint16(0), contract.UpdateCounter)
|
|
require.Equal(t, h, contract.Hash)
|
|
require.Equal(t, ne, &contract.NEF)
|
|
require.Equal(t, *manif, contract.Manifest)
|
|
|
|
// Double deploy.
|
|
_, err = mgmt.Deploy(d, sender, ne, manif)
|
|
require.Error(t, err)
|
|
|
|
// Different sender.
|
|
sender2 := util.Uint160{3, 2, 1}
|
|
contract2, err := mgmt.Deploy(d, sender2, ne, manif)
|
|
require.NoError(t, err)
|
|
require.Equal(t, int32(2), contract2.ID)
|
|
require.Equal(t, uint16(0), contract2.UpdateCounter)
|
|
require.Equal(t, state.CreateContractHash(sender2, ne.Checksum, manif.Name), contract2.Hash)
|
|
require.Equal(t, ne, &contract2.NEF)
|
|
require.Equal(t, *manif, contract2.Manifest)
|
|
|
|
refContract, err := mgmt.GetContract(d, h)
|
|
require.NoError(t, err)
|
|
require.Equal(t, contract, refContract)
|
|
|
|
upContract, err := mgmt.Update(d, h, ne, manif)
|
|
refContract.UpdateCounter++
|
|
require.NoError(t, err)
|
|
require.Equal(t, refContract, upContract)
|
|
|
|
err = mgmt.Destroy(d, h)
|
|
require.NoError(t, err)
|
|
_, err = mgmt.GetContract(d, h)
|
|
require.Error(t, err)
|
|
}
|
|
|
|
func TestManagement_Initialize(t *testing.T) {
|
|
t.Run("good", func(t *testing.T) {
|
|
d := dao.NewSimple(storage.NewMemoryStore(), false, false)
|
|
mgmt := newManagement()
|
|
require.NoError(t, mgmt.InitializeCache(d))
|
|
})
|
|
t.Run("invalid contract state", func(t *testing.T) {
|
|
d := dao.NewSimple(storage.NewMemoryStore(), false, false)
|
|
mgmt := newManagement()
|
|
d.PutStorageItem(mgmt.ID, []byte{prefixContract}, state.StorageItem{0xFF})
|
|
require.Error(t, mgmt.InitializeCache(d))
|
|
})
|
|
}
|
|
|
|
func TestManagement_GetNEP17Contracts(t *testing.T) {
|
|
mgmt := newManagement()
|
|
d := dao.NewSimple(storage.NewMemoryStore(), false, false)
|
|
err := mgmt.Initialize(&interop.Context{DAO: d})
|
|
require.NoError(t, err)
|
|
|
|
require.Empty(t, mgmt.GetNEP17Contracts())
|
|
|
|
// Deploy NEP-17 contract
|
|
script := []byte{byte(opcode.RET)}
|
|
sender := util.Uint160{1, 2, 3}
|
|
ne, err := nef.NewFile(script)
|
|
require.NoError(t, err)
|
|
manif := manifest.NewManifest("Test")
|
|
manif.ABI.Methods = append(manif.ABI.Methods, manifest.Method{
|
|
Name: "dummy",
|
|
ReturnType: smartcontract.VoidType,
|
|
Parameters: []manifest.Parameter{},
|
|
})
|
|
manif.SupportedStandards = []string{manifest.NEP17StandardName}
|
|
c1, err := mgmt.Deploy(d, sender, ne, manif)
|
|
require.NoError(t, err)
|
|
|
|
// PostPersist is not yet called, thus no NEP-17 contracts are expected
|
|
require.Empty(t, mgmt.GetNEP17Contracts())
|
|
|
|
// Call PostPersist, check c1 contract hash is returned
|
|
require.NoError(t, mgmt.PostPersist(&interop.Context{DAO: d}))
|
|
require.Equal(t, []util.Uint160{c1.Hash}, mgmt.GetNEP17Contracts())
|
|
|
|
// Update contract
|
|
manif.ABI.Methods = append(manif.ABI.Methods, manifest.Method{
|
|
Name: "dummy2",
|
|
ReturnType: smartcontract.VoidType,
|
|
Parameters: []manifest.Parameter{},
|
|
})
|
|
c2, err := mgmt.Update(d, c1.Hash, ne, manif)
|
|
require.NoError(t, err)
|
|
|
|
// No changes expected before PostPersist call.
|
|
require.Equal(t, []util.Uint160{c1.Hash}, mgmt.GetNEP17Contracts())
|
|
|
|
// Call PostPersist, check c2 contract hash is returned
|
|
require.NoError(t, mgmt.PostPersist(&interop.Context{DAO: d}))
|
|
require.Equal(t, []util.Uint160{c2.Hash}, mgmt.GetNEP17Contracts())
|
|
}
|