From e68b9d00a6e05b3a941f63ffb696f91e554ac5ec Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Fri, 18 Oct 2024 20:33:49 +0200 Subject: Adding upstream version 9.0.3. Signed-off-by: Daniel Baumann --- modules/test/utils.go | 48 ++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 48 insertions(+) create mode 100644 modules/test/utils.go (limited to 'modules/test/utils.go') diff --git a/modules/test/utils.go b/modules/test/utils.go new file mode 100644 index 0000000..3d884b6 --- /dev/null +++ b/modules/test/utils.go @@ -0,0 +1,48 @@ +// Copyright 2017 The Gitea Authors. All rights reserved. +// SPDX-License-Identifier: MIT + +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 { + 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 { + return strings.Contains(s, `