summaryrefslogtreecommitdiffstats
path: root/modules/test/utils.go
diff options
context:
space:
mode:
Diffstat (limited to 'modules/test/utils.go')
-rw-r--r--modules/test/utils.go19
1 files changed, 18 insertions, 1 deletions
diff --git a/modules/test/utils.go b/modules/test/utils.go
index 282895eaa9..2917741c45 100644
--- a/modules/test/utils.go
+++ b/modules/test/utils.go
@@ -5,12 +5,29 @@ package test
import (
"net/http"
+ "net/http/httptest"
"strings"
+
+ "code.gitea.io/gitea/modules/json"
)
// RedirectURL returns the redirect URL of a http response.
+// It also works for JSONRedirect: `{"redirect": "..."}`
func RedirectURL(resp http.ResponseWriter) string {
- return resp.Header().Get("Location")
+ loc := resp.Header().Get("Location")
+ if loc != "" {
+ return loc
+ }
+ if r, ok := resp.(*httptest.ResponseRecorder); ok {
+ m := map[string]any{}
+ err := json.Unmarshal(r.Body.Bytes(), &m)
+ if err == nil {
+ if loc, ok := m["redirect"].(string); ok {
+ return loc
+ }
+ }
+ }
+ return ""
}
func IsNormalPageCompleted(s string) bool {