diff options
Diffstat (limited to 'dhcpv4/option_ntp_servers_test.go')
| -rw-r--r-- | dhcpv4/option_ntp_servers_test.go | 24 |
1 files changed, 5 insertions, 19 deletions
diff --git a/dhcpv4/option_ntp_servers_test.go b/dhcpv4/option_ntp_servers_test.go index e7bcefd..4d321ff 100644 --- a/dhcpv4/option_ntp_servers_test.go +++ b/dhcpv4/option_ntp_servers_test.go @@ -25,37 +25,23 @@ func TestParseOptNTPServers(t *testing.T) { 192, 168, 0, 10, // NTP server #1 192, 168, 0, 20, // NTP server #2 } - o, err := ParseOptNTPServers(data) + o, err := ParseOptNTPServers(data[2:]) require.NoError(t, err) ntpServers := []net.IP{ - net.IPv4(192, 168, 0, 10), - net.IPv4(192, 168, 0, 20), + net.IP{192, 168, 0, 10}, + net.IP{192, 168, 0, 20}, } require.Equal(t, &OptNTPServers{NTPServers: ntpServers}, o) - // Short byte stream - data = []byte{byte(OptionNTPServers)} - _, err = ParseOptNTPServers(data) - require.Error(t, err, "should get error from short byte stream") - - // Wrong code - data = []byte{54, 2, 1, 1} - _, err = ParseOptNTPServers(data) - require.Error(t, err, "should get error from wrong code") - // Bad length - data = []byte{byte(OptionNTPServers), 6, 1, 1, 1} + data = []byte{1, 1, 1} _, err = ParseOptNTPServers(data) require.Error(t, err, "should get error from bad length") } func TestParseOptNTPserversNoNTPServers(t *testing.T) { // RFC2132 requires that at least one NTP server IP is specified - data := []byte{ - byte(OptionNTPServers), - 0, // Length - } - _, err := ParseOptNTPServers(data) + _, err := ParseOptNTPServers([]byte{}) require.Error(t, err) } |