response_add_header_test.go

// nolint dupl
package functional_tests

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

	"github.com/maxcnunes/httpfake"
)

// TestResponseAddHeader tests a fake server handling a GET request
// and responding with a speficied header
func TestResponseAddHeader(t *testing.T) {
	fakeService := httpfake.New()
	defer fakeService.Server.Close()

	// register a handler for our fake service
	fakeService.NewHandler().
		Get("/users").
		Reply(200).
		AddHeader("X-My-Header", "My Header value").
		BodyString("[]")

	req, err := http.NewRequest("GET", fakeService.ResolveURL("/users"), nil)
	if err != nil {
		t.Fatal(err)
	}

	res, err := http.DefaultClient.Do(req)
	if err != nil {
		t.Fatal(err)
	}
	defer res.Body.Close() // nolint errcheck

	// Check the status code is what we expect
	if status := res.StatusCode; status != 200 {
		t.Errorf("request returned wrong status code: got %v want %v",
			status, 200)
	}

	// Check the response body is what we expect
	expected := "[]"
	body, _ := ioutil.ReadAll(res.Body)
	if bodyString := string(body); bodyString != expected {
		t.Errorf("request returned unexpected body: got %v want %v",
			bodyString, expected)
	}

	// Check the response header is what we expect
	expected = "My Header value"
	if header := res.Header.Get("X-My-Header"); header != expected {
		t.Errorf("request returned unexpected value for header X-My-Header: got %v want %v",
			header, expected)
	}
}