From ccddb24e1d0ab751b5a60b053ffd817dc26d2929 Mon Sep 17 00:00:00 2001
From: Aleksey Lobanov <alekseylobanov1@gmail.com>
Date: Sun, 27 Mar 2016 02:59:35 +0300
Subject: [PATCH] Refactoring of benchmarks

---
 src/saving_test.go | 30 +++++++++++++-----------------
 1 file changed, 13 insertions(+), 17 deletions(-)

diff --git a/src/saving_test.go b/src/saving_test.go
index b3d781b..e269eaa 100644
--- a/src/saving_test.go
+++ b/src/saving_test.go
@@ -3,17 +3,16 @@ package main
 
 import (
 	"io/ioutil"
-	"log"
 	"math/rand"
 	"os"
 	"testing"
 )
 
-func BenchmarkCsvSavingOnlySum(b *testing.B) {
+func GenerateRandStatsForTime(N int) []StatForTime {
 	data := make([]StatForTime, 0)
 	keyMap := GetKeymap()
 	rnd := rand.New(rand.NewSource(42))
-	for i := 0; i < b.N; i++ {
+	for i := 0; i < N; i++ {
 		var curStat StatForTime
 		curStat.Init()
 		for keyNum := range keyMap {
@@ -24,9 +23,16 @@ func BenchmarkCsvSavingOnlySum(b *testing.B) {
 		data = append(data, curStat)
 	}
 
+	return data
+}
+
+func BenchmarkCsvSavingOnlySum(b *testing.B) {
+	data := GenerateRandStatsForTime(b.N)
+	keyMap := GetKeymap()
+
 	tmpFile, err := ioutil.TempFile(os.TempDir(), "benchmark")
 	if err != nil {
-		log.Fatal(err)
+		b.Fatal(err)
 	}
 
 	defer os.Remove(tmpFile.Name())
@@ -37,22 +43,12 @@ func BenchmarkCsvSavingOnlySum(b *testing.B) {
 }
 
 func BenchmarkCsvSaving(b *testing.B) {
-	data := make([]StatForTime, 0)
+	data := GenerateRandStatsForTime(b.N)
 	keyMap := GetKeymap()
-	rnd := rand.New(rand.NewSource(42))
-	for i := 0; i < b.N; i++ {
-		var curStat StatForTime
-		curStat.Init()
-		for keyNum := range keyMap {
-			if rnd.Float32() > 0.7 {
-				curStat.keys[keyNum] = rnd.Intn(5000)
-			}
-		}
-		data = append(data, curStat)
-	}
+
 	tmpFile, err := ioutil.TempFile(os.TempDir(), "benchmark")
 	if err != nil {
-		log.Fatal(err)
+		b.Fatal(err)
 	}
 
 	defer os.Remove(tmpFile.Name())