如何在Golang中测试http.Redirect目标URL

Question

I have set up a http redirect in the go handler, and I'm trying to verify if the target URL is composed as expected. I'm currently using httptest for relevant unit test.

In the handler:

url := fmt.Sprintf("%s%s?token=%s", domain, someURL, token)
http.Redirect(w, r, url, 302)

The current unit test can only verify the respond code but not the target URL:

resp := httptest.NewRecorder()
router.ServeHTTP(resp, req)
assert.Equal(t, 302, resp.Code)

I'm expecting something like

assert.Equal(t, expectedURL, resp.redirectURL)

Answer 1

Assert that the Location header is as expected:

assert.Equal(t, expectedURL, resp.HeaderMap.Get("Location"))