diff options
Diffstat (limited to 'dhcpv4/option_subnet_mask_test.go')
| -rw-r--r-- | dhcpv4/option_subnet_mask_test.go | 37 |
1 files changed, 15 insertions, 22 deletions
diff --git a/dhcpv4/option_subnet_mask_test.go b/dhcpv4/option_subnet_mask_test.go index f04a481..bc82cf1 100644 --- a/dhcpv4/option_subnet_mask_test.go +++ b/dhcpv4/option_subnet_mask_test.go @@ -7,30 +7,23 @@ import ( "github.com/stretchr/testify/require" ) -func TestOptSubnetMaskInterfaceMethods(t *testing.T) { - mask := net.IPMask{255, 255, 255, 0} - o := OptSubnetMask{SubnetMask: mask} - - require.Equal(t, OptionSubnetMask, o.Code(), "Code") - - expectedBytes := []byte{255, 255, 255, 0} - require.Equal(t, expectedBytes, o.ToBytes(), "ToBytes") - - require.Equal(t, "Subnet Mask -> ffffff00", o.String(), "String") +func TestOptSubnetMask(t *testing.T) { + o := OptSubnetMask(net.IPMask{255, 255, 255, 0}) + require.Equal(t, o.Code, OptionSubnetMask, "Code") + require.Equal(t, "Subnet Mask: ffffff00", o.String(), "String") + require.Equal(t, []byte{255, 255, 255, 0}, o.Value.ToBytes(), "ToBytes") } -func TestParseOptSubnetMask(t *testing.T) { - var ( - o *OptSubnetMask - err error - ) - o, err = ParseOptSubnetMask([]byte{}) - require.Error(t, err, "empty byte stream") +func TestGetSubnetMask(t *testing.T) { + o := OptionsFromList(OptSubnetMask(net.IPMask{})) + mask := GetSubnetMask(o) + require.Nil(t, mask, "empty byte stream") - o, err = ParseOptSubnetMask([]byte{255}) - require.Error(t, err, "short byte stream") + o = OptionsFromList(OptSubnetMask(net.IPMask{255})) + mask = GetSubnetMask(o) + require.Nil(t, mask, "short byte stream") - o, err = ParseOptSubnetMask([]byte{255, 255, 255, 0}) - require.NoError(t, err) - require.Equal(t, net.IPMask{255, 255, 255, 0}, o.SubnetMask) + o = OptionsFromList(OptSubnetMask(net.IPMask{255, 255, 255, 0})) + mask = GetSubnetMask(o) + require.Equal(t, net.IPMask{255, 255, 255, 0}, mask) } |