summaryrefslogtreecommitdiffhomepage
path: root/dhcpv6/modifiers_test.go
blob: da1ef56b60d87c5a4678939a0186a5788ded5b86 (plain)
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
package dhcpv6

import (
	"net"
	"testing"

	"github.com/insomniacslk/dhcp/iana"
	"github.com/stretchr/testify/require"
)

func TestWithClientID(t *testing.T) {
	duid := Duid{
		Type:          DUID_LL,
		HwType:        iana.HwTypeEthernet,
		LinkLayerAddr: net.HardwareAddr([]byte{0xfa, 0xce, 0xb0, 0x00, 0x00, 0x0c}),
	}
	m, err := NewMessage(WithClientID(duid))
	require.NoError(t, err)
	opt := m.GetOneOption(OPTION_CLIENTID)
	require.NotNil(t, opt)
	cid := opt.(*OptClientId)
	require.Equal(t, cid.Cid, duid)
}

func TestWithServerID(t *testing.T) {
	duid := Duid{
		Type:          DUID_LL,
		HwType:        iana.HwTypeEthernet,
		LinkLayerAddr: net.HardwareAddr([]byte{0xfa, 0xce, 0xb0, 0x00, 0x00, 0x0c}),
	}
	m, err := NewMessage(WithServerID(duid))
	require.NoError(t, err)
	opt := m.GetOneOption(OPTION_SERVERID)
	require.NotNil(t, opt)
	sid := opt.(*OptServerId)
	require.Equal(t, sid.Sid, duid)
}