cltest: SetRuntime

2024-04-30 14:16:46 +08:00
parent 2d6869bdbd
commit 4eedb4fb69
1 changed files with 8 additions and 0 deletions
--- a/cl/cltest/cltest.go
+++ b/cl/cltest/cltest.go
@@ -113,6 +113,14 @@ func TestCompileEx(t *testing.T, src any, fname, expected string) {
 	}
 	foo.WriteTo(os.Stderr)
 	prog := llssa.NewProgram(nil)
+	prog.SetRuntime(func() *types.Package {
+		rt, err := imp.Import(llssa.PkgRuntime)
+		if err != nil {
+			t.Fatal("load runtime failed:", err)
+		}
+		return rt
+	})
+
 	ret, err := cl.NewPackage(prog, foo, files)
 	if err != nil {
 		t.Fatal("cl.NewPackage failed:", err)