Merge pull request #1267 from fenggw-fnst/work

test: add unit test case for func WithConsole
dragonflyoss · Apr 9, 2020 · cbfc36c · cbfc36c
2 parents 05b2f8e + 0ab5157
commit cbfc36c
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/pkg/dflog/log_test.go b/pkg/dflog/log_test.go
@@ -72,6 +72,16 @@ func (ts *logTestSuite) TestMaxSizeMB() {
 	r.Equal(20, lumberjack.MaxSize)
 }
 
+func (ts *logTestSuite) TestConsole() {
+	r := ts.Require()
+
+	l := logrus.New()
+	r.Nil(Init(l, WithConsole()))
+	for _, level := range logrus.AllLevels {
+		r.Equal(1, len(l.Hooks[level]))
+	}
+}
+
 func (ts *logTestSuite) TestFormatter() {
 	r := ts.Require()