Added support for a custom logger when instantiating the server4 or

server6 object

Signed-off-by: Valerio Santinelli <santinelli@altralogica.it>

1 files changed, 42 insertions, 0 deletions

diff --git a/dhcpv6/server6/logger_test.go b/dhcpv6/server6/logger_test.go
new file mode 100644
index 0000000..b31a5e6
--- /dev/null
+++ b/dhcpv6/server6/logger_test.go
@@ -0,0 +1,42 @@
+// +build go1.12
+
+package server6
+
+import(
+	"log"
+	"os"
+	"testing"
+
+	"github.com/insomniacslk/dhcp/dhcpv6"
+	"github.com/stretchr/testify/require"
+)
+
+func TestEmptyLogger(t *testing.T) {
+	l := EmptyLogger{}
+	msg, err := dhcpv6.NewMessage()
+	require.Nil(t, err)
+	l.Printf("test")
+	l.PrintMessage("prefix", msg)
+}
+
+func TestShortSummaryLogger(t *testing.T) {
+	l := ShortSummaryLogger{
+		Printfer: log.New(os.Stderr, "[dhcpv6] ", log.LstdFlags),
+	}
+	msg, err := dhcpv6.NewMessage()
+	require.Nil(t, err)
+	require.NotNil(t, msg)
+	l.Printf("test")
+	l.PrintMessage("prefix", msg)
+}
+
+func TestDebugLogger(t *testing.T) {
+	l := DebugLogger{
+		Printfer: log.New(os.Stderr, "[dhcpv6] ", log.LstdFlags),
+	}
+	msg, err := dhcpv6.NewMessage()
+	require.Nil(t, err)
+	require.NotNil(t, msg)
+	l.Printf("test")
+	l.PrintMessage("prefix", msg)
+}
\ No newline at end of file