plot: Rename BenchmarkHTMLPlot

danielsig727 · Aug 12, 2018 · 48ba0db · 48ba0db
1 parent c9b9ae4
commit 48ba0db
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/lib/plot/plot_test.go b/lib/plot/plot_test.go
@@ -8,7 +8,7 @@ import (
 	vegeta "github.com/tsenart/vegeta/lib"
 )
 
-func BenchmarkHTMLPlot(b *testing.B) {
+func BenchmarkPlot(b *testing.B) {
 	b.StopTimer()
 	// Build result set
 	rs := make(vegeta.Results, 50000000)
@@ -27,12 +27,7 @@ func BenchmarkHTMLPlot(b *testing.B) {
 	plot := New(
 		Title("Vegeta Plot"),
 		Downsample(5000),
-		Label(func(r *vegeta.Result) string {
-			if r.Code >= 200 && r.Code < 300 {
-				return "OK"
-			}
-			return "Error"
-		}),
+		Label(ErrorLabeler),
 	)
 
 	b.Run("Add", func(b *testing.B) {