TestSetPrevRealm/TestSetPrevAddr

gnolang · Mar 15, 2024 · 0453cf8 · 0453cf8
1 parent 01e91be
commit 0453cf8
Show file tree

Hide file tree

Showing 3 changed files with 57 additions and 0 deletions.
diff --git a/gnovm/tests/stdlibs/native.go b/gnovm/tests/stdlibs/native.go
diff --git a/gnovm/tests/stdlibs/std/std.gno b/gnovm/tests/stdlibs/std/std.gno
@@ -6,6 +6,8 @@ func TestCurrentRealm() string                 // injected
 func TestSkipHeights(count int64)              // injected
 func ClearStoreCache()                         // injected
 func GetCallerAt(n int) Address                // injected
+func TestSetPrevRealm(pkgPath string)          // injected
+func TestSetPrevAddr(addr Address)             // injected
 func TestSetOrigCaller(addr Address)           // injected
 func TestSetOrigPkgAddr(addr Address)          // injected
 func TestSetOrigSend(sent, spent Coins)        // injected

diff --git a/gnovm/tests/stdlibs/std/std.go b/gnovm/tests/stdlibs/std/std.go
@@ -88,6 +88,21 @@ func GetCallerAt(m *gno.Machine, n int) crypto.Bech32Address {
 	return m.LastCallFrame(n).LastPackage.GetPkgAddr().Bech32()
 }
 
+func TestSetPrevRealm(m *gno.Machine, pkgPath string) {
+	m.Frames[m.NumFrames()-1].LastPackage = &gno.PackageValue{
+		PkgPath: pkgPath,
+	}
+}
+
+func TestSetPrevAddr(m *gno.Machine, addr crypto.Bech32Address) {
+	// clear prev realm in stack frame just in case
+	m.Frames[m.NumFrames()-1].LastPackage = nil
+
+	ctx := m.Context.(stdlibs.ExecContext)
+	ctx.OrigCaller = addr
+	m.Context = ctx
+}
+
 func TestSetOrigCaller(m *gno.Machine, addr crypto.Bech32Address) {
 	ctx := m.Context.(std.ExecContext)
 	ctx.OrigCaller = addr