add more unit tests

gookit · Sep 10, 2020 · 9d1cafa · 9d1cafa
1 parent 83177b1
commit 9d1cafa
Showing 1 changed file with 17 additions and 2 deletions.
diff --git a/router_test.go b/router_test.go
@@ -358,13 +358,28 @@ func TestRouter_Controller(t *testing.T) {
 	is.Equal(200, w.Code)
 	is.Equal("hello, in /", w.Body.String())
 
-	w = testutil.MockRequest(r, http.MethodGet, "/", nil)
+	w = testutil.MockRequest(r, http.MethodGet, "", nil)
 	is.Equal(200, w.Code)
-	is.Equal("hello, in /", w.Body.String())
+	is.Equal("hello, in ", w.Body.String())
 
 	Debug(false)
 }
 
+func TestRouter_Controller2(t *testing.T) {
+	is := assert.New(t)
+	r := New()
+
+	r.Controller("", &SiteController{})
+
+	w := testutil.MockRequest(r, http.MethodGet, "/", nil)
+	is.Equal(200, w.Code)
+	is.Equal("hello, in /", w.Body.String())
+
+	w = testutil.MockRequest(r, http.MethodGet, "", nil)
+	is.Equal(200, w.Code)
+	is.Equal("hello, in ", w.Body.String())
+}
+
 func TestDynamicRoute(t *testing.T) {
 	is := assert.New(t)