127 lines
2.8 KiB
Go
127 lines
2.8 KiB
Go
package httpr
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func Test(t *testing.T) {
|
|
r := New()
|
|
|
|
err := r.Handler(http.MethodGet, "/", func(w http.ResponseWriter, r *http.Request) {})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = r.Handler(http.MethodGet, "/", func(w http.ResponseWriter, r *http.Request) {})
|
|
if err == nil {
|
|
t.Fatal("path redefinition wasn't catched")
|
|
}
|
|
|
|
err = r.Handler(http.MethodGet, "/a/b", func(w http.ResponseWriter, r *http.Request) {})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = r.Handler(http.MethodGet, "/:a/:b", func(w http.ResponseWriter, r *http.Request) {})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = r.Handler(http.MethodGet, "/:a/:lol", func(w http.ResponseWriter, r *http.Request) {})
|
|
if err == nil {
|
|
t.Fatal("here is a different last param name is supplied, should be catched")
|
|
}
|
|
|
|
err = r.ServeStatic("/assets/*filepath", http.FS(os.DirFS(".")))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = r.ServeStatic("/assets/*filepath/*filepath", nil)
|
|
if err == nil {
|
|
t.Fatal("multiple catch-all params wasn't catched")
|
|
}
|
|
|
|
}
|
|
|
|
func TestPaths(t *testing.T) {
|
|
found := false
|
|
|
|
r := New()
|
|
|
|
err := r.Handler(http.MethodGet, "/:lel", func(w http.ResponseWriter, r *http.Request) { found = true })
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
r.NotFoundHandler = func(w http.ResponseWriter, r *http.Request) { found = false }
|
|
|
|
w := httptest.NewRecorder()
|
|
p := "/xmpp://me@arav.su"
|
|
req := httptest.NewRequest(http.MethodGet, p, strings.NewReader(""))
|
|
r.ServeHTTP(w, req)
|
|
if found {
|
|
t.Error("Path", p, "should return 404")
|
|
}
|
|
|
|
p = "/lel"
|
|
req = httptest.NewRequest(http.MethodGet, p, strings.NewReader(""))
|
|
r.ServeHTTP(w, req)
|
|
if !found {
|
|
t.Error("Path", p, "should return 200")
|
|
}
|
|
|
|
p = "/lel/lol"
|
|
req = httptest.NewRequest(http.MethodGet, p, strings.NewReader(""))
|
|
r.ServeHTTP(w, req)
|
|
if found {
|
|
t.Error("Path", p, "should return 404")
|
|
}
|
|
}
|
|
|
|
func TestSubPaths(t *testing.T) {
|
|
found := true
|
|
|
|
r := New()
|
|
|
|
s := r.Sub("/api/v1")
|
|
|
|
err := s.Handler(http.MethodGet, "/", func(w http.ResponseWriter, r *http.Request) { found = true })
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
err = s.Handler(http.MethodGet, "/test", func(w http.ResponseWriter, r *http.Request) { found = true })
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
r.NotFoundHandler = func(w http.ResponseWriter, r *http.Request) { found = false }
|
|
|
|
w := httptest.NewRecorder()
|
|
p := "/api/v1/"
|
|
req := httptest.NewRequest(http.MethodGet, p, strings.NewReader(""))
|
|
r.ServeHTTP(w, req)
|
|
if !found {
|
|
t.Error("Path", p, "should return 200")
|
|
}
|
|
|
|
p = "/api/v1/test"
|
|
req = httptest.NewRequest(http.MethodGet, p, strings.NewReader(""))
|
|
r.ServeHTTP(w, req)
|
|
if !found {
|
|
t.Error("Path", p, "should return 200")
|
|
}
|
|
|
|
p = "/api/v1/nonexistent"
|
|
req = httptest.NewRequest(http.MethodGet, p, strings.NewReader(""))
|
|
r.ServeHTTP(w, req)
|
|
if found {
|
|
t.Error(found, "Path", p, "should return 404")
|
|
}
|
|
}
|