Use a null writer + null logger when running tests

connection_test.go

@@ -2,8 +2,8 @@ package main
 
 import (
 	"io"
-	"net/http"
 	"net"
+	"net/http"
 	"strings"
 	"testing"
 )
@@ -50,7 +50,7 @@ func readMessage(reader io.Reader) string {
 }
 
 func TestMain(m *testing.M) {
-	InitLogger()
+	InitNullLogger()
 	m.Run()
 }
 
@@ -163,7 +163,6 @@ func TestSampleProxyViaConnect(t *testing.T) {
 		t.Fatalf("Expected '%s' but got '%s'", expected_response, response)
 	}
 
-
 	// Mimic a regular http request
 	request := "GET / HTTP/1.1\r\nHost: localhost\r\n\r\n"
 	incoming.ClientWriter.Write([]byte(request))
@@ -173,7 +172,6 @@ func TestSampleProxyViaConnect(t *testing.T) {
 		t.Fatalf("Expected '%s' but got '%s'", expected_response, response)
 	}
 
-
 	incoming.CloseClient()
 	<-cleanedUp
 }

logger.go

@@ -5,6 +5,12 @@ import (
 	"os"
 )
 
+type nullWriter struct{}
+
+func (nullWriter) Write(p []byte) (n int, err error) {
+	return len(p), nil
+}
+
 type Logger struct {
 	Info  *log.Logger
 	Warn  *log.Logger
@@ -17,6 +23,14 @@ func InitLogger() {
 	logger = NewLogger()
 }
 
+func InitNullLogger() {
+	nw := nullWriter{}
+	InitLogger()
+	logger.Info.SetOutput(nw)
+	logger.Warn.SetOutput(nw)
+	logger.Fatal.SetOutput(nw)
+}
+
 func NewLogger() *Logger {
 	return &Logger{
 		Info:  log.New(os.Stdout, "[info] ", log.Ldate|log.Ltime|log.Lshortfile),