middleware_test.go

package main

import (
	"bytes"
	"log"
	"net/http"
	"net/http/httptest"
	"testing"
)

func emptyTestHandler() service {
	return func(h http.Handler) http.Handler {
		return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
			w.WriteHeader(http.StatusAccepted)
		})
	}
}

func TestServiceLoader(t *testing.T) {
	req, err := http.NewRequest("GET", "/api/v1/stats", nil)
	if err != nil {
		t.Error(err)
	}
	rr := httptest.NewRecorder()
	testHandlers := serviceLoader(cacheIndexHandler(), emptyTestHandler())
	testHandlers.ServeHTTP(rr, req)
	if status := rr.Code; status != http.StatusAccepted {
		t.Errorf("handlers not loading properly. want: 202, got: %d", rr.Code)
	}
}

func TestRequestMetrics(t *testing.T) {
	var b bytes.Buffer
	logger := log.New(&b, "", log.LstdFlags)
	req, err := http.NewRequest("GET", "/api/v1/cache/empty", nil)
	if err != nil {
		t.Error(err)
	}
	rr := httptest.NewRecorder()
	testHandlers := serviceLoader(cacheIndexHandler(), requestMetrics(logger))
	testHandlers.ServeHTTP(rr, req)
	targetTestString := b.String()
	if len(targetTestString) == 0 {
		t.Errorf("we are not logging request length strings.")
	}
	t.Log(targetTestString)
}