abouthandler_test.go

package j8a

import (
	"bytes"
	"io/ioutil"
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"
)

// this testHandler binds the mock HTTP server to proxyHandler.
type AboutHttpHandler struct{}

func (t AboutHttpHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
	aboutHandler(w, r)
}

func TestAboutHandlerAcceptEncodingIdentitySendsIdentity(t *testing.T) {
	Runner = mockRuntime()

	server := httptest.NewServer(&AboutHttpHandler{})
	defer server.Close()

	c := &http.Client{}
	req, _ := http.NewRequest("GET", server.URL, nil)
	req.Header.Set(acceptEncoding, "identity")
	resp, err := c.Do(req)
	if err != nil {
		t.Fatal(err)
	}

	gotBody, _ := ioutil.ReadAll(resp.Body)
	if c := bytes.Compare(gotBody[0:2], gzipMagicBytes); c == 0 {
		t.Errorf("body should not have gzip response magic bytes but does: %v", gotBody[0:2])
	}

	want := "identity"
	got := resp.Header[contentEncoding][0]
	if got != want {
		t.Errorf("response does have correct Content-Encoding header, want %v, got %v", want, got)
	}
}

func TestAboutHandlerAcceptEncodingNotSpecifiedSendsIdentity(t *testing.T) {
	Runner = mockRuntime()

	server := httptest.NewServer(&AboutHttpHandler{})
	defer server.Close()

	c := &http.Client{
		Transport: &http.Transport{
			DisableCompression: true,
		},
	}
	req, _ := http.NewRequest("GET", server.URL, nil)
	req.Header.Del(acceptEncoding)
	resp, err := c.Do(req)
	if err != nil {
		t.Fatal(err)
	}

	gotBody, _ := ioutil.ReadAll(resp.Body)
	if c := bytes.Compare(gotBody[0:2], gzipMagicBytes); c == 0 {
		t.Errorf("body should not have gzip response magic bytes but does: %v", gotBody[0:2])
	}

	want := "identity"
	got := resp.Header[contentEncoding][0]
	if got != want {
		t.Errorf("response does have correct Content-Encoding header, want %v, got %v", want, got)
	}
}

func TestAboutHandlerAcceptEncodingGzipSendsGzip(t *testing.T) {
	Runner = mockRuntime()

	server := httptest.NewServer(&AboutHttpHandler{})
	defer server.Close()

	c := &http.Client{}
	req, _ := http.NewRequest("GET", server.URL, nil)
	req.Header.Set(acceptEncoding, "gzip")
	resp, err := c.Do(req)
	if err != nil {
		t.Fatal(err)
	}

	gotBody, _ := ioutil.ReadAll(resp.Body)
	if c := bytes.Compare(gotBody[0:2], gzipMagicBytes); c != 0 {
		t.Errorf("body should have gzip response magic bytes but does not: %v", gotBody[0:2])
	}

	want := "gzip"
	got := resp.Header[contentEncoding][0]
	if got != want {
		t.Errorf("response does have correct Content-Encoding header, want %v, got %v", want, got)
	}
}

func TestAboutHandlerAcceptEncodingBrotliSendsBrotli(t *testing.T) {
	Runner = mockRuntime()

	server := httptest.NewServer(&AboutHttpHandler{})
	defer server.Close()

	c := &http.Client{}
	req, _ := http.NewRequest("GET", server.URL, nil)
	req.Header.Set(acceptEncoding, "br")
	resp, err := c.Do(req)
	if err != nil {
		t.Fatal(err)
	}

	want := "br"
	got := resp.Header[contentEncoding][0]
	if got != want {
		t.Errorf("response does have correct Content-Encoding header, want %v, got %v", want, got)
	}
}

func TestAboutHandlerAcceptEncodingDeflateSends406AsIdentity(t *testing.T) {
	Runner = mockRuntime()

	server := httptest.NewServer(&AboutHttpHandler{})
	defer server.Close()

	c := &http.Client{}
	req, _ := http.NewRequest("GET", server.URL, nil)
	req.Header.Set(acceptEncoding, "deflate")
	resp, err := c.Do(req)
	if err != nil {
		t.Fatal(err)
	}

	gotBody, _ := ioutil.ReadAll(resp.Body)
	stringBody := string(gotBody)
	if !strings.Contains(stringBody, "406") {
		t.Errorf("response should contain 406 for deflate request")
	}

	want := "identity"
	got := resp.Header[contentEncoding][0]
	if got != want {
		t.Errorf("response does have correct Content-Encoding header, want %v, got %v", want, got)
	}
}

func TestStatusCodeResponse_FromCode(t *testing.T) {
	res301 := new(StatusCodeResponse)
	res301.withCode(301)
	want := "moved permanently"
	got := res301.Message
	if got != want {
		t.Errorf("invalid status Code response, want %v, got %v", want, got)
	}
}

func TestStatusCodeResponse_AsString(t *testing.T) {
	res := StatusCodeResponse{
		AboutResponse: AboutResponse{
			Version: "1",
		},
		Code:    0,
		Message: "msg",
	}

	str := res.AsString()

	if !strings.Contains(str, "j8a") {
		t.Errorf("about response j8a not included")
	}
	if !strings.Contains(str, "0") {
		t.Errorf("about response Code not included")
	}
	if !strings.Contains(str, "msg") {
		t.Errorf("about response Message not included")
	}
}