forked from TrueCloudLab/neoneo-go
e5b5a3b118
It doesn't trigger the problem fixed in the previous commit because here CreateTxFromScript is used with pre-calculated GAS value from InvokeFunction and thus it's not performing invocation. Still, it's a nice test extension.
84 lines
2 KiB
Go
84 lines
2 KiB
Go
package deploy
|
|
|
|
import (
|
|
"github.com/nspcc-dev/neo-go/cli/testdata/deploy/sub"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/contract"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/iterator"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/runtime"
|
|
"github.com/nspcc-dev/neo-go/pkg/interop/storage"
|
|
)
|
|
|
|
var key = "key"
|
|
|
|
const mgmtKey = "mgmt"
|
|
|
|
func _deploy(data interface{}, isUpdate bool) {
|
|
var value string
|
|
|
|
ctx := storage.GetContext()
|
|
if isUpdate {
|
|
value = "on update"
|
|
} else {
|
|
value = "on create"
|
|
sh := runtime.GetCallingScriptHash()
|
|
storage.Put(ctx, mgmtKey, sh)
|
|
|
|
if data != nil {
|
|
arr := data.([]interface{})
|
|
for i := 0; i < len(arr)-1; i += 2 {
|
|
storage.Put(ctx, arr[i], arr[i+1])
|
|
}
|
|
}
|
|
}
|
|
|
|
storage.Put(ctx, key, value)
|
|
}
|
|
|
|
// Fail just fails.
|
|
func Fail() {
|
|
panic("as expected")
|
|
}
|
|
|
|
// CheckSenderWitness checks sender's witness.
|
|
func CheckSenderWitness() {
|
|
tx := runtime.GetScriptContainer()
|
|
if !runtime.CheckWitness(tx.Sender) {
|
|
panic("not witnessed")
|
|
}
|
|
}
|
|
|
|
// Update updates contract with the new one.
|
|
func Update(script, manifest []byte) {
|
|
ctx := storage.GetReadOnlyContext()
|
|
mgmt := storage.Get(ctx, mgmtKey).(interop.Hash160)
|
|
contract.Call(mgmt, "update", contract.All, script, manifest)
|
|
}
|
|
|
|
// GetValue returns stored value.
|
|
func GetValue() string {
|
|
ctx := storage.GetReadOnlyContext()
|
|
val1 := storage.Get(ctx, key)
|
|
val2 := storage.Get(ctx, sub.Key)
|
|
return val1.(string) + "|" + val2.(string)
|
|
}
|
|
|
|
// GetValueWithKey returns stored value with the specified key.
|
|
func GetValueWithKey(key string) string {
|
|
ctx := storage.GetReadOnlyContext()
|
|
return storage.Get(ctx, key).(string)
|
|
}
|
|
|
|
// TestFind finds items with the specified prefix.
|
|
func TestFind(f storage.FindFlags) []interface{} {
|
|
ctx := storage.GetContext()
|
|
storage.Put(ctx, "findkey1", "value1")
|
|
storage.Put(ctx, "findkey2", "value2")
|
|
|
|
var result []interface{}
|
|
iter := storage.Find(ctx, "findkey", f)
|
|
for iterator.Next(iter) {
|
|
result = append(result, iterator.Value(iter))
|
|
}
|
|
return result
|
|
}
|