Browse Source

Fixed log tests

Signed-off-by: Vishal Rana <vr@labstack.com>
Vishal Rana 10 years ago
parent
commit
4bcd18e0c2
1 changed files with 7 additions and 2 deletions
  1. 7 2
      log/log_test.go

+ 7 - 2
log/log_test.go

@@ -1,6 +1,11 @@
 package log
 
-import "testing"
+import (
+	"bytes"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
 
 func TestLog(t *testing.T) {
 	l := New("test")
@@ -9,7 +14,7 @@ func TestLog(t *testing.T) {
 	test(l, TRACE, t)
 	assert.Contains(t, b.String(), "trace")
 	assert.Contains(t, b.String(), "fatal")
-	
+
 	b.Reset()
 	SetOutput(b)
 	test(global, NOTICE, t)