140 lines
3.1 KiB
Go
140 lines
3.1 KiB
Go
package utils
|
|
|
|
import (
|
|
"bytes"
|
|
"log"
|
|
"os"
|
|
"time"
|
|
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
var _ = Describe("Log", func() {
|
|
var (
|
|
b *bytes.Buffer
|
|
|
|
initialTimeFormat string
|
|
)
|
|
|
|
BeforeEach(func() {
|
|
b = bytes.NewBuffer([]byte{})
|
|
log.SetOutput(b)
|
|
initialTimeFormat = timeFormat
|
|
})
|
|
|
|
AfterEach(func() {
|
|
log.SetOutput(os.Stdout)
|
|
SetLogLevel(LogLevelNothing)
|
|
timeFormat = initialTimeFormat
|
|
})
|
|
|
|
It("the log level has the correct numeric value", func() {
|
|
Expect(LogLevelNothing).To(BeEquivalentTo(0))
|
|
Expect(LogLevelError).To(BeEquivalentTo(1))
|
|
Expect(LogLevelInfo).To(BeEquivalentTo(2))
|
|
Expect(LogLevelDebug).To(BeEquivalentTo(3))
|
|
})
|
|
|
|
It("log level nothing", func() {
|
|
SetLogLevel(LogLevelNothing)
|
|
Debugf("debug")
|
|
Infof("info")
|
|
Errorf("err")
|
|
Expect(b.Bytes()).To(Equal([]byte("")))
|
|
})
|
|
|
|
It("log level err", func() {
|
|
SetLogLevel(LogLevelError)
|
|
Debugf("debug")
|
|
Infof("info")
|
|
Errorf("err")
|
|
Expect(b.Bytes()).To(ContainSubstring("err\n"))
|
|
Expect(b.Bytes()).ToNot(ContainSubstring("info"))
|
|
Expect(b.Bytes()).ToNot(ContainSubstring("debug"))
|
|
})
|
|
|
|
It("log level info", func() {
|
|
SetLogLevel(LogLevelInfo)
|
|
Debugf("debug")
|
|
Infof("info")
|
|
Errorf("err")
|
|
Expect(b.Bytes()).To(ContainSubstring("err\n"))
|
|
Expect(b.Bytes()).To(ContainSubstring("info\n"))
|
|
Expect(b.Bytes()).ToNot(ContainSubstring("debug"))
|
|
})
|
|
|
|
It("log level debug", func() {
|
|
SetLogLevel(LogLevelDebug)
|
|
Debugf("debug")
|
|
Infof("info")
|
|
Errorf("err")
|
|
Expect(b.Bytes()).To(ContainSubstring("err\n"))
|
|
Expect(b.Bytes()).To(ContainSubstring("info\n"))
|
|
Expect(b.Bytes()).To(ContainSubstring("debug\n"))
|
|
})
|
|
|
|
It("doesn't add a timestamp if the time format is empty", func() {
|
|
SetLogLevel(LogLevelDebug)
|
|
SetLogTimeFormat("")
|
|
Debugf("debug")
|
|
Expect(b.Bytes()).To(Equal([]byte("debug\n")))
|
|
})
|
|
|
|
It("adds a timestamp", func() {
|
|
format := "Jan 2, 2006"
|
|
SetLogTimeFormat(format)
|
|
SetLogLevel(LogLevelInfo)
|
|
Infof("info")
|
|
t, err := time.Parse(format, string(b.Bytes()[:b.Len()-6]))
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(t).To(BeTemporally("~", time.Now(), 25*time.Hour))
|
|
})
|
|
|
|
It("says whether debug is enabled", func() {
|
|
Expect(Debug()).To(BeFalse())
|
|
SetLogLevel(LogLevelDebug)
|
|
Expect(Debug()).To(BeTrue())
|
|
})
|
|
|
|
Context("reading from env", func() {
|
|
BeforeEach(func() {
|
|
Expect(logLevel).To(Equal(LogLevelNothing))
|
|
})
|
|
|
|
It("reads DEBUG", func() {
|
|
os.Setenv(logEnv, "DEBUG")
|
|
readLoggingEnv()
|
|
Expect(logLevel).To(Equal(LogLevelDebug))
|
|
})
|
|
|
|
It("reads debug", func() {
|
|
os.Setenv(logEnv, "debug")
|
|
readLoggingEnv()
|
|
Expect(logLevel).To(Equal(LogLevelDebug))
|
|
})
|
|
|
|
It("reads INFO", func() {
|
|
os.Setenv(logEnv, "INFO")
|
|
readLoggingEnv()
|
|
Expect(logLevel).To(Equal(LogLevelInfo))
|
|
})
|
|
|
|
It("reads ERROR", func() {
|
|
os.Setenv(logEnv, "ERROR")
|
|
readLoggingEnv()
|
|
Expect(logLevel).To(Equal(LogLevelError))
|
|
})
|
|
|
|
It("does not error reading invalid log levels from env", func() {
|
|
Expect(logLevel).To(Equal(LogLevelNothing))
|
|
os.Setenv(logEnv, "")
|
|
readLoggingEnv()
|
|
Expect(logLevel).To(Equal(LogLevelNothing))
|
|
os.Setenv(logEnv, "asdf")
|
|
readLoggingEnv()
|
|
Expect(logLevel).To(Equal(LogLevelNothing))
|
|
})
|
|
})
|
|
})
|