From 4eedb4fb695b0a8cd14fc999378b608d771288de Mon Sep 17 00:00:00 2001
From: visualfc <visualfc@gmail.com>
Date: Tue, 30 Apr 2024 14:16:46 +0800
Subject: [PATCH] cltest: SetRuntime

---
 cl/cltest/cltest.go | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/cl/cltest/cltest.go b/cl/cltest/cltest.go
index 8ef858f1..d44edca3 100644
--- 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)