func (*writerBasicsSuite) TestRemoveDefaultWriter(c *gc.C) {
defaultWriter, level, err := loggo.RemoveWriter("default")
c.Assert(err, gc.IsNil)
c.Assert(level, gc.Equals, loggo.TRACE)
c.Assert(defaultWriter, gc.NotNil)
// Trying again fails.
defaultWriter, level, err = loggo.RemoveWriter("default")
c.Assert(err, gc.ErrorMatches, `Writer "default" is not registered`)
c.Assert(level, gc.Equals, loggo.UNSPECIFIED)
c.Assert(defaultWriter, gc.IsNil)
}
func (s *writerSuite) SetUpTest(c *gc.C) {
loggo.ResetLoggers()
loggo.RemoveWriter("default")
s.logger = loggo.GetLogger("test.writer")
// Make it so the logger itself writes all messages.
s.logger.SetLogLevel(loggo.TRACE)
}
func (s *logwriterSuite) BenchmarkLoggingNoWritersNoFormat(c *gc.C) {
// No writers
loggo.RemoveWriter("test")
for i := 0; i < c.N; i++ {
s.logger.Warningf("just a simple warning")
}
}
func (s *logwriterSuite) SetUpTest(c *gc.C) {
loggo.ResetLoggers()
loggo.RemoveWriter("default")
s.writer = &loggo.TestWriter{}
err := loggo.RegisterWriter("test", s.writer, loggo.TRACE)
c.Assert(err, gc.IsNil)
s.logger = loggo.GetLogger("test.writer")
// Make it so the logger itself writes all messages.
s.logger.SetLogLevel(loggo.TRACE)
}
func setupTempFileWriter(c *gc.C) (logFile *os.File, cleanup func()) {
loggo.RemoveWriter("test")
logFile, err := ioutil.TempFile("", "loggo-test")
c.Assert(err, gc.IsNil)
cleanup = func() {
logFile.Close()
os.Remove(logFile.Name())
}
writer := loggo.NewSimpleWriter(logFile, &loggo.DefaultFormatter{})
err = loggo.RegisterWriter("testfile", writer, loggo.TRACE)
c.Assert(err, gc.IsNil)
return
}
func (s *writerBasicsSuite) TestWillWrite(c *gc.C) {
// By default, the root logger watches TRACE messages
c.Assert(loggo.WillWrite(loggo.TRACE), gc.Equals, true)
// Note: ReplaceDefaultWriter doesn't let us change the default log
// level :(
writer, _, err := loggo.RemoveWriter("default")
c.Assert(err, gc.IsNil)
c.Assert(writer, gc.NotNil)
err = loggo.RegisterWriter("default", writer, loggo.CRITICAL)
c.Assert(err, gc.IsNil)
c.Assert(loggo.WillWrite(loggo.TRACE), gc.Equals, false)
c.Assert(loggo.WillWrite(loggo.DEBUG), gc.Equals, false)
c.Assert(loggo.WillWrite(loggo.INFO), gc.Equals, false)
c.Assert(loggo.WillWrite(loggo.WARNING), gc.Equals, false)
c.Assert(loggo.WillWrite(loggo.CRITICAL), gc.Equals, true)
}
func (s *logwriterSuite) BenchmarkLoggingDiskWriterNoMessages(c *gc.C) {
logFile, cleanup := setupTempFileWriter(c)
defer cleanup()
// Change the log level
writer, _, err := loggo.RemoveWriter("testfile")
c.Assert(err, gc.IsNil)
loggo.RegisterWriter("testfile", writer, loggo.WARNING)
msg := "just a simple warning for %d"
for i := 0; i < c.N; i++ {
s.logger.Debugf(msg, i)
}
offset, err := logFile.Seek(0, os.SEEK_CUR)
c.Assert(err, gc.IsNil)
c.Assert(offset, gc.Equals, int64(0),
gc.Commentf("Data was written to the log file."))
}
func (*writerBasicsSuite) TestReplaceDefaultWriterNoDefault(c *gc.C) {
loggo.RemoveWriter("default")
oldWriter, err := loggo.ReplaceDefaultWriter(&loggo.TestWriter{})
c.Assert(oldWriter, gc.IsNil)
c.Assert(err, gc.ErrorMatches, `there is no "default" writer`)
}