1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53
|
package dhcpv6
import (
"testing"
"github.com/insomniacslk/dhcp/iana"
"github.com/stretchr/testify/require"
)
func TestParseOptStatusCode(t *testing.T) {
data := []byte{
0, 5, // StatusUseMulticast
'u', 's', 'e', ' ', 'm', 'u', 'l', 't', 'i', 'c', 'a', 's', 't',
}
opt, err := ParseOptStatusCode(data)
require.NoError(t, err)
require.Equal(t, iana.StatusUseMulticast, opt.StatusCode)
require.Equal(t, "use multicast", opt.StatusMessage)
}
func TestOptStatusCodeToBytes(t *testing.T) {
expected := []byte{
0, 0, // StatusSuccess
's', 'u', 'c', 'c', 'e', 's', 's',
}
opt := OptStatusCode{
StatusCode: iana.StatusSuccess,
StatusMessage: "success",
}
actual := opt.ToBytes()
require.Equal(t, expected, actual)
}
func TestOptStatusCodeParseOptStatusCodeTooShort(t *testing.T) {
_, err := ParseOptStatusCode([]byte{0})
require.Error(t, err, "ParseOptStatusCode: Expected error on truncated option")
}
func TestOptStatusCodeString(t *testing.T) {
data := []byte{
0, 5, // StatusUseMulticast
'u', 's', 'e', ' ', 'm', 'u', 'l', 't', 'i', 'c', 'a', 's', 't',
}
opt, err := ParseOptStatusCode(data)
require.NoError(t, err)
require.Contains(
t,
opt.String(),
"Code: UseMulticast (5); Message: use multicast",
"String() should contain the code and message",
)
}
|