如何在golang中测试http.Redirect目标地址

package main

import (
    "net/http"
    "net/http/httptest"
    "testing"
)

func TestRedirect(t *testing.T) {
    handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        http.Redirect(w, r, "/target", http.StatusFound)
    })

    req, err := http.NewRequest("GET", "/source", nil)
    if err != nil {
        t.Fatal(err)
    }

    rr := httptest.NewRecorder()
    handler.ServeHTTP(rr, req)

    if rr.Code != http.StatusFound {
        t.Errorf("Expected status code %d, but got %d", http.StatusFound, rr.Code)
    }

    expectedLocation := "/target"
    if rr.Header().Get("Location") != expectedLocation {
        t.Errorf("Expected redirect location %s, but got %s", expectedLocation, rr.Header().Get("Location"))
    }
}