test: add unit test case for func WithConsole

Signed-off-by: Guangwen Feng <fenggw-fnst@cn.fujitsu.com>
dragonflyoss · Apr 3, 2020 · f9a86be · f9a86be
1 parent 18b652c
commit f9a86be
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()