diff options
Diffstat (limited to 'dhcpv4/option_ip_address_lease_time_test.go')
| -rw-r--r-- | dhcpv4/option_ip_address_lease_time_test.go | 15 |
1 files changed, 8 insertions, 7 deletions
diff --git a/dhcpv4/option_ip_address_lease_time_test.go b/dhcpv4/option_ip_address_lease_time_test.go index 70c4047..edd709f 100644 --- a/dhcpv4/option_ip_address_lease_time_test.go +++ b/dhcpv4/option_ip_address_lease_time_test.go @@ -15,20 +15,21 @@ func TestOptIPAddressLeaseTime(t *testing.T) { } func TestGetIPAddressLeaseTime(t *testing.T) { - o := Options{OptionIPAddressLeaseTime.Code(): []byte{0, 0, 168, 192}} - leaseTime := GetIPAddressLeaseTime(o, 0) + m, _ := New(WithGeneric(OptionIPAddressLeaseTime, []byte{0, 0, 168, 192})) + leaseTime := m.IPAddressLeaseTime(0) require.Equal(t, 43200*time.Second, leaseTime) // Too short. - o = Options{OptionIPAddressLeaseTime.Code(): []byte{168, 192}} - leaseTime = GetIPAddressLeaseTime(o, 0) + m, _ = New(WithGeneric(OptionIPAddressLeaseTime, []byte{168, 192})) + leaseTime = m.IPAddressLeaseTime(0) require.Equal(t, time.Duration(0), leaseTime) // Too long. - o = Options{OptionIPAddressLeaseTime.Code(): []byte{1, 1, 1, 1, 1}} - leaseTime = GetIPAddressLeaseTime(o, 0) + m, _ = New(WithGeneric(OptionIPAddressLeaseTime, []byte{1, 1, 1, 1, 1})) + leaseTime = m.IPAddressLeaseTime(0) require.Equal(t, time.Duration(0), leaseTime) // Empty. - require.Equal(t, time.Duration(10), GetIPAddressLeaseTime(Options{}, 10)) + m, _ = New() + require.Equal(t, time.Duration(10), m.IPAddressLeaseTime(10)) } |