30 lines
563 B
Go
30 lines
563 B
Go
package main
|
|
|
|
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)
|
|
}
|
|
})
|
|
}
|
|
}
|