interfaces: added package with interface facilities

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)
 }