From 3425950f2104007f4b903d4737ad07313aca0963 Mon Sep 17 00:00:00 2001
From: Carl Jackson <carl@avtok.com>
Date: Sun, 4 May 2014 09:15:09 -0700
Subject: [PATCH] DRY up the benchmarks a bit

---
 web/bench_test.go | 31 ++++++++-----------------------
 1 file changed, 8 insertions(+), 23 deletions(-)

diff --git a/web/bench_test.go b/web/bench_test.go
index 051fe26..6e198c3 100644
--- a/web/bench_test.go
+++ b/web/bench_test.go
@@ -85,45 +85,30 @@ func permuteRequests(reqs []*http.Request) []*http.Request {
 	return out
 }
 
-func testingMux(n int) (*Mux, []*http.Request) {
+func benchN(b *testing.B, n int) {
 	m := New()
 	prefixes := genPrefixes(n)
 	for _, prefix := range prefixes {
 		addRoutes(m, prefix)
 	}
 	reqs := permuteRequests(genRequests(prefixes))
-	return m, reqs
-}
 
-func BenchmarkRoute5(b *testing.B) {
-	m, reqs := testingMux(1)
 	b.ResetTimer()
 	b.ReportAllocs()
 	for i := 0; i < b.N; i++ {
 		m.ServeHTTP(w, reqs[i%len(reqs)])
 	}
 }
+
+func BenchmarkRoute5(b *testing.B) {
+	benchN(b, 1)
+}
 func BenchmarkRoute50(b *testing.B) {
-	m, reqs := testingMux(10)
-	b.ResetTimer()
-	b.ReportAllocs()
-	for i := 0; i < b.N; i++ {
-		m.ServeHTTP(w, reqs[i%len(reqs)])
-	}
+	benchN(b, 10)
 }
 func BenchmarkRoute500(b *testing.B) {
-	m, reqs := testingMux(100)
-	b.ResetTimer()
-	b.ReportAllocs()
-	for i := 0; i < b.N; i++ {
-		m.ServeHTTP(w, reqs[i%len(reqs)])
-	}
+	benchN(b, 100)
 }
 func BenchmarkRoute5000(b *testing.B) {
-	m, reqs := testingMux(1000)
-	b.ResetTimer()
-	b.ReportAllocs()
-	for i := 0; i < b.N; i++ {
-		m.ServeHTTP(w, reqs[i%len(reqs)])
-	}
+	benchN(b, 1000)
 }