prog: measure memory allocation in benchmarks

Enable ReportAllocs.
Also factor out common code into a helper function
(duplicated in 3 places now).

1 files changed, 4 insertions, 14 deletions

diff --git a/prog/mutation_test.go b/prog/mutation_test.go
index 6b50b975e..b041f02d2 100644
--- a/prog/mutation_test.go
+++ b/prog/mutation_test.go
@@ -220,13 +220,8 @@ func runMutationTests(t *testing.T, tests [][2]string) {
 }
 
 func BenchmarkMutate(b *testing.B) {
-	olddebug := debug
-	debug = false
-	defer func() { debug = olddebug }()
-	target, err := GetTarget("linux", "amd64")
-	if err != nil {
-		b.Fatal(err)
-	}
+	target, cleanup := initBench(b)
+	defer cleanup()
 	ct := linuxAmd64ChoiceTable(target)
 	const progLen = 30
 	p := target.Generate(rand.NewSource(0), progLen, nil)
@@ -240,13 +235,8 @@ func BenchmarkMutate(b *testing.B) {
 }
 
 func BenchmarkGenerate(b *testing.B) {
-	olddebug := debug
-	debug = false
-	defer func() { debug = olddebug }()
-	target, err := GetTarget("linux", "amd64")
-	if err != nil {
-		b.Fatal(err)
-	}
+	target, cleanup := initBench(b)
+	defer cleanup()
 	ct := linuxAmd64ChoiceTable(target)
 	const progLen = 30
 	b.ResetTimer()