stat_handler_wrapper_test.go

//go:build go1.8
// +build go1.8

package stats

import (
	"net/http"
	"testing"
)

func TestHTTPHandler_WrapResponse(t *testing.T) {
	tests := []http.ResponseWriter{
		struct {
			http.ResponseWriter
			http.Flusher
			http.Hijacker
			http.Pusher
			http.CloseNotifier
		}{},
		struct {
			http.ResponseWriter
			http.Flusher
			http.Hijacker
			http.Pusher
		}{},
		struct {
			http.ResponseWriter
			http.Flusher
			http.Hijacker
			http.CloseNotifier
		}{},
		struct {
			http.ResponseWriter
			http.Flusher
			http.Hijacker
		}{},
		struct {
			http.ResponseWriter
			http.Flusher
			http.Pusher
			http.CloseNotifier
		}{},
		struct {
			http.ResponseWriter
			http.Flusher
			http.Pusher
		}{},
		struct {
			http.ResponseWriter
			http.Flusher
			http.CloseNotifier
		}{},
		struct {
			http.ResponseWriter
			http.Flusher
		}{},
		struct {
			http.ResponseWriter
			http.Hijacker
			http.Pusher
			http.CloseNotifier
		}{},
		struct {
			http.ResponseWriter
			http.Hijacker
			http.Pusher
		}{},
		struct {
			http.ResponseWriter
			http.Hijacker
			http.CloseNotifier
		}{},
		struct {
			http.ResponseWriter
			http.Hijacker
		}{},
		struct {
			http.ResponseWriter
			http.Pusher
			http.CloseNotifier
		}{},
		struct {
			http.ResponseWriter
			http.Pusher
		}{},
		struct {
			http.ResponseWriter
			http.CloseNotifier
		}{},
		struct{ http.ResponseWriter }{},
	}

	h := NewStatHandler(
		NewStore(NewNullSink(), false),
		http.HandlerFunc(func(http.ResponseWriter, *http.Request) {})).(*httpHandler)

	for i, test := range tests {
		tc := test
		_, canFlush := tc.(http.Flusher)
		_, canHijack := tc.(http.Hijacker)
		_, canPush := tc.(http.Pusher)

		rw := h.wrapResponse(tc)

		if _, ok := rw.(http.Flusher); ok != canFlush {
			t.Errorf("Test(%d): Flusher: wanted %t", i, canFlush)
		}
		if _, ok := rw.(http.Hijacker); ok != canHijack {
			t.Errorf("Test(%d): Hijacker: wanted %t", i, canHijack)
		}
		if _, ok := rw.(http.Pusher); ok != canPush {
			t.Errorf("Test(%d): Pusher: wanted %t", i, canPush)
		}
	}
}