compiler: add go test command

compiler/chore/ssadump/ssadump.go

@@ -148,7 +148,7 @@ func doMain() error {
 
 	for i, p := range pkgs {
 		if p == nil {
-			return fmt.Errorf("cannot build SSA for package %s", initial[i])
+			return fmt.Errorf("cannot build SSA for package %s", initial[i].ID)
 		}
 	}
 

compiler/cmd/internal/test/test.go

@@ -0,0 +1,33 @@
+package test
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/goplus/llgo/compiler/cmd/internal/base"
+	"github.com/goplus/llgo/compiler/internal/build"
+)
+
+// llgo test
+var Cmd = &base.Command{
+	UsageLine: "llgo test [build flags] package [arguments...]",
+	Short:     "Compile and run Go test",
+}
+
+func init() {
+	Cmd.Run = runCmd
+}
+
+func runCmd(cmd *base.Command, args []string) {
+	runCmdEx(cmd, args, build.ModeRun)
+}
+
+func runCmdEx(_ *base.Command, args []string, mode build.Mode) {
+	conf := build.NewDefaultConf(mode)
+	conf.Mode = build.ModeTest
+	_, err := build.Do(args, conf)
+	if err != nil {
+		fmt.Fprintln(os.Stderr, err)
+		os.Exit(1)
+	}
+}

compiler/cmd/llgo/llgo.go

@@ -31,6 +31,7 @@ import (
 	"github.com/goplus/llgo/compiler/cmd/internal/help"
 	"github.com/goplus/llgo/compiler/cmd/internal/install"
 	"github.com/goplus/llgo/compiler/cmd/internal/run"
+	"github.com/goplus/llgo/compiler/cmd/internal/test"
 	"github.com/goplus/llgo/compiler/cmd/internal/version"
 	"github.com/goplus/llgo/compiler/internal/mockable"
 )
@@ -47,6 +48,7 @@ func init() {
 		get.Cmd,
 		run.Cmd,
 		run.CmpTestCmd,
+		test.Cmd,
 		clean.Cmd,
 		version.Cmd,
 	}

compiler/go.mod

@@ -1,6 +1,6 @@
 module github.com/goplus/llgo/compiler
 
