916ec9acab
defer test.MockVariable(&variable, 1234)() (cherry picked from commit9c78752444) (cherry picked from commit8ab559df0d) (cherry picked from commit2e7fe1ec95) (cherry picked from commitf9618b8896) Conflicts: modules/test/utils.go https://codeberg.org/forgejo/forgejo/issues/1219
32 lines
682 B
Go
32 lines
682 B
Go
// Copyright 2017 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package test
|
|
|
|
import (
|
|
"net/http"
|
|
"strings"
|
|
)
|
|
|
|
// RedirectURL returns the redirect URL of a http response.
|
|
func RedirectURL(resp http.ResponseWriter) string {
|
|
return resp.Header().Get("Location")
|
|
}
|
|
|
|
func IsNormalPageCompleted(s string) bool {
|
|
return strings.Contains(s, `<footer class="page-footer"`) && strings.Contains(s, `</html>`)
|
|
}
|
|
|
|
func MockVariableValue[T any](p *T, v T) (reset func()) {
|
|
old := *p
|
|
*p = v
|
|
return func() { *p = old }
|
|
}
|
|
|
|
func MockVariable[T any](variable *T, mock T) func() {
|
|
original := *variable
|
|
*variable = mock
|
|
return func() {
|
|
*variable = original
|
|
}
|
|
}
|