Add SetTemplate to console logger. Closes #2

aphistic · Oct 2, 2015 · 5389fca · 5389fca
1 parent 331edbe
commit 5389fca
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/console_logger.go b/console_logger.go
@@ -1,6 +1,7 @@
 package gomol
 
 import (
+	"errors"
 	"fmt"
 	"github.com/mgutz/ansi"
 )
@@ -74,6 +75,15 @@ func (l *ConsoleLogger) SetBase(base *Base) {
 	l.base = base
 }
 
+func (l *ConsoleLogger) SetTemplate(tpl *Template) error {
+	if tpl == nil {
+		return errors.New("A template must be provided")
+	}
+	l.tpl = tpl
+
+	return nil
+}
+
 func (l *ConsoleLogger) InitLogger() error {
 	l.isInitialized = true
 	return nil

diff --git a/console_logger_test.go b/console_logger_test.go
@@ -54,6 +54,19 @@ func (s *GomolSuite) TestIssue5StringFormatting(c *C) {
 
 // General tests
 
+func (s *GomolSuite) TestConsoleSetTemplate(c *C) {
+	cl, err := NewConsoleLogger(nil)
+	c.Check(cl.tpl, NotNil)
+
+	err = cl.SetTemplate(nil)
+	c.Check(err, NotNil)
+
+	tpl, err := NewTemplate("")
+	c.Assert(err, IsNil)
+	err = cl.SetTemplate(tpl)
+	c.Check(err, IsNil)
+}
+
 func (s *GomolSuite) TestConsoleInitLogger(c *C) {
 	cl, err := NewConsoleLogger(nil)
 	c.Assert(err, IsNil)