// 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 := dhcp6.NewServeMux()
mt := dhcp6.MessageTypeSolicit
mux.HandleFunc(mt, solicit)
r, err := dhcp6.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)
}
}
// 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 := dhcp6.NewServeMux()
r, err := dhcp6.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)
}
}