dynamic leases

dhcpd/v6_test.go

@@ -86,9 +86,11 @@ func TestV6GetLease(t *testing.T) {
 	resp.AddOption(dhcpv6.OptServerID(s.conf.sid))
 
 	// check "Advertise"
+	assert.Equal(t, dhcpv6.MessageTypeAdvertise, resp.Type())
 	oia := resp.Options.OneIANA()
 	oiaAddr := oia.Options.OneAddress()
 	assert.Equal(t, "2001::1", oiaAddr.IPv6Addr.String())
+	assert.Equal(t, s.conf.leaseTime.Seconds(), oiaAddr.ValidLifetime.Seconds())
 
 	// "Request"
 	req, _ = dhcpv6.NewRequestFromAdvertise(resp)
@@ -97,12 +99,75 @@ func TestV6GetLease(t *testing.T) {
 	assert.True(t, s.process(msg, req, resp))
 
 	// check "Reply"
+	assert.Equal(t, dhcpv6.MessageTypeReply, resp.Type())
 	oia = resp.Options.OneIANA()
 	oiaAddr = oia.Options.OneAddress()
 	assert.Equal(t, "2001::1", oiaAddr.IPv6Addr.String())
+	assert.Equal(t, s.conf.leaseTime.Seconds(), oiaAddr.ValidLifetime.Seconds())
+
 	dnsAddrs := resp.Options.DNS()
 	assert.Equal(t, 1, len(dnsAddrs))
 	assert.Equal(t, "2000::1", dnsAddrs[0].String())
 
+	// check lease
+	ls := s.GetLeases(LeasesStatic)
+	assert.Equal(t, 1, len(ls))
+	assert.Equal(t, "2001::1", ls[0].IP.String())
+	assert.Equal(t, "aa:aa:aa:aa:aa:aa", ls[0].HWAddr.String())
+
+	s.Stop()
+}
+
+func TestV6GetDynamicLease(t *testing.T) {
+	conf := V6ServerConf{
+		Enabled:    true,
+		RangeStart: "2001::2",
+		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")
+
+	// "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"
+	assert.Equal(t, dhcpv6.MessageTypeAdvertise, resp.Type())
+	oia := resp.Options.OneIANA()
+	oiaAddr := oia.Options.OneAddress()
+	assert.Equal(t, "2001::2", 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"
+	assert.Equal(t, dhcpv6.MessageTypeReply, resp.Type())
+	oia = resp.Options.OneIANA()
+	oiaAddr = oia.Options.OneAddress()
+	assert.Equal(t, "2001::2", oiaAddr.IPv6Addr.String())
+
+	dnsAddrs := resp.Options.DNS()
+	assert.Equal(t, 1, len(dnsAddrs))
+	assert.Equal(t, "2000::1", dnsAddrs[0].String())
+
+	// check lease
+	ls := s.GetLeases(LeasesDynamic)
+	assert.Equal(t, 1, len(ls))
+	assert.Equal(t, "2001::2", ls[0].IP.String())
+	assert.Equal(t, "aa:aa:aa:aa:aa:aa", ls[0].HWAddr.String())
+
 	s.Stop()
 }