mirror of
https://gitea.com/Sirherobrine23/tea.git
synced 2024-08-20 22:07:24 +00:00
update go min version Update Vendors: * code.gitea.io/gitea-vet v0.2.0 -> v0.2.1 * code.gitea.io/sdk/gitea v0.13.0 -> v0.13.1 * github.com/AlecAivazis/survey v2.1.1 -> v2.2.2 * github.com/adrg/xdg v0.2.1 -> v0.2.2 * github.com/araddon/dateparse d820a6159ab1 -> 8aadafed4dc4 * github.com/go-git/go-git v5.1.0 -> v5.2.0 * github.com/muesli/termenv v0.7.2 -> v0.7.4 * github.com/stretchr/testify v1.5.1 -> v1.6.1 * github.com/urfave/cli v2.2.0 -> v2.3.0 Co-authored-by: 6543 <6543@obermui.de> Reviewed-on: https://gitea.com/gitea/tea/pulls/250 Reviewed-by: Andrew Thornton <art27@cantab.net> Reviewed-by: mrsdizzie <info@mrsdizzie.com> Co-Authored-By: 6543 <6543@noreply.gitea.io> Co-Committed-By: 6543 <6543@noreply.gitea.io>
163 lines
5.4 KiB
Go
163 lines
5.4 KiB
Go
package assert
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"strings"
|
|
)
|
|
|
|
// httpCode is a helper that returns HTTP code of the response. It returns -1 and
|
|
// an error if building a new request fails.
|
|
func httpCode(handler http.HandlerFunc, method, url string, values url.Values) (int, error) {
|
|
w := httptest.NewRecorder()
|
|
req, err := http.NewRequest(method, url, nil)
|
|
if err != nil {
|
|
return -1, err
|
|
}
|
|
req.URL.RawQuery = values.Encode()
|
|
handler(w, req)
|
|
return w.Code, nil
|
|
}
|
|
|
|
// HTTPSuccess asserts that a specified handler returns a success status code.
|
|
//
|
|
// assert.HTTPSuccess(t, myHandler, "POST", "http://www.google.com", nil)
|
|
//
|
|
// Returns whether the assertion was successful (true) or not (false).
|
|
func HTTPSuccess(t TestingT, handler http.HandlerFunc, method, url string, values url.Values, msgAndArgs ...interface{}) bool {
|
|
if h, ok := t.(tHelper); ok {
|
|
h.Helper()
|
|
}
|
|
code, err := httpCode(handler, method, url, values)
|
|
if err != nil {
|
|
Fail(t, fmt.Sprintf("Failed to build test request, got error: %s", err))
|
|
}
|
|
|
|
isSuccessCode := code >= http.StatusOK && code <= http.StatusPartialContent
|
|
if !isSuccessCode {
|
|
Fail(t, fmt.Sprintf("Expected HTTP success status code for %q but received %d", url+"?"+values.Encode(), code))
|
|
}
|
|
|
|
return isSuccessCode
|
|
}
|
|
|
|
// HTTPRedirect asserts that a specified handler returns a redirect status code.
|
|
//
|
|
// assert.HTTPRedirect(t, myHandler, "GET", "/a/b/c", url.Values{"a": []string{"b", "c"}}
|
|
//
|
|
// Returns whether the assertion was successful (true) or not (false).
|
|
func HTTPRedirect(t TestingT, handler http.HandlerFunc, method, url string, values url.Values, msgAndArgs ...interface{}) bool {
|
|
if h, ok := t.(tHelper); ok {
|
|
h.Helper()
|
|
}
|
|
code, err := httpCode(handler, method, url, values)
|
|
if err != nil {
|
|
Fail(t, fmt.Sprintf("Failed to build test request, got error: %s", err))
|
|
}
|
|
|
|
isRedirectCode := code >= http.StatusMultipleChoices && code <= http.StatusTemporaryRedirect
|
|
if !isRedirectCode {
|
|
Fail(t, fmt.Sprintf("Expected HTTP redirect status code for %q but received %d", url+"?"+values.Encode(), code))
|
|
}
|
|
|
|
return isRedirectCode
|
|
}
|
|
|
|
// HTTPError asserts that a specified handler returns an error status code.
|
|
//
|
|
// assert.HTTPError(t, myHandler, "POST", "/a/b/c", url.Values{"a": []string{"b", "c"}}
|
|
//
|
|
// Returns whether the assertion was successful (true) or not (false).
|
|
func HTTPError(t TestingT, handler http.HandlerFunc, method, url string, values url.Values, msgAndArgs ...interface{}) bool {
|
|
if h, ok := t.(tHelper); ok {
|
|
h.Helper()
|
|
}
|
|
code, err := httpCode(handler, method, url, values)
|
|
if err != nil {
|
|
Fail(t, fmt.Sprintf("Failed to build test request, got error: %s", err))
|
|
}
|
|
|
|
isErrorCode := code >= http.StatusBadRequest
|
|
if !isErrorCode {
|
|
Fail(t, fmt.Sprintf("Expected HTTP error status code for %q but received %d", url+"?"+values.Encode(), code))
|
|
}
|
|
|
|
return isErrorCode
|
|
}
|
|
|
|
// HTTPStatusCode asserts that a specified handler returns a specified status code.
|
|
//
|
|
// assert.HTTPStatusCode(t, myHandler, "GET", "/notImplemented", nil, 501)
|
|
//
|
|
// Returns whether the assertion was successful (true) or not (false).
|
|
func HTTPStatusCode(t TestingT, handler http.HandlerFunc, method, url string, values url.Values, statuscode int, msgAndArgs ...interface{}) bool {
|
|
if h, ok := t.(tHelper); ok {
|
|
h.Helper()
|
|
}
|
|
code, err := httpCode(handler, method, url, values)
|
|
if err != nil {
|
|
Fail(t, fmt.Sprintf("Failed to build test request, got error: %s", err))
|
|
}
|
|
|
|
successful := code == statuscode
|
|
if !successful {
|
|
Fail(t, fmt.Sprintf("Expected HTTP status code %d for %q but received %d", statuscode, url+"?"+values.Encode(), code))
|
|
}
|
|
|
|
return successful
|
|
}
|
|
|
|
// HTTPBody is a helper that returns HTTP body of the response. It returns
|
|
// empty string if building a new request fails.
|
|
func HTTPBody(handler http.HandlerFunc, method, url string, values url.Values) string {
|
|
w := httptest.NewRecorder()
|
|
req, err := http.NewRequest(method, url+"?"+values.Encode(), nil)
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
handler(w, req)
|
|
return w.Body.String()
|
|
}
|
|
|
|
// HTTPBodyContains asserts that a specified handler returns a
|
|
// body that contains a string.
|
|
//
|
|
// assert.HTTPBodyContains(t, myHandler, "GET", "www.google.com", nil, "I'm Feeling Lucky")
|
|
//
|
|
// Returns whether the assertion was successful (true) or not (false).
|
|
func HTTPBodyContains(t TestingT, handler http.HandlerFunc, method, url string, values url.Values, str interface{}, msgAndArgs ...interface{}) bool {
|
|
if h, ok := t.(tHelper); ok {
|
|
h.Helper()
|
|
}
|
|
body := HTTPBody(handler, method, url, values)
|
|
|
|
contains := strings.Contains(body, fmt.Sprint(str))
|
|
if !contains {
|
|
Fail(t, fmt.Sprintf("Expected response body for \"%s\" to contain \"%s\" but found \"%s\"", url+"?"+values.Encode(), str, body))
|
|
}
|
|
|
|
return contains
|
|
}
|
|
|
|
// HTTPBodyNotContains asserts that a specified handler returns a
|
|
// body that does not contain a string.
|
|
//
|
|
// assert.HTTPBodyNotContains(t, myHandler, "GET", "www.google.com", nil, "I'm Feeling Lucky")
|
|
//
|
|
// Returns whether the assertion was successful (true) or not (false).
|
|
func HTTPBodyNotContains(t TestingT, handler http.HandlerFunc, method, url string, values url.Values, str interface{}, msgAndArgs ...interface{}) bool {
|
|
if h, ok := t.(tHelper); ok {
|
|
h.Helper()
|
|
}
|
|
body := HTTPBody(handler, method, url, values)
|
|
|
|
contains := strings.Contains(body, fmt.Sprint(str))
|
|
if contains {
|
|
Fail(t, fmt.Sprintf("Expected response body for \"%s\" to NOT contain \"%s\" but found \"%s\"", url+"?"+values.Encode(), str, body))
|
|
}
|
|
|
|
return !contains
|
|
}
|