feat: auto add slash for url

ysmood · Apr 18, 2024 · 4562fdf · 4562fdf
1 parent 0798ede
commit 4562fdf
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/utils_serve.go b/utils_serve.go
@@ -46,7 +46,12 @@ type Router struct {
 
 // URL will prefix the path with the server's host
 func (rt *Router) URL(path ...string) string {
-	return rt.HostURL.String() + strings.Join(path, "")
+	p := strings.Join(path, "")
+	if !strings.HasPrefix(p, "/") {
+		p = "/" + p
+	}
+
+	return rt.HostURL.String() + p
 }
 
 // Route on the pattern. Check the doc of [http.ServeMux] for the syntax of pattern.

diff --git a/utils_test.go b/utils_test.go
@@ -88,7 +88,7 @@ func TestHelper(t *testing.T) {
 		ut.Eq(ut.Req("", s.URL()).String(), "")
 		ut.Has(ut.Req("", s.URL("/file")).String(), "ysmood/got")
 		ut.Eq(ut.Req("", s.URL("/a")).String(), "ok")
-		ut.Eq(ut.Req("", s.URL("/a")).String(), "ok")
+		ut.Eq(ut.Req("", s.URL("a")).String(), "ok")
 
 		ut.Has(ut.Req("", s.URL("/c")).String(), "ysmood/got")
 		ut.Req(http.MethodPost, s.URL("/d"), 1)