mirrors/coredhcp
1
0
Fork 0
mirror of https://github.com/coredhcp/coredhcp.git synced 2026-05-05 12:06:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

plugins/server_id: Add a happy-path test

Browse Source 
Signed-off-by: Anatole Denis <anatole@unverle.fr>
This commit is contained in:
Anatole Denis 2019-09-11 18:48:59 +02:00 committed by Anatole Denis
parent 4a73abd6b6
commit 5cd963f1a3
1 changed files with 30 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
plugins/server_id/plugin_test.go
View File

@ -37,3 +37,33 @@ func TestRejectBadServerIDV6(t *testing.T) {
t.Error("server_id did not interrupt processing on a request with mismatched ServerID")
}
}
func TestAddServerIDV6(t *testing.T) {
req, err := dhcpv6.NewMessage()
if err != nil {
t.Fatal(err)
}
v6ServerID = makeTestDUID("0000000000000000")
req.MessageType = dhcpv6.MessageTypeRebind
dhcpv6.WithClientID(*makeTestDUID("1000000000000000"))(req)
stub, err := dhcpv6.NewReplyFromMessage(req)
if err != nil {
t.Fatal(err)
}
resp, _ := Handler6(req, stub)
if resp == nil {
t.Fatal("plugin did not return an answer")
}
if opt := resp.GetOneOption(dhcpv6.OptionServerID); opt == nil {
t.Fatal("plugin did not add a ServerID option")
} else {
sid := opt.(*dhcpv6.OptServerId)
if !sid.Sid.Equal(*v6ServerID) {
t.Fatalf("Got unexpected DUID: expected %v, got %v", v6ServerID, sid.Sid)
}
}
}