87 lines
3.5 KiB
Go
87 lines
3.5 KiB
Go
package assert
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
"testing"
|
|
)
|
|
|
|
func httpOK(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusOK)
|
|
}
|
|
|
|
func httpRedirect(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusTemporaryRedirect)
|
|
}
|
|
|
|
func httpError(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
}
|
|
|
|
func TestHTTPStatuses(t *testing.T) {
|
|
assert := New(t)
|
|
mockT := new(testing.T)
|
|
|
|
assert.Equal(HTTPSuccess(mockT, httpOK, "GET", "/", nil), true)
|
|
assert.Equal(HTTPSuccess(mockT, httpRedirect, "GET", "/", nil), false)
|
|
assert.Equal(HTTPSuccess(mockT, httpError, "GET", "/", nil), false)
|
|
|
|
assert.Equal(HTTPRedirect(mockT, httpOK, "GET", "/", nil), false)
|
|
assert.Equal(HTTPRedirect(mockT, httpRedirect, "GET", "/", nil), true)
|
|
assert.Equal(HTTPRedirect(mockT, httpError, "GET", "/", nil), false)
|
|
|
|
assert.Equal(HTTPError(mockT, httpOK, "GET", "/", nil), false)
|
|
assert.Equal(HTTPError(mockT, httpRedirect, "GET", "/", nil), false)
|
|
assert.Equal(HTTPError(mockT, httpError, "GET", "/", nil), true)
|
|
}
|
|
|
|
func TestHTTPStatusesWrapper(t *testing.T) {
|
|
assert := New(t)
|
|
mockAssert := New(new(testing.T))
|
|
|
|
assert.Equal(mockAssert.HTTPSuccess(httpOK, "GET", "/", nil), true)
|
|
assert.Equal(mockAssert.HTTPSuccess(httpRedirect, "GET", "/", nil), false)
|
|
assert.Equal(mockAssert.HTTPSuccess(httpError, "GET", "/", nil), false)
|
|
|
|
assert.Equal(mockAssert.HTTPRedirect(httpOK, "GET", "/", nil), false)
|
|
assert.Equal(mockAssert.HTTPRedirect(httpRedirect, "GET", "/", nil), true)
|
|
assert.Equal(mockAssert.HTTPRedirect(httpError, "GET", "/", nil), false)
|
|
|
|
assert.Equal(mockAssert.HTTPError(httpOK, "GET", "/", nil), false)
|
|
assert.Equal(mockAssert.HTTPError(httpRedirect, "GET", "/", nil), false)
|
|
assert.Equal(mockAssert.HTTPError(httpError, "GET", "/", nil), true)
|
|
}
|
|
|
|
func httpHelloName(w http.ResponseWriter, r *http.Request) {
|
|
name := r.FormValue("name")
|
|
w.Write([]byte(fmt.Sprintf("Hello, %s!", name)))
|
|
}
|
|
|
|
func TestHttpBody(t *testing.T) {
|
|
assert := New(t)
|
|
mockT := new(testing.T)
|
|
|
|
assert.True(HTTPBodyContains(mockT, httpHelloName, "GET", "/", url.Values{"name": []string{"World"}}, "Hello, World!"))
|
|
assert.True(HTTPBodyContains(mockT, httpHelloName, "GET", "/", url.Values{"name": []string{"World"}}, "World"))
|
|
assert.False(HTTPBodyContains(mockT, httpHelloName, "GET", "/", url.Values{"name": []string{"World"}}, "world"))
|
|
|
|
assert.False(HTTPBodyNotContains(mockT, httpHelloName, "GET", "/", url.Values{"name": []string{"World"}}, "Hello, World!"))
|
|
assert.False(HTTPBodyNotContains(mockT, httpHelloName, "GET", "/", url.Values{"name": []string{"World"}}, "World"))
|
|
assert.True(HTTPBodyNotContains(mockT, httpHelloName, "GET", "/", url.Values{"name": []string{"World"}}, "world"))
|
|
}
|
|
|
|
func TestHttpBodyWrappers(t *testing.T) {
|
|
assert := New(t)
|
|
mockAssert := New(new(testing.T))
|
|
|
|
assert.True(mockAssert.HTTPBodyContains(httpHelloName, "GET", "/", url.Values{"name": []string{"World"}}, "Hello, World!"))
|
|
assert.True(mockAssert.HTTPBodyContains(httpHelloName, "GET", "/", url.Values{"name": []string{"World"}}, "World"))
|
|
assert.False(mockAssert.HTTPBodyContains(httpHelloName, "GET", "/", url.Values{"name": []string{"World"}}, "world"))
|
|
|
|
assert.False(mockAssert.HTTPBodyNotContains(httpHelloName, "GET", "/", url.Values{"name": []string{"World"}}, "Hello, World!"))
|
|
assert.False(mockAssert.HTTPBodyNotContains(httpHelloName, "GET", "/", url.Values{"name": []string{"World"}}, "World"))
|
|
assert.True(mockAssert.HTTPBodyNotContains(httpHelloName, "GET", "/", url.Values{"name": []string{"World"}}, "world"))
|
|
|
|
}
|