Stowage/go
2
0
Fork 0
mirror of https://github.com/golang/go.git synced 2025-12-08 06:10:04 +00:00
Code Issues Projects Releases Packages Wiki Activity

testing: add -test.memprofile and -test.memprofilerate flags.

Browse source 
These allow a test to generate memory profiles automatically.

R=golang-dev, rsc1
CC=golang-dev
https://golang.org/cl/4273064
This commit is contained in:
Rob Pike 2011-03-16 09:53:58 -07:00
parent df184ff2f0
commit afaa30694c
4 changed files with 52 additions and 48 deletions
Download patch file Download diff file Expand all files Collapse all files

37
src/pkg/testing/testing.go
View file

@ -43,12 +43,17 @@ import (
"fmt"
"os"
"runtime"
"runtime/pprof"
"time"
)
// Report as tests are run; default is silent for success.
var chatty = flag.Bool("test.v", false, "verbose: print additional output")
var match = flag.String("test.run", "", "regular expression to select tests to run")
var (
// Report as tests are run; default is silent for success.
chatty = flag.Bool("test.v", false, "verbose: print additional output")
match = flag.String("test.run", "", "regular expression to select tests to run")
memProfile = flag.String("test.memprofile", "", "after execution write the memory profile to the named file")
memProfileRate = flag.Int("test.memprofilerate", 0, "if >=0, sets runtime.MemProfileRate")
)
// Insert final newline if needed and tabs after internal newlines.
@ -138,6 +143,8 @@ func tRunner(t *T, test *InternalTest) {
// of gotest.
func Main(matchString func(pat, str string) (bool, os.Error), tests []InternalTest) {
flag.Parse()
before()
ok := true
if len(tests) == 0 {
println("testing: warning: no tests to run")
@ -170,9 +177,33 @@ func Main(matchString func(pat, str string) (bool, os.Error), tests []InternalTe
print(t.errors)
}
}
after()
if !ok {
println("FAIL")
os.Exit(1)
}
println("PASS")
}
// before runs before all testing.
func before() {
if *memProfileRate > 0 {
runtime.MemProfileRate = *memProfileRate
}
}
// after runs after all testing.
func after() {
if *memProfile == "" {
return
}
fd, err := os.Open(*memProfile, os.O_WRONLY|os.O_CREAT|os.O_TRUNC, 0666)
if err != nil {
fmt.Fprintf(os.Stderr, "testing: can't open %s: %s", *memProfile, err)
return
}
if err = pprof.WriteHeapProfile(fd); err != nil {
fmt.Fprintf(os.Stderr, "testing: can't write %s: %s", *memProfile, err)
}
fd.Close()
}