summaryrefslogtreecommitdiffhomepage
path: root/dhcpv6
diff options
context:
space:
mode:
authorPablo Mazzini <pmazzini@gmail.com>2020-03-28 11:02:34 +0000
committerPablo Mazzini <pmazzini@gmail.com>2020-03-28 11:13:56 +0000
commita6429049b99acdc3d40b2379917cd3e7cc634239 (patch)
tree62fc12b162af900429b06a762e21c97bf3acc3d3 /dhcpv6
parent0444670a63335cc631222e99f59f4c7bfdff83bd (diff)
remove some tests
Signed-off-by: Pablo Mazzini <pmazzini@gmail.com>
Diffstat (limited to 'dhcpv6')
-rw-r--r--dhcpv6/server6/logger_test.go4
-rw-r--r--dhcpv6/server6/server_test.go139
2 files changed, 3 insertions, 140 deletions
diff --git a/dhcpv6/server6/logger_test.go b/dhcpv6/server6/logger_test.go
index b31a5e6..1ffacb7 100644
--- a/dhcpv6/server6/logger_test.go
+++ b/dhcpv6/server6/logger_test.go
@@ -1,5 +1,3 @@
-// +build go1.12
-
package server6
import(
@@ -39,4 +37,4 @@ func TestDebugLogger(t *testing.T) {
require.NotNil(t, msg)
l.Printf("test")
l.PrintMessage("prefix", msg)
-} \ No newline at end of file
+}
diff --git a/dhcpv6/server6/server_test.go b/dhcpv6/server6/server_test.go
index f7f7008..09cde4c 100644
--- a/dhcpv6/server6/server_test.go
+++ b/dhcpv6/server6/server_test.go
@@ -4,9 +4,7 @@ import (
"context"
"log"
"net"
- "sync"
"testing"
- "time"
"github.com/insomniacslk/dhcp/dhcpv6"
"github.com/insomniacslk/dhcp/dhcpv6/nclient6"
@@ -30,7 +28,7 @@ func (f unconnectedConn) ReadFrom(b []byte) (int, net.Addr, error) {
// utility function to set up a client and a server instance and run it in
// background. The caller needs to call Server.Close() once finished.
-func setUpClientAndServer(handler Handler, logger *customLogger) (*nclient6.Client, *Server) {
+func setUpClientAndServer(handler Handler) (*nclient6.Client, *Server) {
laddr := &net.UDPAddr{
IP: net.ParseIP("::1"),
Port: 0,
@@ -39,11 +37,6 @@ func setUpClientAndServer(handler Handler, logger *customLogger) (*nclient6.Clie
if err != nil {
panic(err)
}
-
- if logger != nil {
- s.logger = logger
- }
-
go func() {
_ = s.Serve()
}()
@@ -60,30 +53,6 @@ func setUpClientAndServer(handler Handler, logger *customLogger) (*nclient6.Clie
return c, s
}
-type customLogger struct {
- tb testing.TB
- called bool
- mux sync.Mutex
-}
-
-func (s *customLogger) Printf(format string, v ...interface{}) {
- s.mux.Lock()
- s.called = true
- s.mux.Unlock()
- s.tb.Logf("===CustomLogger BEGIN===")
- s.tb.Logf(format, v...)
- s.tb.Logf("===CustomLogger END===")
-}
-
-func (s *customLogger) PrintMessage(prefix string, message *dhcpv6.Message) {
- s.mux.Lock()
- s.called = true
- s.mux.Unlock()
- s.tb.Logf("===CustomLogger BEGIN===")
- s.tb.Logf("%s: %s", prefix, message)
- s.tb.Logf("===CustomLogger END===")
-}
-
func TestServer(t *testing.T) {
handler := func(conn net.PacketConn, peer net.Addr, m dhcpv6.DHCPv6) {
msg := m.(*dhcpv6.Message)
@@ -97,7 +66,7 @@ func TestServer(t *testing.T) {
}
}
- c, s := setUpClientAndServer(handler, nil)
+ c, s := setUpClientAndServer(handler)
defer s.Close()
ifaces, err := interfaces.GetLoopbackInterfaces()
@@ -107,107 +76,3 @@ func TestServer(t *testing.T) {
_, err = c.Solicit(context.Background(), dhcpv6.WithRapidCommit)
require.NoError(t, err)
}
-
-func TestCustomLoggerForServer(t *testing.T) {
- handler := func(conn net.PacketConn, peer net.Addr, m dhcpv6.DHCPv6) {
- msg := m.(*dhcpv6.Message)
- adv, err := dhcpv6.NewAdvertiseFromSolicit(msg)
- if err != nil {
- log.Printf("NewAdvertiseFromSolicit failed: %v", err)
- return
- }
- if _, err := conn.WriteTo(adv.ToBytes(), peer); err != nil {
- log.Printf("Cannot reply to client: %v", err)
- }
- }
-
- c, s := setUpClientAndServer(handler, &customLogger{
- tb: t,
- })
- defer s.Close()
-
- ifaces, err := interfaces.GetLoopbackInterfaces()
- require.NoError(t, err)
- require.NotEqual(t, 0, len(ifaces))
-
- _, err = c.Solicit(context.Background(), dhcpv6.WithRapidCommit)
- require.NoError(t, err)
- go func() {
- time.Sleep(time.Second * 5)
- require.Equal(t, true, s.logger.(*customLogger).called)
- }()
-}
-
-func TestServerInstantiationWithCustomLogger(t *testing.T) {
- handler := func(conn net.PacketConn, peer net.Addr, m dhcpv6.DHCPv6) {
- msg := m.(*dhcpv6.Message)
- adv, err := dhcpv6.NewAdvertiseFromSolicit(msg)
- if err != nil {
- log.Printf("NewAdvertiseFromSolicit failed: %v", err)
- return
- }
- if _, err := conn.WriteTo(adv.ToBytes(), peer); err != nil {
- log.Printf("Cannot reply to client: %v", err)
- }
- }
-
- laddr := &net.UDPAddr{
- IP: net.ParseIP("::1"),
- Port: 0,
- }
- s, err := NewServer("", laddr, handler, WithLogger(&customLogger{
- tb: t,
- }))
- if err != nil {
- t.Fatal(err)
- }
- require.NotNil(t, s)
-}
-
-func TestServerInstantiationWithSummaryLogger(t *testing.T) {
- handler := func(conn net.PacketConn, peer net.Addr, m dhcpv6.DHCPv6) {
- msg := m.(*dhcpv6.Message)
- adv, err := dhcpv6.NewAdvertiseFromSolicit(msg)
- if err != nil {
- log.Printf("NewAdvertiseFromSolicit failed: %v", err)
- return
- }
- if _, err := conn.WriteTo(adv.ToBytes(), peer); err != nil {
- log.Printf("Cannot reply to client: %v", err)
- }
- }
-
- laddr := &net.UDPAddr{
- IP: net.ParseIP("::1"),
- Port: 0,
- }
- s, err := NewServer("", laddr, handler, WithSummaryLogger())
- if err != nil {
- t.Fatal(err)
- }
- require.NotNil(t, s)
-}
-
-func TestServerInstantiationWithDebugLogger(t *testing.T) {
- handler := func(conn net.PacketConn, peer net.Addr, m dhcpv6.DHCPv6) {
- msg := m.(*dhcpv6.Message)
- adv, err := dhcpv6.NewAdvertiseFromSolicit(msg)
- if err != nil {
- log.Printf("NewAdvertiseFromSolicit failed: %v", err)
- return
- }
- if _, err := conn.WriteTo(adv.ToBytes(), peer); err != nil {
- log.Printf("Cannot reply to client: %v", err)
- }
- }
-
- laddr := &net.UDPAddr{
- IP: net.ParseIP("::1"),
- Port: 0,
- }
- s, err := NewServer("", laddr, handler, WithDebugLogger())
- if err != nil {
- t.Fatal(err)
- }
- require.NotNil(t, s)
-}