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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
|
package dhcp6server_test
import (
"testing"
"github.com/mdlayher/dhcp6"
"github.com/mdlayher/dhcp6/dhcp6server"
"github.com/mdlayher/dhcp6/dhcp6test"
)
// TestServeMuxHandleNoResponse verifies that no Handler is invoked when a
// ServeMux does not have a Handler registered for a given message type.
func TestServeMuxHandleNoResponse(t *testing.T) {
mux := dhcp6server.NewServeMux()
r, err := dhcp6server.ParseRequest([]byte{1, 1, 2, 3}, nil)
if err != nil {
t.Fatal(err)
}
w := dhcp6test.NewRecorder(r.TransactionID)
mux.ServeDHCP(w, r)
if mt := w.MessageType; mt != dhcp6.MessageType(0) {
t.Fatalf("reply packet empty, but got message type: %v", mt)
}
if l := len(w.Options()); l > 0 {
t.Fatalf("reply packet empty, but got %d options", l)
}
}
// TestServeMuxHandleOK verifies that a Handler is invoked when a ServeMux
// has a Handler registered for a given message type.
func TestServeMuxHandleOK(t *testing.T) {
mux := dhcp6server.NewServeMux()
mt := dhcp6.MessageTypeSolicit
mux.Handle(mt, &solicitHandler{})
r, err := dhcp6server.ParseRequest([]byte{byte(mt), 0, 1, 2}, nil)
if err != nil {
t.Fatal(err)
}
w := dhcp6test.NewRecorder(r.TransactionID)
mux.ServeDHCP(w, r)
if want, got := dhcp6.MessageTypeAdvertise, w.MessageType; want != got {
t.Fatalf("unexpected response message type: %v != %v", want, got)
}
}
// TestServeMuxHandleFuncOK verifies that a normal function which can be used
// as a Handler is invoked when a ServeMux has a HandlerFunc registered for
// a given message type.
func TestServeMuxHandleFuncOK(t *testing.T) {
mux := dhcp6server.NewServeMux()
mt := dhcp6.MessageTypeSolicit
mux.HandleFunc(mt, solicit)
r, err := dhcp6server.ParseRequest([]byte{byte(mt), 0, 1, 2}, nil)
if err != nil {
t.Fatal(err)
}
w := dhcp6test.NewRecorder(r.TransactionID)
mux.ServeDHCP(w, r)
if want, got := dhcp6.MessageTypeAdvertise, w.MessageType; want != got {
t.Fatalf("unexpected response message type: %v != %v", want, got)
}
}
// solicitHandler is a Handler which returns an Advertise in reply
// to a Solicit request.
type solicitHandler struct{}
func (h *solicitHandler) ServeDHCP(w dhcp6server.ResponseSender, r *dhcp6server.Request) {
solicit(w, r)
}
// solicit is a function which can be adapted as a HandlerFunc.
func solicit(w dhcp6server.ResponseSender, r *dhcp6server.Request) {
w.Send(dhcp6.MessageTypeAdvertise)
}
|
