handler_test.go

package main

import (
	"io/ioutil"
	"net/http"
	"net/http/httptest"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestEchoHandler(t *testing.T) {
	s := httptest.NewServer(http.HandlerFunc(EchoHandler("hello")))

	res, err := http.Get(s.URL)
	assert.NoError(t, err)
	assert.Equal(t, "text/plain", res.Header.Get("Content-Type"))
	assert.Equal(t, 200, res.StatusCode)

	defer res.Body.Close()
	body, err := ioutil.ReadAll(res.Body)
	assert.NoError(t, err)
	assert.Equal(t, "hello", string(body))
}