76 lines
1.4 KiB
Go
76 lines
1.4 KiB
Go
package server
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/url"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
type fakeTransport struct{}
|
|
|
|
func (ft fakeTransport) RoundTrip(r *http.Request) (*http.Response, error) {
|
|
return &http.Response{
|
|
Body: r.Body,
|
|
}, nil
|
|
}
|
|
|
|
// empty url -> OK //TODO
|
|
func TestRewrite(t *testing.T) {
|
|
transport := &rewrite{
|
|
rewrites: map[string]string{
|
|
"a": "b",
|
|
},
|
|
baseTransport: fakeTransport{},
|
|
}
|
|
|
|
r, err := http.NewRequest("GET", "asdf", strings.NewReader("mary had a little lamb"))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
resp, err := transport.RoundTrip(r)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
b, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if string(b) != "mbry hbd b little lbmb" {
|
|
t.Errorf("failed to replace: got %q, want \"b\"", b)
|
|
}
|
|
}
|
|
|
|
func TestMapKey(t *testing.T) {
|
|
r := &http.Request{
|
|
Host: "a.b.c:123",
|
|
URL: &url.URL{
|
|
Path: "/c/d/e",
|
|
},
|
|
}
|
|
|
|
if v := mapKey(r, "domain"); v != "a" {
|
|
t.Errorf("failed to get domain: got %v", v)
|
|
}
|
|
|
|
if v := mapKey(r, "path"); v != "c" {
|
|
t.Errorf("failed to get domain: got %v", v)
|
|
}
|
|
|
|
r.Host = "a:123"
|
|
if v := mapKey(r, "domain"); v != "a" {
|
|
t.Errorf("failed to get domain: got %v", v)
|
|
}
|
|
|
|
r.URL.Path = ""
|
|
if v := mapKey(r, "path"); v != "" {
|
|
t.Errorf("failed to get domain: got %v", v)
|
|
}
|
|
|
|
r.URL.Path = "/"
|
|
if v := mapKey(r, "path"); v != "" {
|
|
t.Errorf("failed to get domain: got %v", v)
|
|
}
|
|
}
|