-go 1.22.0
+go 1.22.4
 
 require (
 	github.com/goplus/gogen v1.16.6

compiler/internal/build/build.go

@@ -55,6 +55,7 @@ const (
 	ModeBuild Mode = iota
 	ModeInstall
 	ModeRun
+	ModeTest
 	ModeCmpTest
 	ModeGen
 )
@@ -130,6 +131,10 @@ func Do(args []string, conf *Config) ([]Package, error) {
 		Mode:       loadSyntax | packages.NeedDeps | packages.NeedModule | packages.NeedExportFile,
 		BuildFlags: flags,
 		Fset:       token.NewFileSet(),
+		Tests:      conf.Mode == ModeTest,
+	}
+	if conf.Mode == ModeTest {
+		cfg.Mode |= packages.NeedForTest
 	}
 
 	if len(overlayFiles) > 0 {
@@ -164,15 +169,14 @@ func Do(args []string, conf *Config) ([]Package, error) {
 	initial, err := packages.LoadEx(dedup, sizes, cfg, patterns...)
 	check(err)
 	mode := conf.Mode
-	if len(initial) == 1 && len(initial[0].CompiledGoFiles) > 0 {
-		if mode == ModeBuild {
+	switch mode {
+	case ModeBuild:
+		if len(initial) == 1 && len(initial[0].CompiledGoFiles) > 0 {
 			mode = ModeInstall
 		}
-	} else if mode == ModeRun {
+	case ModeRun:
 		if len(initial) > 1 {
 			return nil, fmt.Errorf("cannot run multiple packages")
-		} else {
-			return nil, fmt.Errorf("no Go files in matched packages")
 		}
 	}
 
@@ -205,7 +209,8 @@ func Do(args []string, conf *Config) ([]Package, error) {
 	os.Setenv("PATH", env.BinDir()+":"+os.Getenv("PATH")) // TODO(xsw): check windows
 
 	ctx := &context{env, cfg, progSSA, prog, dedup, patches, make(map[string]none), initial, mode, 0}
-	pkgs := buildAllPkgs(ctx, initial, verbose)
+	pkgs, err := buildAllPkgs(ctx, initial, verbose)
+	check(err)
 	if mode == ModeGen {
 		for _, pkg := range pkgs {
 			if pkg.Package == initial[0] {
@@ -215,7 +220,8 @@ func Do(args []string, conf *Config) ([]Package, error) {
 		return nil, fmt.Errorf("initial package not found")
 	}
 
-	dpkg := buildAllPkgs(ctx, altPkgs[noRt:], verbose)
+	dpkg, err := buildAllPkgs(ctx, altPkgs[noRt:], verbose)
+	check(err)
 	var linkArgs []string
 	for _, pkg := range dpkg {
 		linkArgs = append(linkArgs, pkg.LinkArgs...)
@@ -266,7 +272,7 @@ type context struct {
 	nLibdir int
 }
 
-func buildAllPkgs(ctx *context, initial []*packages.Package, verbose bool) (pkgs []*aPackage) {
+func buildAllPkgs(ctx *context, initial []*packages.Package, verbose bool) (pkgs []*aPackage, err error) {
 	prog := ctx.prog
 	pkgs, errPkgs := allPkgs(ctx, initial, verbose)
 	for _, errPkg := range errPkgs {
@@ -276,7 +282,7 @@ func buildAllPkgs(ctx *context, initial []*packages.Package, verbose bool) (pkgs
 		fmt.Fprintln(os.Stderr, "cannot build SSA for package", errPkg)
 	}
 	if len(errPkgs) > 0 {
-		mockable.Exit(1)
+		return nil, fmt.Errorf("cannot build SSA for packages")
 	}
 	built := ctx.built
 	for _, aPkg := range pkgs {
@@ -862,6 +868,7 @@ var hasAltPkg = map[string]none{
 	"crypto/sha512":            {},
 	"crypto/subtle":            {},
 	"fmt":                      {},
+	"go/parser":                {},
 	"hash/crc32":               {},
 	"internal/abi":             {},
 	"internal/bytealg":         {},

compiler/internal/packages/load.go

@@ -58,6 +58,8 @@ const (
 	NeedTypesSizes = packages.NeedTypesSizes
 	NeedTypesInfo  = packages.NeedTypesInfo
 
+	NeedForTest = packages.NeedForTest
+
 	typecheckCgo = NeedModule - 1 // TODO(xsw): how to check
 )
 
@@ -129,18 +131,18 @@ func (p Deduper) SetPkgPath(fn func(path, name string) string) {
 	p.setpath = fn
 }
 
-func (p Deduper) Check(pkgPath string) *Cached {
-	if v, ok := p.cache.Load(pkgPath); ok {
+func (p Deduper) Check(id string) *Cached {
+	if v, ok := p.cache.Load(id); ok {
 		return v.(*Cached)
 	}
 	return nil
 }
 
-func (p Deduper) set(pkgPath string, cp *Cached) {
+func (p Deduper) set(id string, cp *Cached) {
 	if DebugPackagesLoad {
-		log.Println("==> Import", pkgPath)
+		log.Println("==> Import", id)
 	}
-	p.cache.Store(pkgPath, cp)
+	p.cache.Store(id, cp)
 }
 
 //go:linkname defaultDriver golang.org/x/tools/go/packages.defaultDriver
@@ -176,7 +178,7 @@ func loadPackageEx(dedup Deduper, ld *loader, lpkg *loaderPackage) {
 	}
 
 	if dedup != nil {
-		if cp := dedup.Check(lpkg.PkgPath); cp != nil {
+		if cp := dedup.Check(lpkg.ID); cp != nil {
 			lpkg.Types = cp.Types
 			lpkg.Fset = ld.Fset
 			lpkg.TypesInfo = cp.TypesInfo