246 lines
6.0 KiB
Go
246 lines
6.0 KiB
Go
|
package http_response
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
"io/ioutil"
|
||
|
|
"net/http"
|
||
|
|
"net/http/httptest"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestParseHeaders(t *testing.T) {
|
||
|
|
fakeHeaders := `
|
||
|
|
Accept: text/plain
|
||
|
|
Content-Type: application/json
|
||
|
|
Cache-Control: no-cache
|
||
|
|
`
|
||
|
|
headers, err := ParseHeaders(fakeHeaders)
|
||
|
|
require.NoError(t, err)
|
||
|
|
testHeaders := make(http.Header)
|
||
|
|
testHeaders.Add("Accept", "text/plain")
|
||
|
|
testHeaders.Add("Content-Type", "application/json")
|
||
|
|
testHeaders.Add("Cache-Control", "no-cache")
|
||
|
|
assert.Equal(t, testHeaders, headers)
|
||
|
|
|
||
|
|
headers, err = ParseHeaders("Accept text/plain")
|
||
|
|
require.Error(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func setUpTestMux() http.Handler {
|
||
|
|
mux := http.NewServeMux()
|
||
|
|
mux.HandleFunc("/redirect", func(w http.ResponseWriter, req *http.Request) {
|
||
|
|
http.Redirect(w, req, "/good", http.StatusMovedPermanently)
|
||
|
|
})
|
||
|
|
mux.HandleFunc("/good", func(w http.ResponseWriter, req *http.Request) {
|
||
|
|
fmt.Fprintf(w, "hit the good page!")
|
||
|
|
})
|
||
|
|
mux.HandleFunc("/badredirect", func(w http.ResponseWriter, req *http.Request) {
|
||
|
|
http.Redirect(w, req, "/badredirect", http.StatusMovedPermanently)
|
||
|
|
})
|
||
|
|
mux.HandleFunc("/mustbepostmethod", func(w http.ResponseWriter, req *http.Request) {
|
||
|
|
if req.Method != "POST" {
|
||
|
|
http.Error(w, "method wasn't post", http.StatusMethodNotAllowed)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
fmt.Fprintf(w, "used post correctly!")
|
||
|
|
})
|
||
|
|
mux.HandleFunc("/musthaveabody", func(w http.ResponseWriter, req *http.Request) {
|
||
|
|
body, err := ioutil.ReadAll(req.Body)
|
||
|
|
req.Body.Close()
|
||
|
|
if err != nil {
|
||
|
|
http.Error(w, "couldn't read request body", http.StatusBadRequest)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
if string(body) == "" {
|
||
|
|
http.Error(w, "body was empty", http.StatusBadRequest)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
fmt.Fprintf(w, "sent a body!")
|
||
|
|
})
|
||
|
|
mux.HandleFunc("/twosecondnap", func(w http.ResponseWriter, req *http.Request) {
|
||
|
|
time.Sleep(time.Second * 2)
|
||
|
|
return
|
||
|
|
})
|
||
|
|
return mux
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestFields(t *testing.T) {
|
||
|
|
mux := setUpTestMux()
|
||
|
|
ts := httptest.NewServer(mux)
|
||
|
|
defer ts.Close()
|
||
|
|
|
||
|
|
h := &HTTPResponse{
|
||
|
|
Address: ts.URL + "/good",
|
||
|
|
Body: "{ 'test': 'data'}",
|
||
|
|
Method: "GET",
|
||
|
|
ResponseTimeout: 20,
|
||
|
|
Headers: `
|
||
|
|
Content-Type: application/json
|
||
|
|
`,
|
||
|
|
FollowRedirects: true,
|
||
|
|
}
|
||
|
|
fields, err := h.HTTPGather()
|
||
|
|
require.NoError(t, err)
|
||
|
|
assert.NotEmpty(t, fields)
|
||
|
|
if assert.NotNil(t, fields["http_response_code"]) {
|
||
|
|
assert.Equal(t, http.StatusOK, fields["http_response_code"])
|
||
|
|
}
|
||
|
|
assert.NotNil(t, fields["response_time"])
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestRedirects(t *testing.T) {
|
||
|
|
mux := setUpTestMux()
|
||
|
|
ts := httptest.NewServer(mux)
|
||
|
|
defer ts.Close()
|
||
|
|
|
||
|
|
h := &HTTPResponse{
|
||
|
|
Address: ts.URL + "/redirect",
|
||
|
|
Body: "{ 'test': 'data'}",
|
||
|
|
Method: "GET",
|
||
|
|
ResponseTimeout: 20,
|
||
|
|
Headers: `
|
||
|
|
Content-Type: application/json
|
||
|
|
`,
|
||
|
|
FollowRedirects: true,
|
||
|
|
}
|
||
|
|
fields, err := h.HTTPGather()
|
||
|
|
require.NoError(t, err)
|
||
|
|
assert.NotEmpty(t, fields)
|
||
|
|
if assert.NotNil(t, fields["http_response_code"]) {
|
||
|
|
assert.Equal(t, http.StatusOK, fields["http_response_code"])
|
||
|
|
}
|
||
|
|
|
||
|
|
h = &HTTPResponse{
|
||
|
|
Address: ts.URL + "/badredirect",
|
||
|
|
Body: "{ 'test': 'data'}",
|
||
|
|
Method: "GET",
|
||
|
|
ResponseTimeout: 20,
|
||
|
|
Headers: `
|
||
|
|
Content-Type: application/json
|
||
|
|
`,
|
||
|
|
FollowRedirects: true,
|
||
|
|
}
|
||
|
|
fields, err = h.HTTPGather()
|
||
|
|
require.Error(t, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestMethod(t *testing.T) {
|
||
|
|
mux := setUpTestMux()
|
||
|
|
ts := httptest.NewServer(mux)
|
||
|
|
defer ts.Close()
|
||
|
|
|
||
|
|
h := &HTTPResponse{
|
||
|
|
Address: ts.URL + "/mustbepostmethod",
|
||
|
|
Body: "{ 'test': 'data'}",
|
||
|
|
Method: "POST",
|
||
|
|
ResponseTimeout: 20,
|
||
|
|
Headers: `
|
||
|
|
Content-Type: application/json
|
||
|
|
`,
|
||
|
|
FollowRedirects: true,
|
||
|
|
}
|
||
|
|
fields, err := h.HTTPGather()
|
||
|
|
require.NoError(t, err)
|
||
|
|
assert.NotEmpty(t, fields)
|
||
|
|
if assert.NotNil(t, fields["http_response_code"]) {
|
||
|
|
assert.Equal(t, http.StatusOK, fields["http_response_code"])
|
||
|
|
}
|
||
|
|
|
||
|
|
h = &HTTPResponse{
|
||
|
|
Address: ts.URL + "/mustbepostmethod",
|
||
|
|
Body: "{ 'test': 'data'}",
|
||
|
|
Method: "GET",
|
||
|
|
ResponseTimeout: 20,
|
||
|
|
Headers: `
|
||
|
|
Content-Type: application/json
|
||
|
|
`,
|
||
|
|
FollowRedirects: true,
|
||
|
|
}
|
||
|
|
fields, err = h.HTTPGather()
|
||
|
|
require.NoError(t, err)
|
||
|
|
assert.NotEmpty(t, fields)
|
||
|
|
if assert.NotNil(t, fields["http_response_code"]) {
|
||
|
|
assert.Equal(t, http.StatusMethodNotAllowed, fields["http_response_code"])
|
||
|
|
}
|
||
|
|
|
||
|
|
//check that lowercase methods work correctly
|
||
|
|
h = &HTTPResponse{
|
||
|
|
Address: ts.URL + "/mustbepostmethod",
|
||
|
|
Body: "{ 'test': 'data'}",
|
||
|
|
Method: "head",
|
||
|
|
ResponseTimeout: 20,
|
||
|
|
Headers: `
|
||
|
|
Content-Type: application/json
|
||
|
|
`,
|
||
|
|
FollowRedirects: true,
|
||
|
|
}
|
||
|
|
fields, err = h.HTTPGather()
|
||
|
|
require.NoError(t, err)
|
||
|
|
assert.NotEmpty(t, fields)
|
||
|
|
if assert.NotNil(t, fields["http_response_code"]) {
|
||
|
|
assert.Equal(t, http.StatusMethodNotAllowed, fields["http_response_code"])
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestBody(t *testing.T) {
|
||
|
|
mux := setUpTestMux()
|
||
|
|
ts := httptest.NewServer(mux)
|
||
|
|
defer ts.Close()
|
||
|
|
|
||
|
|
h := &HTTPResponse{
|
||
|
|
Address: ts.URL + "/musthaveabody",
|
||
|
|
Body: "{ 'test': 'data'}",
|
||
|
|
Method: "GET",
|
||
|
|
ResponseTimeout: 20,
|
||
|
|
Headers: `
|
||
|
|
Content-Type: application/json
|
||
|
|
`,
|
||
|
|
FollowRedirects: true,
|
||
|
|
}
|
||
|
|
fields, err := h.HTTPGather()
|
||
|
|
require.NoError(t, err)
|
||
|
|
assert.NotEmpty(t, fields)
|
||
|
|
if assert.NotNil(t, fields["http_response_code"]) {
|
||
|
|
assert.Equal(t, http.StatusOK, fields["http_response_code"])
|
||
|
|
}
|
||
|
|
|
||
|
|
h = &HTTPResponse{
|
||
|
|
Address: ts.URL + "/musthaveabody",
|
||
|
|
Method: "GET",
|
||
|
|
ResponseTimeout: 20,
|
||
|
|
Headers: `
|
||
|
|
Content-Type: application/json
|
||
|
|
`,
|
||
|
|
FollowRedirects: true,
|
||
|
|
}
|
||
|
|
fields, err = h.HTTPGather()
|
||
|
|
require.NoError(t, err)
|
||
|
|
assert.NotEmpty(t, fields)
|
||
|
|
if assert.NotNil(t, fields["http_response_code"]) {
|
||
|
|
assert.Equal(t, http.StatusBadRequest, fields["http_response_code"])
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestTimeout(t *testing.T) {
|
||
|
|
mux := setUpTestMux()
|
||
|
|
ts := httptest.NewServer(mux)
|
||
|
|
defer ts.Close()
|
||
|
|
|
||
|
|
h := &HTTPResponse{
|
||
|
|
Address: ts.URL + "/twosecondnap",
|
||
|
|
Body: "{ 'test': 'data'}",
|
||
|
|
Method: "GET",
|
||
|
|
ResponseTimeout: 1,
|
||
|
|
Headers: `
|
||
|
|
Content-Type: application/json
|
||
|
|
`,
|
||
|
|
FollowRedirects: true,
|
||
|
|
}
|
||
|
|
_, err := h.HTTPGather()
|
||
|
|
require.Error(t, err)
|
||
|
|
}
|