1 files changed, 5 insertions, 19 deletions
diff --git a/dhcpv6/dhcpv6_test.go b/dhcpv6/dhcpv6_test.go
index bc56ea3..6b7d2df 100644
--- a/dhcpv6/dhcpv6_test.go
+++ b/dhcpv6/dhcpv6_test.go
@@ -37,7 +37,7 @@ func TestNewMessage(t *testing.T) {
 	require.Empty(t, d.(*DHCPv6Message).options)
 }
 
-func TestDecapsulateRelay(t *testing.T) {
+func TestDecapsulateRelayIndex(t *testing.T) {
 	m := DHCPv6Message{}
 	r1, err := EncapsulateRelay(&m, RELAY_FORW, net.IPv6linklocalallnodes, net.IPv6interfacelocalallnodes)
 	require.NoError(t, err)
@@ -46,7 +46,7 @@ func TestDecapsulateRelay(t *testing.T) {
 	r3, err := EncapsulateRelay(r2, RELAY_FORW, net.IPv6unspecified, net.IPv6linklocalallrouters)
 	require.NoError(t, err)
 
-	first, err := DecapsulateRelay(r3, 0)
+	first, err := DecapsulateRelayIndex(r3, 0)
 	require.NoError(t, err)
 	relay, ok := first.(*DHCPv6Relay)
 	require.True(t, ok)
@@ -54,7 +54,7 @@ func TestDecapsulateRelay(t *testing.T) {
 	require.Equal(t, relay.LinkAddr(), net.IPv6loopback)
 	require.Equal(t, relay.PeerAddr(), net.IPv6linklocalallnodes)
 
-	second, err := DecapsulateRelay(r3, 1)
+	second, err := DecapsulateRelayIndex(r3, 1)
 	require.NoError(t, err)
 	relay, ok = second.(*DHCPv6Relay)
 	require.True(t, ok)
@@ -62,32 +62,18 @@ func TestDecapsulateRelay(t *testing.T) {
 	require.Equal(t, relay.LinkAddr(), net.IPv6linklocalallnodes)
 	require.Equal(t, relay.PeerAddr(), net.IPv6interfacelocalallnodes)
 
-	third, err := DecapsulateRelay(r3, 2)
+	third, err := DecapsulateRelayIndex(r3, 2)
 	require.NoError(t, err)
 	_, ok = third.(*DHCPv6Message)
 	require.True(t, ok)
 
-	_, err = DecapsulateRelay(r3, 3)
-	require.Error(t, err)
-
-	rfirst, err := DecapsulateRelay(r3, -1)
+	rfirst, err := DecapsulateRelayIndex(r3, -1)
 	require.NoError(t, err)
 	relay, ok = rfirst.(*DHCPv6Relay)
 	require.True(t, ok)
 	require.Equal(t, relay.HopCount(), uint8(0))
 	require.Equal(t, relay.LinkAddr(), net.IPv6linklocalallnodes)
 	require.Equal(t, relay.PeerAddr(), net.IPv6interfacelocalallnodes)
-
-	rsecond, err := DecapsulateRelay(r3, -2)
-	require.NoError(t, err)
-	relay, ok = rsecond.(*DHCPv6Relay)
-	require.True(t, ok)
-	require.Equal(t, relay.HopCount(), uint8(1))
-	require.Equal(t, relay.LinkAddr(), net.IPv6loopback)
-	require.Equal(t, relay.PeerAddr(), net.IPv6linklocalallnodes)
-
-	_, err = DecapsulateRelay(r3, -3)
-	require.Error(t, err)
 }
 
 func TestSettersAndGetters(t *testing.T) {