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 | |
parent | 0444670a63335cc631222e99f59f4c7bfdff83bd (diff) |
remove some tests
Signed-off-by: Pablo Mazzini <pmazzini@gmail.com>
-rw-r--r-- | dhcpv4/server4/logger_test.go | 4 | ||||
-rw-r--r-- | dhcpv4/server4/server_test.go | 115 | ||||
-rw-r--r-- | dhcpv6/server6/logger_test.go | 4 | ||||
-rw-r--r-- | dhcpv6/server6/server_test.go | 139 |
4 files changed, 6 insertions, 256 deletions
diff --git a/dhcpv4/server4/logger_test.go b/dhcpv4/server4/logger_test.go index c33d83e..bbeb91c 100644 --- a/dhcpv4/server4/logger_test.go +++ b/dhcpv4/server4/logger_test.go @@ -1,5 +1,3 @@ -// +build go1.12 - package server4 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/dhcpv4/server4/server_test.go b/dhcpv4/server4/server_test.go index 9005e0e..43314ad 100644 --- a/dhcpv4/server4/server_test.go +++ b/dhcpv4/server4/server_test.go @@ -7,7 +7,6 @@ import ( "log" "math/rand" "net" - "sync" "testing" "time" @@ -56,7 +55,7 @@ func DORAHandler(conn net.PacketConn, peer net.Addr, m *dhcpv4.DHCPv4) { // 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(t *testing.T, iface net.Interface, handler Handler, logger *customLogger) (*nclient4.Client, *Server) { +func setUpClientAndServer(t *testing.T, iface net.Interface, handler Handler) (*nclient4.Client, *Server) { // strong assumption, I know loAddr := net.ParseIP("127.0.0.1") saddr := &net.UDPAddr{ @@ -71,11 +70,6 @@ func setUpClientAndServer(t *testing.T, iface net.Interface, handler Handler, lo if err != nil { t.Fatal(err) } - - if logger != nil { - s.logger = logger - } - go func() { _ = s.Serve() }() @@ -92,30 +86,6 @@ func setUpClientAndServer(t *testing.T, iface net.Interface, handler Handler, lo 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 *dhcpv4.DHCPv4) { - 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) { ifaces, err := interfaces.GetLoopbackInterfaces() require.NoError(t, err) @@ -126,7 +96,7 @@ func TestServer(t *testing.T) { hwaddr := net.HardwareAddr{1, 2, 3, 4, 5, 6} ifaces[0].HardwareAddr = hwaddr - c, s := setUpClientAndServer(t, ifaces[0], DORAHandler, nil) + c, s := setUpClientAndServer(t, ifaces[0], DORAHandler) defer func() { require.Nil(t, s.Close()) }() @@ -157,84 +127,3 @@ func TestBadAddrFamily(t *testing.T) { t.Fatal("Expected server4.NewServer to fail with an IPv6 address") } } - -func TestServerWithCustomLogger(t *testing.T) { - ifaces, err := interfaces.GetLoopbackInterfaces() - require.NoError(t, err) - require.NotEqual(t, 0, len(ifaces)) - - // lo has a HardwareAddr of "nil". The client will drop all packets - // that don't match the HWAddr of the client interface. - hwaddr := net.HardwareAddr{1, 2, 3, 4, 5, 6} - ifaces[0].HardwareAddr = hwaddr - - c, s := setUpClientAndServer(t, ifaces[0], DORAHandler, &customLogger{ - tb: t, - }) - defer func() { - require.Nil(t, s.Close()) - }() - - xid := dhcpv4.TransactionID{0xaa, 0xbb, 0xcc, 0xdd} - - modifiers := []dhcpv4.Modifier{ - dhcpv4.WithTransactionID(xid), - dhcpv4.WithHwAddr(ifaces[0].HardwareAddr), - } - - offer, ack, err := c.Request(context.Background(), modifiers...) - require.NoError(t, err) - require.NotNil(t, offer, ack) - for _, p := range []*dhcpv4.DHCPv4{offer, ack} { - require.Equal(t, xid, p.TransactionID) - require.Equal(t, ifaces[0].HardwareAddr, p.ClientHWAddr) - } - go func() { - time.Sleep(time.Second * 5) - require.Equal(t, true, s.logger.(*customLogger).called) - }() -} - -func TestServerInstantiationWithCustomLogger(t *testing.T) { - // strong assumption, I know - loAddr := net.ParseIP("127.0.0.1") - saddr := &net.UDPAddr{ - IP: loAddr, - Port: 0, - } - s, err := NewServer("", saddr, DORAHandler, WithLogger(&customLogger{ - tb: t, - })) - if err != nil { - t.Fatal(err) - } - require.NotNil(t, s) -} - -func TestServerInstantiationWithSummaryLogger(t *testing.T) { - // strong assumption, I know - loAddr := net.ParseIP("127.0.0.1") - saddr := &net.UDPAddr{ - IP: loAddr, - Port: 0, - } - s, err := NewServer("", saddr, DORAHandler, WithSummaryLogger()) - if err != nil { - t.Fatal(err) - } - require.NotNil(t, s) -} - -func TestServerInstantiationWithDebugLogger(t *testing.T) { - // strong assumption, I know - loAddr := net.ParseIP("127.0.0.1") - saddr := &net.UDPAddr{ - IP: loAddr, - Port: 0, - } - s, err := NewServer("", saddr, DORAHandler, WithDebugLogger()) - if err != nil { - t.Fatal(err) - } - require.NotNil(t, s) -} 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) -} |