package guard

import (
	"errors"
	"net/http"
	"testing"
)

func TestGuard(t *testing.T) {
	mux := http.NewServeMux()
	mux.HandleFunc("/panic", func(rw http.ResponseWriter, r *http.Request) {
		panic(nil)
	})
	mux.HandleFunc("/custerr", func(rw http.ResponseWriter, r *http.Request) {
		panic(Error{
			Code:    404,
			Message: "custerr",
			ID:      "test",
		})
	})
	mux.HandleFunc("/err", func(rw http.ResponseWriter, r *http.Request) {
		panic(errors.New("err"))
	})
	mux.HandleFunc("/500", func(rw http.ResponseWriter, r *http.Request) {
		rw.WriteHeader(500)
		rw.Write([]byte("err"))
	})
	mux.HandleFunc("/", func(rw http.ResponseWriter, r *http.Request) {
		rw.Write([]byte("ok"))
	})
	http.ListenAndServe(":8000", New().Guard(mux))
}