Suppress logs during benchmarks

The logs are messing up the output meaning that perfcheck isn't able
to compare some benchmarks.

filter/rules_small_test.go

@@ -22,6 +22,7 @@ import (
 	"testing"
 	"time"
 
+	"github.com/jumptrading/influx-spout/spouttest"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 )
@@ -153,6 +154,9 @@ func TestMeasurementName(t *testing.T) {
 var result int
 
 func BenchmarkLineLookup(b *testing.B) {
+	spouttest.SuppressLogs()
+	defer spouttest.RestoreLogs()
+
 	rs := new(RuleSet)
 	rs.Append(CreateBasicRule("hello", ""))
 	line := []byte("hello world=42")
@@ -164,6 +168,9 @@ func BenchmarkLineLookup(b *testing.B) {
 }
 
 func BenchmarkLineLookupRegex(b *testing.B) {
+	spouttest.SuppressLogs()
+	defer spouttest.RestoreLogs()
+
 	rs := new(RuleSet)
 	rs.Append(CreateRegexRule("hello|abcde", ""))
 	line := []byte("hello world=42")
@@ -175,6 +182,9 @@ func BenchmarkLineLookupRegex(b *testing.B) {
 }
 
 func BenchmarkLineLookupNegativeRegex(b *testing.B) {
+	spouttest.SuppressLogs()
+	defer spouttest.RestoreLogs()
+
 	rs := new(RuleSet)
 	rs.Append(CreateNegativeRegexRule("hello|abcde", ""))
 	line := []byte("hello world=42")
@@ -186,6 +196,9 @@ func BenchmarkLineLookupNegativeRegex(b *testing.B) {
 }
 
 func BenchmarkProcessBatch(b *testing.B) {
+	spouttest.SuppressLogs()
+	defer spouttest.RestoreLogs()
+
 	// Run the Filter worker with a fake NATS connection.
 	rs := new(RuleSet)
 	rs.Append(CreateBasicRule("hello", "hello-out"))

listener/listener_medium_test.go

@@ -300,6 +300,12 @@ func TestHTTPListenerConcurrency(t *testing.T) {
 }
 
 func BenchmarkListenerLatency(b *testing.B) {
+	spouttest.SuppressLogs()
+	defer spouttest.RestoreLogs()
+
+	s := spouttest.RunGnatsd(natsPort)
+	defer s.Shutdown()
+
 	listener := startListener(b, testConfig())
 	defer listener.Stop()
 

spouttest/logs.go

@@ -0,0 +1,17 @@
+package spouttest
+
+import (
+	"io/ioutil"
+	"log"
+	"os"
+)
+
+// SuppressLogs causes log output globally to be discarded.
+func SuppressLogs() {
+	log.SetOutput(ioutil.Discard)
+}
+
+// RestoreLogs causes log output globally to be sent to stderr.
+func RestoreLogs() {
+	log.SetOutput(os.Stderr)
+}

writer/writer_medium_test.go

@@ -263,6 +263,9 @@ func TestNegativeRegexFilterRule(t *testing.T) {
 }
 
 func BenchmarkWriterLatency(b *testing.B) {
+	spouttest.SuppressLogs()
+	defer spouttest.RestoreLogs()
+
 	nc, closeNATS := runGnatsd(b)
 	defer closeNATS()