Add test TestNewGHDocWithDebug

Author: ekalinin

ghdoc.go

@@ -39,16 +39,28 @@ type GHDoc struct {
 	Indent     int
 	Debug      bool
 	html       string
+	logger     *log.Logger
 }
 
 // NewGHDoc create GHDoc
 func NewGHDoc(Path string, AbsPaths bool, StartDepth int, Depth int, Escape bool, Token string, Indent int, Debug bool) *GHDoc {
-	return &GHDoc{Path, AbsPaths, StartDepth, Depth, Escape, Token, Indent, Debug, ""}
+	return &GHDoc{
+		Path:       Path,
+		AbsPaths:   AbsPaths,
+		StartDepth: StartDepth,
+		Depth:      Depth,
+		Escape:     Escape,
+		GhToken:    Token,
+		Indent:     Indent,
+		Debug:      Debug,
+		html:       "",
+		logger:     log.New(os.Stderr, "", log.LstdFlags),
+	}
 }
 
 func (doc *GHDoc) d(msg string) {
 	if doc.Debug {
-		log.Println(msg)
+		doc.logger.Println(msg)
 	}
 }
 

ghdoc_test.go

@@ -2,6 +2,7 @@ package ghtoc
 
 import (
 	"bytes"
+	"log"
 	"testing"
 )
 
@@ -362,5 +363,18 @@ func TestGHToc_Print(t *testing.T) {
 }
 
 func TestNewGHDocWithDebug(t *testing.T) {
+	noMatterN := 1
+	noMatterS := "test"
+	noMatterB := false
+	var got bytes.Buffer
+
+	doc := NewGHDoc(noMatterS, noMatterB, noMatterN, noMatterN,
+		noMatterB, noMatterS, noMatterN, true)
+	doc.logger = log.New(&got, "", 0)
 
+	want := "test"
+	doc.d(want)
+	if got.String() != want+"\n" {
+		t.Error("\nGot :", got.String(), "\nWant:", want)
+	}
 }