diff options
Diffstat (limited to 'dhcpv6/server_test.go')
| -rw-r--r-- | dhcpv6/server_test.go | 26 |
1 files changed, 5 insertions, 21 deletions
diff --git a/dhcpv6/server_test.go b/dhcpv6/server_test.go index 08cb507..436e808 100644 --- a/dhcpv6/server_test.go +++ b/dhcpv6/server_test.go @@ -5,28 +5,11 @@ import ( "net" "testing" "time" - "errors" + "github.com/insomniacslk/dhcp/interfaces" "github.com/stretchr/testify/require" ) -// utility function to return the loopback interface name -func getLoopbackInterface() (string, error) { - var ifaces []net.Interface - var err error - if ifaces, err = net.Interfaces(); err != nil { - return "", err - } - - for _, iface := range ifaces { - if iface.Flags & net.FlagLoopback != 0 || iface.Name[:2] == "lo" { - return iface.Name, nil - } - } - - return "", errors.New("No loopback interface found") -} - // 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) (*Client, *Server) { @@ -39,7 +22,7 @@ func setUpClientAndServer(handler Handler) (*Client, *Server) { c := NewClient() c.LocalAddr = &net.UDPAddr{ - IP: net.ParseIP("::1"), + IP: net.ParseIP("::1"), } for { if s.LocalAddr() != nil { @@ -85,9 +68,10 @@ func TestServerActivateAndServe(t *testing.T) { c, s := setUpClientAndServer(handler) defer s.Close() - iface, err := getLoopbackInterface() + ifaces, err := interfaces.GetLoopbackInterfaces() require.NoError(t, err) + require.NotEqual(t, 0, len(ifaces)) - _, _, err = c.Solicit(iface) + _, _, err = c.Solicit(ifaces[0].Name) require.NoError(t, err) } |