package server import ( "net/http" "testing" ) func TestDirector(t *testing.T) { s := &Server{} req, err := http.NewRequest("GET", "https://cetacean.club/", nil) if err != nil { t.Fatal(err) } req.Header.Add("X-Forwarded-For", "Rick-James") req.Header.Add("X-Client-Ip", "56.32.51.84") s.Director(req) for _, header := range []string{"X-Forwarded-For", "X-Client-Ip"} { t.Run(header, func(t *testing.T) { val := req.Header.Get(header) if val != "" { t.Fatalf("expected header %q to have no value, got: %v", header, val) } }) } }