tests

dhcpd/v6_test.go

@@ -4,6 +4,8 @@ import (
 	"net"
 	"testing"
 
+	"github.com/insomniacslk/dhcp/dhcpv6"
+	"github.com/insomniacslk/dhcp/iana"
 	"github.com/stretchr/testify/assert"
 )
 
@@ -54,3 +56,53 @@ func TestV6StaticLease(t *testing.T) {
 
 	s.Stop()
 }
+
+func TestV6GetLease(t *testing.T) {
+	conf := V6ServerConf{
+		Enabled:    true,
+		RangeStart: "2001::1",
+		notify:     notify,
+	}
+	s, err := v6Create(conf)
+	assert.True(t, err == nil)
+	s.conf.dnsIPAddrs = []net.IP{net.ParseIP("2000::1")}
+	s.conf.sid = dhcpv6.Duid{
+		Type:   dhcpv6.DUID_LLT,
+		HwType: iana.HWTypeEthernet,
+	}
+	s.conf.sid.LinkLayerAddr, _ = net.ParseMAC("aa:aa:aa:aa:aa:aa")
+
+	l := Lease{}
+	l.IP = net.ParseIP("2001::1")
+	l.HWAddr, _ = net.ParseMAC("aa:aa:aa:aa:aa:aa")
+	assert.True(t, s.AddStaticLease(l) == nil)
+
+	// "Solicit"
+	mac, _ := net.ParseMAC("aa:aa:aa:aa:aa:aa")
+	req, _ := dhcpv6.NewSolicit(mac)
+	msg, _ := req.GetInnerMessage()
+	resp, _ := dhcpv6.NewAdvertiseFromSolicit(msg)
+	assert.True(t, s.process(msg, req, resp))
+	resp.AddOption(dhcpv6.OptServerID(s.conf.sid))
+
+	// check "Advertise"
+	oia := resp.Options.OneIANA()
+	oiaAddr := oia.Options.OneAddress()
+	assert.Equal(t, "2001::1", oiaAddr.IPv6Addr.String())
+
+	// "Request"
+	req, _ = dhcpv6.NewRequestFromAdvertise(resp)
+	msg, _ = req.GetInnerMessage()
+	resp, _ = dhcpv6.NewReplyFromMessage(msg)
+	assert.True(t, s.process(msg, req, resp))
+
+	// check "Reply"
+	oia = resp.Options.OneIANA()
+	oiaAddr = oia.Options.OneAddress()
+	assert.Equal(t, "2001::1", oiaAddr.IPv6Addr.String())
+	dnsAddrs := resp.Options.DNS()
+	assert.Equal(t, 1, len(dnsAddrs))
+	assert.Equal(t, "2000::1", dnsAddrs[0].String())
+
+	s.Stop()
+}