~themue/juju-core/050-prepare-openstack

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
// Copyright 2011, 2012, 2013 Canonical Ltd.
// Licensed under the AGPLv3, see LICENCE file for details.

package log_test

import (
	"bytes"
	"fmt"
	"testing"
	"time"

	gc "launchpad.net/gocheck"
	"launchpad.net/loggo"

	"launchpad.net/juju-core/log"
)

func Test(t *testing.T) {
	gc.TestingT(t)
}

type testWriter struct {
	bytes.Buffer
}

type suite struct {
	writer    *testWriter
	oldWriter loggo.Writer
	oldLevel  loggo.Level
}

var _ = gc.Suite(&suite{})

func (t *testWriter) Write(level loggo.Level, module, filename string, line int, timestamp time.Time, message string) {
	t.Buffer.WriteString(fmt.Sprintf("%s %s %s", level, module, message))
}

func (s *suite) SetUpTest(c *gc.C) {
	var err error
	s.writer = &testWriter{}
	s.oldWriter, s.oldLevel, err = loggo.RemoveWriter("default")
	c.Assert(err, gc.IsNil)
	err = loggo.RegisterWriter("test", s.writer, loggo.TRACE)
	c.Assert(err, gc.IsNil)
	logger := loggo.GetLogger("juju")
	logger.SetLogLevel(loggo.TRACE)
}

func (s *suite) TearDownTest(c *gc.C) {
	_, _, err := loggo.RemoveWriter("test")
	c.Assert(err, gc.IsNil)
	err = loggo.RegisterWriter("default", s.oldWriter, s.oldLevel)
	c.Assert(err, gc.IsNil)
}

func (s *suite) TestLoggerDebug(c *gc.C) {
	input := "Hello World"
	log.Debugf(input)
	c.Assert(s.writer.String(), gc.Equals, "DEBUG juju "+input)
}

func (s *suite) TestInfoLogger(c *gc.C) {
	input := "Hello World"
	log.Infof(input)
	c.Assert(s.writer.String(), gc.Equals, "INFO juju "+input)
}

func (s *suite) TestErrorLogger(c *gc.C) {
	input := "Hello World"
	log.Errorf(input)
	c.Assert(s.writer.String(), gc.Equals, "ERROR juju "+input)
}

func (s *suite) TestWarningLogger(c *gc.C) {
	input := "Hello World"
	log.Warningf(input)
	c.Assert(s.writer.String(), gc.Equals, "WARNING juju "+input)
}

func (s *suite) TestNoticeLogger(c *gc.C) {
	input := "Hello World"
	log.Noticef(input)
	c.Assert(s.writer.String(), gc.Equals, "INFO juju "+input)
}