|
|
@ -17,12 +17,14 @@ |
|
|
|
package cmd |
|
|
|
package cmd |
|
|
|
|
|
|
|
|
|
|
|
import ( |
|
|
|
import ( |
|
|
|
|
|
|
|
"encoding/json" |
|
|
|
"fmt" |
|
|
|
"fmt" |
|
|
|
"net" |
|
|
|
"net" |
|
|
|
"net/http" |
|
|
|
"net/http" |
|
|
|
"net/url" |
|
|
|
"net/url" |
|
|
|
"reflect" |
|
|
|
"reflect" |
|
|
|
"runtime" |
|
|
|
"runtime" |
|
|
|
|
|
|
|
"strings" |
|
|
|
"testing" |
|
|
|
"testing" |
|
|
|
) |
|
|
|
) |
|
|
|
|
|
|
|
|
|
|
@ -413,3 +415,44 @@ func TestCheckURL(t *testing.T) { |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// Testing dumping request function.
|
|
|
|
|
|
|
|
func TestDumpRequest(t *testing.T) { |
|
|
|
|
|
|
|
req, err := http.NewRequest("GET", "http://localhost:9000?prefix=Hello%2AWorld%2A", nil) |
|
|
|
|
|
|
|
if err != nil { |
|
|
|
|
|
|
|
t.Fatal(err) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
req.Header.Set("content-md5", "====test") |
|
|
|
|
|
|
|
jsonReq := dumpRequest(req) |
|
|
|
|
|
|
|
type jsonResult struct { |
|
|
|
|
|
|
|
Method string `json:"method"` |
|
|
|
|
|
|
|
Path string `json:"path"` |
|
|
|
|
|
|
|
Query string `json:"query"` |
|
|
|
|
|
|
|
Header http.Header `json:"header"` |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
jsonReq = strings.Replace(jsonReq, "%%", "%", -1) |
|
|
|
|
|
|
|
res := jsonResult{} |
|
|
|
|
|
|
|
if err = json.Unmarshal([]byte(jsonReq), &res); err != nil { |
|
|
|
|
|
|
|
t.Fatal(err) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// Look for expected method.
|
|
|
|
|
|
|
|
if res.Method != "GET" { |
|
|
|
|
|
|
|
t.Fatalf("Unexpected method %s, expected 'GET'", res.Method) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// Look for expected query values
|
|
|
|
|
|
|
|
expectedQuery := url.Values{} |
|
|
|
|
|
|
|
expectedQuery.Set("prefix", "Hello*World*") |
|
|
|
|
|
|
|
if !reflect.DeepEqual(res.Query, expectedQuery.Encode()) { |
|
|
|
|
|
|
|
t.Fatalf("Expected %#v, got %#v", expectedQuery, res.Query) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// Look for expected header.
|
|
|
|
|
|
|
|
expectedHeader := http.Header{} |
|
|
|
|
|
|
|
expectedHeader.Set("content-md5", "====test") |
|
|
|
|
|
|
|
expectedHeader.Set("host", "localhost:9000") |
|
|
|
|
|
|
|
if !reflect.DeepEqual(res.Header, expectedHeader) { |
|
|
|
|
|
|
|
t.Fatalf("Expected %#v, got %#v", expectedHeader, res.Header) |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|