From 4b0974ec10e6b43dd38cacda43424ee0cdde039e Mon Sep 17 00:00:00 2001
From: Kjell Kvinge <kjell@kvinge.biz>
Date: Tue, 3 Jan 2017 03:52:09 +0100
Subject: [PATCH] Fix benchmarktests (#557)

---
 models/graph_test.go | 28 ++++++++++++++++------------
 1 file changed, 16 insertions(+), 12 deletions(-)

diff --git a/models/graph_test.go b/models/graph_test.go
index 23d8aa849..47c9dbb08 100644
--- a/models/graph_test.go
+++ b/models/graph_test.go
@@ -17,25 +17,29 @@ func BenchmarkGetCommitGraph(b *testing.B) {
 		b.Error("Could not open repository")
 	}
 
-	graph, err := GetCommitGraph(currentRepo)
-	if err != nil {
-		b.Error("Could get commit graph")
-	}
+	for i := 0; i < b.N; i++ {
+		graph, err := GetCommitGraph(currentRepo)
+		if err != nil {
+			b.Error("Could get commit graph")
+		}
 
-	if len(graph) < 100 {
-		b.Error("Should get 100 log lines.")
+		if len(graph) < 100 {
+			b.Error("Should get 100 log lines.")
+		}
 	}
 }
 
 func BenchmarkParseCommitString(b *testing.B) {
 	testString := "* DATA:||4e61bacab44e9b4730e44a6615d04098dd3a8eaf|2016-12-20 21:10:41 +0100|Kjell Kvinge|kjell@kvinge.biz|4e61bac|Add route for graph"
 
-	graphItem, err := graphItemFromString(testString, nil)
-	if err != nil {
-		b.Error("could not parse teststring")
-	}
+	for i := 0; i < b.N; i++ {
+		graphItem, err := graphItemFromString(testString, nil)
+		if err != nil {
+			b.Error("could not parse teststring")
+		}
 
-	if graphItem.Author != "Kjell Kvinge" {
-		b.Error("Did not get expected data")
+		if graphItem.Author != "Kjell Kvinge" {
+			b.Error("Did not get expected data")
+		}
 	}
 }