Commit b835c46c authored by hzxuzhonghu's avatar hzxuzhonghu

fix using defer in loop in cors test

parent cd2692c4
...@@ -42,21 +42,22 @@ func TestCORSAllowedOrigins(t *testing.T) { ...@@ -42,21 +42,22 @@ func TestCORSAllowedOrigins(t *testing.T) {
http.HandlerFunc(func(http.ResponseWriter, *http.Request) {}), http.HandlerFunc(func(http.ResponseWriter, *http.Request) {}),
item.allowedOrigins, nil, nil, nil, "true", item.allowedOrigins, nil, nil, nil, "true",
) )
server := httptest.NewServer(handler) var response *http.Response
defer server.Close() func() {
client := http.Client{} server := httptest.NewServer(handler)
defer server.Close()
request, err := http.NewRequest("GET", server.URL+"/version", nil)
if err != nil { request, err := http.NewRequest("GET", server.URL+"/version", nil)
t.Errorf("unexpected error: %v", err) if err != nil {
} t.Errorf("unexpected error: %v", err)
request.Header.Set("Origin", item.origin) }
request.Header.Set("Origin", item.origin)
response, err := client.Do(request) client := http.Client{}
if err != nil { response, err = client.Do(request)
t.Errorf("unexpected error: %v", err) if err != nil {
} t.Errorf("unexpected error: %v", err)
}
}()
if item.allowed { if item.allowed {
if !reflect.DeepEqual(item.origin, response.Header.Get("Access-Control-Allow-Origin")) { if !reflect.DeepEqual(item.origin, response.Header.Get("Access-Control-Allow-Origin")) {
t.Errorf("Expected %#v, Got %#v", item.origin, response.Header.Get("Access-Control-Allow-Origin")) t.Errorf("Expected %#v, Got %#v", item.origin, response.Header.Get("Access-Control-Allow-Origin"))
...@@ -131,20 +132,22 @@ func TestCORSAllowedMethods(t *testing.T) { ...@@ -131,20 +132,22 @@ func TestCORSAllowedMethods(t *testing.T) {
http.HandlerFunc(func(http.ResponseWriter, *http.Request) {}), http.HandlerFunc(func(http.ResponseWriter, *http.Request) {}),
[]string{".*"}, test.allowedMethods, nil, nil, "true", []string{".*"}, test.allowedMethods, nil, nil, "true",
) )
server := httptest.NewServer(handler) var response *http.Response
defer server.Close() func() {
client := http.Client{} server := httptest.NewServer(handler)
defer server.Close()
request, err := http.NewRequest(test.method, server.URL+"/version", nil)
if err != nil { request, err := http.NewRequest(test.method, server.URL+"/version", nil)
t.Fatalf("unexpected error: %v", err) if err != nil {
} t.Fatalf("unexpected error: %v", err)
request.Header.Set("Origin", "allowed.com") }
request.Header.Set("Origin", "allowed.com")
response, err := client.Do(request) client := http.Client{}
if err != nil { response, err = client.Do(request)
t.Fatalf("unexpected error: %v", err) if err != nil {
} t.Fatalf("unexpected error: %v", err)
}
}()
methodAllowed := allowsMethod(response, test.method) methodAllowed := allowsMethod(response, test.method)
switch { switch {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment