195 lines
7.4 KiB
Go
195 lines
7.4 KiB
Go
package native
|
|
|
|
import (
|
|
"math/big"
|
|
"testing"
|
|
|
|
"github.com/tutus-one/tutus-chain/pkg/config"
|
|
"github.com/tutus-one/tutus-chain/pkg/core/dao"
|
|
"github.com/tutus-one/tutus-chain/pkg/core/interop"
|
|
"github.com/tutus-one/tutus-chain/pkg/core/state"
|
|
"github.com/tutus-one/tutus-chain/pkg/core/storage"
|
|
"github.com/tutus-one/tutus-chain/pkg/crypto/keys"
|
|
"github.com/tutus-one/tutus-chain/pkg/smartcontract"
|
|
"github.com/tutus-one/tutus-chain/pkg/smartcontract/manifest"
|
|
"github.com/tutus-one/tutus-chain/pkg/smartcontract/nef"
|
|
"github.com/tutus-one/tutus-chain/pkg/util"
|
|
"github.com/tutus-one/tutus-chain/pkg/vm/opcode"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type annos struct{}
|
|
|
|
func (n annos) Initialize(context *interop.Context, hardfork *config.Hardfork, md *interop.HFSpecificContractMD) error {
|
|
panic("TODO")
|
|
}
|
|
func (n annos) ActiveIn() *config.Hardfork { panic("TODO") }
|
|
func (n annos) InitializeCache(isHardforkEnabled interop.IsHardforkEnabled, blockHeight uint32, d *dao.Simple) error {
|
|
panic("TODO")
|
|
}
|
|
func (n annos) Metadata() *interop.ContractMD { panic("TODO") }
|
|
func (n annos) OnPersist(context *interop.Context) error { panic("TODO") }
|
|
func (n annos) PostPersist(context *interop.Context) error { panic("TODO") }
|
|
func (n annos) GetCommitteeAddress(d *dao.Simple) util.Uint160 { panic("TODO") }
|
|
func (n annos) GetNextBlockValidatorsInternal(d *dao.Simple) keys.PublicKeys { panic("TODO") }
|
|
func (n annos) BalanceOf(d *dao.Simple, acc util.Uint160) (*big.Int, uint32) { panic("TODO") }
|
|
func (n annos) CalculateBonus(ic *interop.Context, acc util.Uint160, endHeight uint32) (*big.Int, error) {
|
|
panic("TODO")
|
|
}
|
|
func (n annos) GetCommitteeMembers(d *dao.Simple) keys.PublicKeys { panic("TODO") }
|
|
func (n annos) ComputeNextBlockValidators(d *dao.Simple) keys.PublicKeys { panic("TODO") }
|
|
func (n annos) GetCandidates(d *dao.Simple) ([]state.Validator, error) { panic("TODO") }
|
|
func (n annos) CheckCommittee(ic *interop.Context) bool { panic("TODO") }
|
|
func (n annos) RevokeVotes(ic *interop.Context, h util.Uint160) error { return nil }
|
|
|
|
func TestDeployGetUpdateDestroyContract(t *testing.T) {
|
|
mgmt := NewManagement()
|
|
p := newPolicy()
|
|
p.Annos = annos{}
|
|
mgmt.Policy = p
|
|
|
|
d := dao.NewSimple(storage.NewMemoryStore(), false)
|
|
ic := &interop.Context{DAO: d}
|
|
err := mgmt.Initialize(ic, nil, nil)
|
|
require.NoError(t, err)
|
|
require.NoError(t, mgmt.Policy.Initialize(&interop.Context{DAO: d}, nil, nil))
|
|
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(ic, 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(ic, sender, ne, manif)
|
|
require.Error(t, err)
|
|
|
|
// Different sender.
|
|
sender2 := util.Uint160{3, 2, 1}
|
|
contract2, err := mgmt.Deploy(ic, 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 := GetContract(d, mgmt.ID, h)
|
|
require.NoError(t, err)
|
|
require.Equal(t, contract, refContract)
|
|
|
|
refContract, err = GetContractByID(d, mgmt.ID, contract.ID)
|
|
require.NoError(t, err)
|
|
require.Equal(t, contract, refContract)
|
|
|
|
upContract, err := mgmt.Update(ic, h, ne, manif)
|
|
refContract.UpdateCounter++
|
|
require.NoError(t, err)
|
|
require.Equal(t, refContract, upContract)
|
|
|
|
_, err = mgmt.Destroy(ic, h)
|
|
require.NoError(t, err)
|
|
_, err = GetContract(d, mgmt.ID, h)
|
|
require.Error(t, err)
|
|
_, err = GetContractByID(d, mgmt.ID, contract.ID)
|
|
require.Error(t, err)
|
|
}
|
|
|
|
func TestManagement_Initialize(t *testing.T) {
|
|
t.Run("good", func(t *testing.T) {
|
|
d := dao.NewSimple(storage.NewMemoryStore(), false)
|
|
mgmt := NewManagement()
|
|
require.NoError(t, mgmt.InitializeCache(func(hf *config.Hardfork, blockHeight uint32) bool { return false }, 0, d))
|
|
})
|
|
t.Run("invalid contract state", func(t *testing.T) {
|
|
d := dao.NewSimple(storage.NewMemoryStore(), false)
|
|
mgmt := NewManagement()
|
|
d.PutStorageItem(mgmt.Metadata().ID, []byte{PrefixContract}, state.StorageItem{0xFF})
|
|
require.Error(t, mgmt.InitializeCache(func(hf *config.Hardfork, blockHeight uint32) bool { return false }, 0, d))
|
|
})
|
|
}
|
|
|
|
func TestManagement_GetNEP17Contracts(t *testing.T) {
|
|
mgmt := NewManagement()
|
|
mgmt.Policy = newPolicy()
|
|
d := dao.NewSimple(storage.NewMemoryStore(), false)
|
|
err := mgmt.Initialize(&interop.Context{DAO: d}, nil, nil)
|
|
require.NoError(t, err)
|
|
require.NoError(t, mgmt.Policy.Initialize(&interop.Context{DAO: d}, nil, nil))
|
|
err = mgmt.InitializeCache(func(hf *config.Hardfork, blockHeight uint32) bool { return false }, 0, d)
|
|
require.NoError(t, err)
|
|
|
|
require.Empty(t, mgmt.GetNEP17Contracts(d))
|
|
private := d.GetPrivate()
|
|
ic := &interop.Context{DAO: private}
|
|
|
|
// 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(ic, sender, ne, manif)
|
|
require.NoError(t, err)
|
|
|
|
// c1 contract hash should be returned, as private DAO already contains changed cache.
|
|
require.Equal(t, []util.Uint160{c1.Hash}, mgmt.GetNEP17Contracts(private))
|
|
|
|
// Lower DAO still shouldn't contain c1, as no Persist was called.
|
|
require.Empty(t, mgmt.GetNEP17Contracts(d))
|
|
|
|
// Call Persist, check c1 contract hash is returned
|
|
_, err = private.Persist()
|
|
require.NoError(t, err)
|
|
require.Equal(t, []util.Uint160{c1.Hash}, mgmt.GetNEP17Contracts(d))
|
|
|
|
// Update contract
|
|
private = d.GetPrivate()
|
|
manif.ABI.Methods = append(manif.ABI.Methods, manifest.Method{
|
|
Name: "dummy2",
|
|
ReturnType: smartcontract.VoidType,
|
|
Parameters: []manifest.Parameter{},
|
|
})
|
|
c1Updated, err := mgmt.Update(&interop.Context{DAO: private}, c1.Hash, ne, manif)
|
|
require.NoError(t, err)
|
|
require.Equal(t, c1.Hash, c1Updated.Hash)
|
|
|
|
// No changes expected in lower store.
|
|
require.Equal(t, []util.Uint160{c1.Hash}, mgmt.GetNEP17Contracts(d))
|
|
c1Lower, err := GetContract(d, mgmt.ID, c1.Hash)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, len(c1Lower.Manifest.ABI.Methods))
|
|
require.Equal(t, []util.Uint160{c1Updated.Hash}, mgmt.GetNEP17Contracts(private))
|
|
c1Upper, err := GetContract(private, mgmt.ID, c1Updated.Hash)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 2, len(c1Upper.Manifest.ABI.Methods))
|
|
|
|
// Call Persist, check c1Updated state is returned from lower.
|
|
_, err = private.Persist()
|
|
require.NoError(t, err)
|
|
require.Equal(t, []util.Uint160{c1.Hash}, mgmt.GetNEP17Contracts(d))
|
|
c1Lower, err = GetContract(d, mgmt.ID, c1.Hash)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 2, len(c1Lower.Manifest.ABI.Methods))
|
|
}
|