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 dhcpv4
import (
"testing"
"github.com/stretchr/testify/require"
)
var (
sampleVIVCOpt = VIVCIdentifiers{
VIVCIdentifier{EntID: 9, Data: []byte("CiscoIdentifier")},
VIVCIdentifier{EntID: 18, Data: []byte("WellfleetIdentifier")},
}
sampleVIVCOptRaw = []byte{
0x0, 0x0, 0x0, 0x9, // enterprise id 9
0xf, // length
'C', 'i', 's', 'c', 'o', 'I', 'd', 'e', 'n', 't', 'i', 'f', 'i', 'e', 'r',
0x0, 0x0, 0x0, 0x12, // enterprise id 18
0x13, // length
'W', 'e', 'l', 'l', 'f', 'l', 'e', 'e', 't', 'I', 'd', 'e', 'n', 't', 'i', 'f', 'i', 'e', 'r',
}
)
func TestOptVIVCInterfaceMethods(t *testing.T) {
opt := OptVIVC(sampleVIVCOpt...)
require.Equal(t, OptionVendorIdentifyingVendorClass, opt.Code, "Code")
require.Equal(t, sampleVIVCOptRaw, opt.Value.ToBytes(), "ToBytes")
require.Equal(t, "Vendor-Identifying Vendor Class: 9:'CiscoIdentifier', 18:'WellfleetIdentifier'",
opt.String())
}
func TestParseOptVICO(t *testing.T) {
m, _ := New(WithGeneric(OptionVendorIdentifyingVendorClass, sampleVIVCOptRaw))
o := m.VIVC()
require.Equal(t, sampleVIVCOpt, o)
// Identifier len too long
data := make([]byte, len(sampleVIVCOptRaw))
copy(data, sampleVIVCOptRaw)
data[4] = 40
m, _ = New(WithGeneric(OptionVendorIdentifyingVendorClass, data))
o = m.VIVC()
require.Nil(t, o, "should get error from bad length")
// Longer than length
data[4] = 5
m, _ = New(WithGeneric(OptionVendorIdentifyingVendorClass, data[:10]))
o = m.VIVC()
require.Equal(t, o[0].Data, []byte("Cisco"))
m, _ = New()
require.Equal(t, VIVCIdentifiers(nil), m.VIVC())
}
|