Merge 469377f into 920ccb4

log.go

@@ -100,6 +100,16 @@ func (l *Logger) GetLevel() string {
 	return LogLevelNames[l.logLevel]
 }
 
+// SetWriter to set the log writer
+func (l *Logger) SetWriter(o io.Writer) {
+	l.out = o
+}
+
+// GetWriter to get the log writer
+func (l *Logger) GetWriter() io.Writer {
+	return l.out
+}
+
 func (l *Logger) log(logLevel level, context interface{}, message string, args ...interface{}) {
 	if logLevel >= l.logLevel {
 		var buf bytes.Buffer

log_test.go

@@ -23,6 +23,10 @@ import (
 	"strings"
 	"testing"
 	"time"
+	"os"
+	"bufio"
+	"io/ioutil"
+	"encoding/json"
 )
 
 func TestLevelByName(t *testing.T) {
@@ -68,6 +72,56 @@ func TestLoggerLevel(t *testing.T) {
 	SetDefaultLogLevel("INFO")
 }
 
+func TestLoggerOutput(t *testing.T) {
+	const TestFile = "/tmp/random.log"
+	fd, err := os.Create(TestFile)
+	if err != nil {
+		t.Errorf("Couldn't create %s file: %s", TestFile, err.Error())
+		return
+	}
+
+	writer := bufio.NewWriter(fd)
+	logger := NewLogger()
+	logger.SetWriter(writer)
+	expectedMsg := "hola"
+	logger.Info(expectedMsg)
+	writer.Flush()
+
+	content, err := ioutil.ReadFile(TestFile)
+	if err != nil {
+		t.Errorf("Couldn't read %s file: %s", TestFile, err.Error())
+		return
+	}
+
+	parsed := map[string]interface{}{}
+
+	err = json.Unmarshal(content, &parsed)
+	if err != nil {
+		t.Errorf("Couldn't unmarshal %s file: %s", TestFile, err.Error())
+		return
+	}
+
+	expectedLevel := logger.GetLevel()
+	gotLevel := parsed["lvl"].(string)
+	if expectedLevel != gotLevel {
+		t.Errorf("Expecting level \"%s\", got \"%s\" instead.", expectedLevel, gotLevel)
+	}
+
+	gotMsg := parsed["msg"].(string)
+	if expectedMsg != gotMsg {
+		t.Errorf("Expecting msg \"%s\", got \"%s\" instead.", expectedMsg, gotMsg)
+	}
+
+	logger.SetWriter(os.Stdout)
+
+	new := logger.GetWriter()
+
+	if new != os.Stdout {
+		t.Error("Not expected output writer interface")
+	}
+}
+
+
 func TestLoggerContext(t *testing.T) {
 	logger := NewLogger()
 	if logger.GetLogContext() != nil {