llgo/cl/cltest

cl/compile_test.go

@@ -14,101 +14,21 @@
  * limitations under the License.
  */
 
-package cl
+package cl_test
 
 import (
-	"go/ast"
-	"go/parser"
-	"go/token"
-	"go/types"
-	"log"
-	"os"
-	"path"
-	"strings"
 	"testing"
 
-	"github.com/goplus/gogen/packages"
-	"golang.org/x/tools/go/ssa"
-	"golang.org/x/tools/go/ssa/ssautil"
-
-	llssa "github.com/goplus/llgo/ssa"
+	"github.com/goplus/llgo/cl/cltest"
 )
 
 func TestFromTestdata(t *testing.T) {
-	testFromDir(t, "", "./_testdata")
-}
-
-func init() {
-	SetDebug(DbgFlagAll)
-	llssa.Initialize(llssa.InitAll)
-	llssa.SetDebug(llssa.DbgFlagAll)
-}
-
-func testFromDir(t *testing.T, sel, relDir string) {
-	dir, err := os.Getwd()
-	if err != nil {
-		t.Fatal("Getwd failed:", err)
-	}
-	dir = path.Join(dir, relDir)
-	fis, err := os.ReadDir(dir)
-	if err != nil {
-		t.Fatal("ReadDir failed:", err)
-	}
-	for _, fi := range fis {
-		name := fi.Name()
-		if !fi.IsDir() || strings.HasPrefix(name, "_") {
-			continue
-		}
-		t.Run(name, func(t *testing.T) {
-			testFrom(t, dir+"/"+name, sel)
-		})
-	}
-}
-
-func testFrom(t *testing.T, pkgDir, sel string) {
-	if sel != "" && !strings.Contains(pkgDir, sel) {
-		return
-	}
-	log.Println("Parsing", pkgDir)
-	in := pkgDir + "/in.go"
-	out := pkgDir + "/out.ll"
-	expected, err := os.ReadFile(out)
-	if err != nil {
-		t.Fatal("ReadFile failed:", err)
-	}
-	testCompileEx(t, nil, in, string(expected))
-}
-
-func testCompileEx(t *testing.T, src any, fname, expected string) {
-	t.Helper()
-	fset := token.NewFileSet()
-	f, err := parser.ParseFile(fset, fname, src, parser.ParseComments)
-	if err != nil {
-		t.Fatal("ParseFile failed:", err)
-	}
-	files := []*ast.File{f}
-	name := f.Name.Name
-	pkg := types.NewPackage(name, name)
-	imp := packages.NewImporter(fset)
-	foo, _, err := ssautil.BuildPackage(
-		&types.Config{Importer: imp}, fset, pkg, files, ssa.SanityCheckFunctions)
-	if err != nil {
-		t.Fatal("BuildPackage failed:", err)
-	}
-	foo.WriteTo(os.Stderr)
-	prog := llssa.NewProgram(nil)
-	ret, err := NewPackage(prog, foo, files)
-	if err != nil {
-		t.Fatal("cl.NewPackage failed:", err)
-	}
-	if v := ret.String(); v != expected {
-		t.Fatalf("\n==> got:\n%s\n==> expected:\n%s\n", v, expected)
-	}
+	cltest.FromDir(t, "", "./_testdata")
 }
 
 func testCompile(t *testing.T, src, expected string) {
 	t.Helper()
-	testCompileEx(t, src, "foo.go", expected)
+	cltest.TestCompileEx(t, src, "foo.go", expected)
 }
 
 func TestVar(t *testing.T) {