github.com/hhsnopek/up@v0.1.1/handler/handler_test.go

package handler

import (
	"net/http"
	"net/http/httptest"
	"os"
	"testing"
	"time"

	"github.com/tj/assert"
)

// TODO: tests integration of each middleware
// TODO: tests for POST etc
// TODO: tests for larger response bodies

func TestNode(t *testing.T) {
	os.Chdir("testdata/node")
	defer os.Chdir("../..")

	h, err := New()
	assert.NoError(t, err)

	res := httptest.NewRecorder()
	req := httptest.NewRequest("GET", "/", nil)

	h.ServeHTTP(res, req)

	actual := res.Header()
	assert.NotEmpty(t, actual.Get("Date"), "date")
	actual.Del("Date")

	expected := make(http.Header)
	expected.Add("X-Powered-By", "up")
	expected.Add("X-Foo", "bar")
	expected.Add("Content-Length", "11")
	expected.Add("Content-Type", "text/plain; charset=utf-8")
	assert.Equal(t, expected, actual)
}

func TestStatic(t *testing.T) {
	os.Chdir("testdata/static")
	defer os.Chdir("../..")

	h, err := New()
	assert.NoError(t, err)

	res := httptest.NewRecorder()
	req := httptest.NewRequest("GET", "/", nil)

	start := time.Now()
	h.ServeHTTP(res, req)
	t.Logf("latency = %s", time.Since(start))

	actual := res.Header()
	assert.NotEmpty(t, actual.Get("Last-Modified"), "last-modified")
	actual.Del("Last-Modified")

	expected := make(http.Header)
	expected.Add("X-Powered-By", "up")
	expected.Add("Content-Length", "12")
	expected.Add("Content-Type", "text/html; charset=utf-8")
	expected.Add("Accept-Ranges", "bytes")

	assert.Equal(t, expected, actual)
}

func TestHandler_conditionalGet(t *testing.T) {
	os.Chdir("testdata/static")
	defer os.Chdir("../..")

	h, err := New()
	assert.NoError(t, err)

	res := httptest.NewRecorder()
	req := httptest.NewRequest("GET", "/style.css", nil)
	req.Header.Set("If-Modified-Since", "Thu, 27 Jul 2030 03:30:31 GMT")
	h.ServeHTTP(res, req)
	assert.Equal(t, 304, res.Code)
	assert.Equal(t, "", res.Header().Get("Content-Length"))
	assert.Equal(t, "", res.Body.String())
}

func TestHandler_rewrite(t *testing.T) {
	os.Chdir("testdata/static-rewrites")
	defer os.Chdir("../..")

	h, err := New()
	assert.NoError(t, err)

	res := httptest.NewRecorder()
	req := httptest.NewRequest("GET", "/docs/ping/guides/alerts", nil)
	h.ServeHTTP(res, req)
	assert.Equal(t, 200, res.Code)
	assert.Equal(t, "14", res.Header().Get("Content-Length"))
	assert.Equal(t, "Alerting docs\n", res.Body.String())
}

func TestHandler_redirect(t *testing.T) {
	os.Chdir("testdata/static-redirects")
	defer os.Chdir("../..")

	h, err := New()
	assert.NoError(t, err)

	res := httptest.NewRecorder()
	req := httptest.NewRequest("GET", "/docs/ping/guides/alerts/", nil)
	h.ServeHTTP(res, req)
	assert.Equal(t, "/help/ping/alerts", res.Header().Get("Location"))
	assert.Equal(t, 302, res.Code)
	assert.Equal(t, "Found\n", res.Body.String())
}

func TestHandler_spa(t *testing.T) {
	os.Chdir("testdata/spa")
	defer os.Chdir("../..")

	h, err := New()
	assert.NoError(t, err)

	t.Run("index", func(t *testing.T) {
		res := httptest.NewRecorder()
		req := httptest.NewRequest("GET", "/", nil)
		h.ServeHTTP(res, req)
		assert.Equal(t, 200, res.Code)
		assert.Equal(t, "Index\n", res.Body.String())
	})

	t.Run("redirect", func(t *testing.T) {
		res := httptest.NewRecorder()
		req := httptest.NewRequest("GET", "/index.html", nil)
		h.ServeHTTP(res, req)
		assert.Equal(t, 301, res.Code)
	})

	t.Run("file does not exist", func(t *testing.T) {
		res := httptest.NewRecorder()
		req := httptest.NewRequest("GET", "/something/here", nil)
		h.ServeHTTP(res, req)
		assert.Equal(t, 200, res.Code)
		assert.Equal(t, "Index\n", res.Body.String())
	})

	t.Run("file exists", func(t *testing.T) {
		res := httptest.NewRecorder()
		req := httptest.NewRequest("GET", "/app.js", nil)
		h.ServeHTTP(res, req)
		assert.Equal(t, 200, res.Code)
		assert.Equal(t, "app js\n", res.Body.String())
	})

	t.Run("file exists nested", func(t *testing.T) {
		res := httptest.NewRecorder()
		req := httptest.NewRequest("GET", "/css/bar.css", nil)
		h.ServeHTTP(res, req)
		assert.Equal(t, 200, res.Code)
		assert.Equal(t, "bar css\n", res.Body.String())
	})
}

func BenchmarkHandler(b *testing.B) {
	b.ReportAllocs()

	b.Run("static server", func(b *testing.B) {
		os.Chdir("testdata/basic")

		h, err := New()
		assert.NoError(b, err)

		b.ResetTimer()
		b.SetParallelism(80)

		b.RunParallel(func(pb *testing.PB) {
			for pb.Next() {
				res := httptest.NewRecorder()
				req := httptest.NewRequest("GET", "/", nil)
				h.ServeHTTP(res, req)
			}
		})
	})

	b.Run("node server relay", func(b *testing.B) {
		os.Chdir("testdata/basic")

		h, err := New()
		assert.NoError(b, err)

		b.ResetTimer()
		b.SetParallelism(80)

		b.RunParallel(func(pb *testing.PB) {
			for pb.Next() {
				res := httptest.NewRecorder()
				req := httptest.NewRequest("GET", "/", nil)
				h.ServeHTTP(res, req)
			}
		})
	})
}