telegraf/plugins/inputs/http_response/http_response_test.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)
}