mirror of
https://github.com/tinygo-org/tinygo.git
synced 2026-08-03 02:27:48 +00:00
Piotr Bocheński
4465360f3d
PR #5220 changed -print-allocs output to the go coverage tool format, which replaced the original human-readable explanation of why each object had to be heap allocated. That explanation is useful on its own, so this restores it as the default behavior of -print-allocs and moves the coverage format behind a -print-allocs-cover flag. Signed-off-by: Piotr Bocheński <piotr@bochen.ski>
72 lines
1.8 KiB
Go
72 lines
1.8 KiB
Go
package transform_test
|
|
|
|
import (
|
|
"go/token"
|
|
"regexp"
|
|
"sort"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/tinygo-org/tinygo/transform"
|
|
"tinygo.org/x/go-llvm"
|
|
)
|
|
|
|
func TestAllocs(t *testing.T) {
|
|
t.Parallel()
|
|
testTransform(t, "testdata/allocs", func(mod llvm.Module) {
|
|
transform.OptimizeAllocs(mod, nil, 256, nil)
|
|
})
|
|
}
|
|
|
|
// Test with a Go file as input (for more accurate tests).
|
|
func TestAllocs2(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
const (
|
|
basePath = "testdata/allocs2"
|
|
goFile = basePath + ".go"
|
|
goldenFile = basePath + ".out"
|
|
)
|
|
mod := compileGoFileForTesting(t, goFile)
|
|
|
|
// Run functionattrs pass, which is necessary for escape analysis.
|
|
po := llvm.NewPassBuilderOptions()
|
|
defer po.Dispose()
|
|
err := mod.RunPasses("function(instcombine),function-attrs", llvm.TargetMachine{}, po)
|
|
if err != nil {
|
|
t.Error("failed to run passes:", err)
|
|
}
|
|
|
|
// Run heap to stack transform.
|
|
type report struct {
|
|
pos token.Position
|
|
reason string
|
|
}
|
|
var reports []report
|
|
transform.OptimizeAllocs(mod, regexp.MustCompile("."), 256, func(pos token.Position, reason string) {
|
|
pos.Filename = goFile
|
|
reports = append(reports, report{pos, reason})
|
|
})
|
|
sort.Slice(reports, func(i, j int) bool { return reports[i].pos.Line < reports[j].pos.Line })
|
|
|
|
// Render every report in each format and diff against its golden file.
|
|
for _, format := range []struct {
|
|
name string
|
|
render func(report) string
|
|
}{
|
|
{"reason", func(r report) string { return transform.FormatAllocReason(r.pos, r.reason) }},
|
|
{"cover", func(r report) string { return transform.FormatAllocCover(r.pos) }},
|
|
} {
|
|
t.Run(format.name, func(t *testing.T) {
|
|
var got strings.Builder
|
|
for _, r := range reports {
|
|
if line := format.render(r); line != "" {
|
|
got.WriteString(line)
|
|
got.WriteByte('\n')
|
|
}
|
|
}
|
|
checkGolden(t, goldenFile+"."+format.name, got.String())
|
|
})
|
|
}
|
|
}
|