diff options
author | Pablo Mazzini <pmazzini@gmail.com> | 2020-03-28 11:02:34 +0000 |
---|---|---|
committer | Pablo Mazzini <pmazzini@gmail.com> | 2020-03-28 11:13:56 +0000 |
commit | a6429049b99acdc3d40b2379917cd3e7cc634239 (patch) | |
tree | 62fc12b162af900429b06a762e21c97bf3acc3d3 /dhcpv6 | |
parent | 0444670a63335cc631222e99f59f4c7bfdff83bd (diff) |
remove some tests
Signed-off-by: Pablo Mazzini <pmazzini@gmail.com>
Diffstat (limited to 'dhcpv6')
-rw-r--r-- | dhcpv6/server6/logger_test.go | 4 | ||||
-rw-r--r-- | dhcpv6/server6/server_test.go | 139 |
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) -} |