From 719c3c2eea3bcd825072af377d26e294c0083ec5 Mon Sep 17 00:00:00 2001
From: Aleksey Lobanov <alekseylobanov1@gmail.com>
Date: Sun, 27 Mar 2016 02:48:33 +0300
Subject: [PATCH] Added saving to system tempfile

---
 src/saving_test.go | 24 ++++++++++++++++++++++--
 1 file changed, 22 insertions(+), 2 deletions(-)

diff --git a/src/saving_test.go b/src/saving_test.go
index 893e877..b3d781b 100644
--- a/src/saving_test.go
+++ b/src/saving_test.go
@@ -2,7 +2,10 @@
 package main
 
 import (
+	"io/ioutil"
+	"log"
 	"math/rand"
+	"os"
 	"testing"
 )
 
@@ -20,8 +23,17 @@ func BenchmarkCsvSavingOnlySum(b *testing.B) {
 		}
 		data = append(data, curStat)
 	}
+
+	tmpFile, err := ioutil.TempFile(os.TempDir(), "benchmark")
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	defer os.Remove(tmpFile.Name())
+
 	b.ResetTimer()
-	SaveToCsvFile(data, keyMap, "/tmp/bla.csv", true)
+
+	SaveToCsvWriter(data, keyMap, tmpFile, true)
 }
 
 func BenchmarkCsvSaving(b *testing.B) {
@@ -38,6 +50,14 @@ func BenchmarkCsvSaving(b *testing.B) {
 		}
 		data = append(data, curStat)
 	}
+	tmpFile, err := ioutil.TempFile(os.TempDir(), "benchmark")
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	defer os.Remove(tmpFile.Name())
+
 	b.ResetTimer()
-	SaveToCsvFile(data, keyMap, "/tmp/bla.csv", false)
+
+	SaveToCsvWriter(data, keyMap, tmpFile, false)
 }