|
@@ -10,224 +10,139 @@ import (
|
|
|
"testing"
|
|
"testing"
|
|
|
)
|
|
)
|
|
|
|
|
|
|
|
-// TestRouterGroupGETRouteOK tests that GET route is correctly invoked.
|
|
|
|
|
-func TestRouterGroupGETRouteOK(t *testing.T) {
|
|
|
|
|
- req, _ := http.NewRequest("GET", "/test", nil)
|
|
|
|
|
|
|
+func PerformRequest(r http.Handler, method, path string) *httptest.ResponseRecorder {
|
|
|
|
|
+ req, _ := http.NewRequest(method, path, nil)
|
|
|
w := httptest.NewRecorder()
|
|
w := httptest.NewRecorder()
|
|
|
- passed := false
|
|
|
|
|
-
|
|
|
|
|
- r := Default()
|
|
|
|
|
- r.GET("/test", func(c *Context) {
|
|
|
|
|
- passed = true
|
|
|
|
|
- })
|
|
|
|
|
-
|
|
|
|
|
- r.ServeHTTP(w, req)
|
|
|
|
|
-
|
|
|
|
|
- if passed == false {
|
|
|
|
|
- t.Errorf("GET route handler was not invoked.")
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- if w.Code != http.StatusOK {
|
|
|
|
|
- t.Errorf("Status code should be %v, was %d", http.StatusOK, w.Code)
|
|
|
|
|
- }
|
|
|
|
|
-}
|
|
|
|
|
-
|
|
|
|
|
-// TestRouterGroupGETNoRootExistsRouteOK tests that a GET requse to root is correctly
|
|
|
|
|
-// handled (404) when no root route exists.
|
|
|
|
|
-func TestRouterGroupGETNoRootExistsRouteOK(t *testing.T) {
|
|
|
|
|
- req, _ := http.NewRequest("GET", "/", nil)
|
|
|
|
|
- w := httptest.NewRecorder()
|
|
|
|
|
-
|
|
|
|
|
- r := Default()
|
|
|
|
|
- r.GET("/test", func(c *Context) {
|
|
|
|
|
- })
|
|
|
|
|
-
|
|
|
|
|
- r.ServeHTTP(w, req)
|
|
|
|
|
-
|
|
|
|
|
- if w.Code != http.StatusNotFound {
|
|
|
|
|
- // If this fails, it's because httprouter needs to be updated to at least f78f58a0db
|
|
|
|
|
- t.Errorf("Status code should be %v, was %d. Location: %s", http.StatusNotFound, w.Code, w.HeaderMap.Get("Location"))
|
|
|
|
|
- }
|
|
|
|
|
-}
|
|
|
|
|
-
|
|
|
|
|
-// TestRouterGroupPOSTRouteOK tests that POST route is correctly invoked.
|
|
|
|
|
-func TestRouterGroupPOSTRouteOK(t *testing.T) {
|
|
|
|
|
- req, _ := http.NewRequest("POST", "/test", nil)
|
|
|
|
|
- w := httptest.NewRecorder()
|
|
|
|
|
- passed := false
|
|
|
|
|
-
|
|
|
|
|
- r := Default()
|
|
|
|
|
- r.POST("/test", func(c *Context) {
|
|
|
|
|
- passed = true
|
|
|
|
|
- })
|
|
|
|
|
-
|
|
|
|
|
r.ServeHTTP(w, req)
|
|
r.ServeHTTP(w, req)
|
|
|
-
|
|
|
|
|
- if passed == false {
|
|
|
|
|
- t.Errorf("POST route handler was not invoked.")
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- if w.Code != http.StatusOK {
|
|
|
|
|
- t.Errorf("Status code should be %v, was %d", http.StatusOK, w.Code)
|
|
|
|
|
- }
|
|
|
|
|
|
|
+ return w
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
-// TestRouterGroupDELETERouteOK tests that DELETE route is correctly invoked.
|
|
|
|
|
-func TestRouterGroupDELETERouteOK(t *testing.T) {
|
|
|
|
|
- req, _ := http.NewRequest("DELETE", "/test", nil)
|
|
|
|
|
- w := httptest.NewRecorder()
|
|
|
|
|
|
|
+// TestSingleRouteOK tests that POST route is correctly invoked.
|
|
|
|
|
+func testRouteOK(method string, t *testing.T) {
|
|
|
|
|
+ // SETUP
|
|
|
passed := false
|
|
passed := false
|
|
|
-
|
|
|
|
|
- r := Default()
|
|
|
|
|
- r.DELETE("/test", func(c *Context) {
|
|
|
|
|
|
|
+ r := New()
|
|
|
|
|
+ r.Handle(method, "/test", []HandlerFunc{func(c *Context) {
|
|
|
passed = true
|
|
passed = true
|
|
|
- })
|
|
|
|
|
|
|
+ }})
|
|
|
|
|
|
|
|
- r.ServeHTTP(w, req)
|
|
|
|
|
|
|
+ // RUN
|
|
|
|
|
+ w := PerformRequest(r, method, "/test")
|
|
|
|
|
|
|
|
|
|
+ // TEST
|
|
|
if passed == false {
|
|
if passed == false {
|
|
|
- t.Errorf("DELETE route handler was not invoked.")
|
|
|
|
|
|
|
+ t.Errorf(method + " route handler was not invoked.")
|
|
|
}
|
|
}
|
|
|
-
|
|
|
|
|
if w.Code != http.StatusOK {
|
|
if w.Code != http.StatusOK {
|
|
|
t.Errorf("Status code should be %v, was %d", http.StatusOK, w.Code)
|
|
t.Errorf("Status code should be %v, was %d", http.StatusOK, w.Code)
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
|
-
|
|
|
|
|
-// TestRouterGroupPATCHRouteOK tests that PATCH route is correctly invoked.
|
|
|
|
|
-func TestRouterGroupPATCHRouteOK(t *testing.T) {
|
|
|
|
|
- req, _ := http.NewRequest("PATCH", "/test", nil)
|
|
|
|
|
- w := httptest.NewRecorder()
|
|
|
|
|
- passed := false
|
|
|
|
|
-
|
|
|
|
|
- r := Default()
|
|
|
|
|
- r.PATCH("/test", func(c *Context) {
|
|
|
|
|
- passed = true
|
|
|
|
|
- })
|
|
|
|
|
-
|
|
|
|
|
- r.ServeHTTP(w, req)
|
|
|
|
|
-
|
|
|
|
|
- if passed == false {
|
|
|
|
|
- t.Errorf("PATCH route handler was not invoked.")
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- if w.Code != http.StatusOK {
|
|
|
|
|
- t.Errorf("Status code should be %v, was %d", http.StatusOK, w.Code)
|
|
|
|
|
- }
|
|
|
|
|
|
|
+func TestRouterGroupRouteOK(t *testing.T) {
|
|
|
|
|
+ testRouteOK("POST", t)
|
|
|
|
|
+ testRouteOK("DELETE", t)
|
|
|
|
|
+ testRouteOK("PATCH", t)
|
|
|
|
|
+ testRouteOK("PUT", t)
|
|
|
|
|
+ testRouteOK("OPTIONS", t)
|
|
|
|
|
+ testRouteOK("HEAD", t)
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
-// TestRouterGroupPUTRouteOK tests that PUT route is correctly invoked.
|
|
|
|
|
-func TestRouterGroupPUTRouteOK(t *testing.T) {
|
|
|
|
|
- req, _ := http.NewRequest("PUT", "/test", nil)
|
|
|
|
|
- w := httptest.NewRecorder()
|
|
|
|
|
|
|
+// TestSingleRouteOK tests that POST route is correctly invoked.
|
|
|
|
|
+func testRouteNotOK(method string, t *testing.T) {
|
|
|
|
|
+ // SETUP
|
|
|
passed := false
|
|
passed := false
|
|
|
-
|
|
|
|
|
- r := Default()
|
|
|
|
|
- r.PUT("/test", func(c *Context) {
|
|
|
|
|
|
|
+ r := New()
|
|
|
|
|
+ r.Handle(method, "/test_2", []HandlerFunc{func(c *Context) {
|
|
|
passed = true
|
|
passed = true
|
|
|
- })
|
|
|
|
|
|
|
+ }})
|
|
|
|
|
|
|
|
- r.ServeHTTP(w, req)
|
|
|
|
|
|
|
+ // RUN
|
|
|
|
|
+ w := PerformRequest(r, method, "/test")
|
|
|
|
|
|
|
|
- if passed == false {
|
|
|
|
|
- t.Errorf("PUT route handler was not invoked.")
|
|
|
|
|
|
|
+ // TEST
|
|
|
|
|
+ if passed == true {
|
|
|
|
|
+ t.Errorf(method + " route handler was invoked, when it should not")
|
|
|
}
|
|
}
|
|
|
-
|
|
|
|
|
- if w.Code != http.StatusOK {
|
|
|
|
|
- t.Errorf("Status code should be %v, was %d", http.StatusOK, w.Code)
|
|
|
|
|
|
|
+ if w.Code != http.StatusNotFound {
|
|
|
|
|
+ // If this fails, it's because httprouter needs to be updated to at least f78f58a0db
|
|
|
|
|
+ t.Errorf("Status code should be %v, was %d. Location: %s", http.StatusNotFound, w.Code, w.HeaderMap.Get("Location"))
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
-// TestRouterGroupOPTIONSRouteOK tests that OPTIONS route is correctly invoked.
|
|
|
|
|
-func TestRouterGroupOPTIONSRouteOK(t *testing.T) {
|
|
|
|
|
- req, _ := http.NewRequest("OPTIONS", "/test", nil)
|
|
|
|
|
- w := httptest.NewRecorder()
|
|
|
|
|
- passed := false
|
|
|
|
|
-
|
|
|
|
|
- r := Default()
|
|
|
|
|
- r.OPTIONS("/test", func(c *Context) {
|
|
|
|
|
- passed = true
|
|
|
|
|
- })
|
|
|
|
|
-
|
|
|
|
|
- r.ServeHTTP(w, req)
|
|
|
|
|
-
|
|
|
|
|
- if passed == false {
|
|
|
|
|
- t.Errorf("OPTIONS route handler was not invoked.")
|
|
|
|
|
- }
|
|
|
|
|
-
|
|
|
|
|
- if w.Code != http.StatusOK {
|
|
|
|
|
- t.Errorf("Status code should be %v, was %d", http.StatusOK, w.Code)
|
|
|
|
|
- }
|
|
|
|
|
|
|
+// TestSingleRouteOK tests that POST route is correctly invoked.
|
|
|
|
|
+func TestRouteNotOK(t *testing.T) {
|
|
|
|
|
+ testRouteNotOK("POST", t)
|
|
|
|
|
+ testRouteNotOK("DELETE", t)
|
|
|
|
|
+ testRouteNotOK("PATCH", t)
|
|
|
|
|
+ testRouteNotOK("PUT", t)
|
|
|
|
|
+ testRouteNotOK("OPTIONS", t)
|
|
|
|
|
+ testRouteNotOK("HEAD", t)
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
-// TestRouterGroupHEADRouteOK tests that HEAD route is correctly invoked.
|
|
|
|
|
-func TestRouterGroupHEADRouteOK(t *testing.T) {
|
|
|
|
|
- req, _ := http.NewRequest("HEAD", "/test", nil)
|
|
|
|
|
- w := httptest.NewRecorder()
|
|
|
|
|
|
|
+// TestSingleRouteOK tests that POST route is correctly invoked.
|
|
|
|
|
+func testRouteNotOK2(method string, t *testing.T) {
|
|
|
|
|
+ // SETUP
|
|
|
passed := false
|
|
passed := false
|
|
|
-
|
|
|
|
|
- r := Default()
|
|
|
|
|
- r.HEAD("/test", func(c *Context) {
|
|
|
|
|
|
|
+ r := New()
|
|
|
|
|
+ var methodRoute string
|
|
|
|
|
+ if method == "POST" {
|
|
|
|
|
+ methodRoute = "GET"
|
|
|
|
|
+ } else {
|
|
|
|
|
+ methodRoute = "POST"
|
|
|
|
|
+ }
|
|
|
|
|
+ r.Handle(methodRoute, "/test", []HandlerFunc{func(c *Context) {
|
|
|
passed = true
|
|
passed = true
|
|
|
- })
|
|
|
|
|
|
|
+ }})
|
|
|
|
|
|
|
|
- r.ServeHTTP(w, req)
|
|
|
|
|
|
|
+ // RUN
|
|
|
|
|
+ w := PerformRequest(r, method, "/test")
|
|
|
|
|
|
|
|
- if passed == false {
|
|
|
|
|
- t.Errorf("HEAD route handler was not invoked.")
|
|
|
|
|
|
|
+ // TEST
|
|
|
|
|
+ if passed == true {
|
|
|
|
|
+ t.Errorf(method + " route handler was invoked, when it should not")
|
|
|
}
|
|
}
|
|
|
-
|
|
|
|
|
- if w.Code != http.StatusOK {
|
|
|
|
|
- t.Errorf("Status code should be %v, was %d", http.StatusOK, w.Code)
|
|
|
|
|
|
|
+ if w.Code != http.StatusNotFound {
|
|
|
|
|
+ // If this fails, it's because httprouter needs to be updated to at least f78f58a0db
|
|
|
|
|
+ t.Errorf("Status code should be %v, was %d. Location: %s", http.StatusNotFound, w.Code, w.HeaderMap.Get("Location"))
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
-// TestRouterGroup404 tests that 404 is returned for a route that does not exist.
|
|
|
|
|
-func TestEngine404(t *testing.T) {
|
|
|
|
|
- req, _ := http.NewRequest("GET", "/", nil)
|
|
|
|
|
- w := httptest.NewRecorder()
|
|
|
|
|
-
|
|
|
|
|
- r := Default()
|
|
|
|
|
- r.ServeHTTP(w, req)
|
|
|
|
|
-
|
|
|
|
|
- if w.Code != http.StatusNotFound {
|
|
|
|
|
- t.Errorf("Response code should be %v, was %d", http.StatusNotFound, w.Code)
|
|
|
|
|
- }
|
|
|
|
|
|
|
+// TestSingleRouteOK tests that POST route is correctly invoked.
|
|
|
|
|
+func TestRouteNotOK2(t *testing.T) {
|
|
|
|
|
+ testRouteNotOK2("POST", t)
|
|
|
|
|
+ testRouteNotOK2("DELETE", t)
|
|
|
|
|
+ testRouteNotOK2("PATCH", t)
|
|
|
|
|
+ testRouteNotOK2("PUT", t)
|
|
|
|
|
+ testRouteNotOK2("OPTIONS", t)
|
|
|
|
|
+ testRouteNotOK2("HEAD", t)
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
// TestHandleStaticFile - ensure the static file handles properly
|
|
// TestHandleStaticFile - ensure the static file handles properly
|
|
|
func TestHandleStaticFile(t *testing.T) {
|
|
func TestHandleStaticFile(t *testing.T) {
|
|
|
-
|
|
|
|
|
|
|
+ // SETUP file
|
|
|
testRoot, _ := os.Getwd()
|
|
testRoot, _ := os.Getwd()
|
|
|
-
|
|
|
|
|
f, err := ioutil.TempFile(testRoot, "")
|
|
f, err := ioutil.TempFile(testRoot, "")
|
|
|
- defer os.Remove(f.Name())
|
|
|
|
|
-
|
|
|
|
|
if err != nil {
|
|
if err != nil {
|
|
|
t.Error(err)
|
|
t.Error(err)
|
|
|
}
|
|
}
|
|
|
-
|
|
|
|
|
|
|
+ defer os.Remove(f.Name())
|
|
|
filePath := path.Join("/", path.Base(f.Name()))
|
|
filePath := path.Join("/", path.Base(f.Name()))
|
|
|
- req, _ := http.NewRequest("GET", filePath, nil)
|
|
|
|
|
-
|
|
|
|
|
f.WriteString("Gin Web Framework")
|
|
f.WriteString("Gin Web Framework")
|
|
|
f.Close()
|
|
f.Close()
|
|
|
|
|
|
|
|
- w := httptest.NewRecorder()
|
|
|
|
|
-
|
|
|
|
|
- r := Default()
|
|
|
|
|
|
|
+ // SETUP gin
|
|
|
|
|
+ r := New()
|
|
|
r.Static("./", testRoot)
|
|
r.Static("./", testRoot)
|
|
|
|
|
|
|
|
- r.ServeHTTP(w, req)
|
|
|
|
|
|
|
+ // RUN
|
|
|
|
|
+ w := PerformRequest(r, "GET", filePath)
|
|
|
|
|
|
|
|
|
|
+ // TEST
|
|
|
if w.Code != 200 {
|
|
if w.Code != 200 {
|
|
|
t.Errorf("Response code should be Ok, was: %s", w.Code)
|
|
t.Errorf("Response code should be Ok, was: %s", w.Code)
|
|
|
}
|
|
}
|
|
|
-
|
|
|
|
|
if w.Body.String() != "Gin Web Framework" {
|
|
if w.Body.String() != "Gin Web Framework" {
|
|
|
t.Errorf("Response should be test, was: %s", w.Body.String())
|
|
t.Errorf("Response should be test, was: %s", w.Body.String())
|
|
|
}
|
|
}
|
|
|
-
|
|
|
|
|
if w.HeaderMap.Get("Content-Type") != "text/plain; charset=utf-8" {
|
|
if w.HeaderMap.Get("Content-Type") != "text/plain; charset=utf-8" {
|
|
|
t.Errorf("Content-Type should be text/plain, was %s", w.HeaderMap.Get("Content-Type"))
|
|
t.Errorf("Content-Type should be text/plain, was %s", w.HeaderMap.Get("Content-Type"))
|
|
|
}
|
|
}
|
|
@@ -235,30 +150,24 @@ func TestHandleStaticFile(t *testing.T) {
|
|
|
|
|
|
|
|
// TestHandleStaticDir - ensure the root/sub dir handles properly
|
|
// TestHandleStaticDir - ensure the root/sub dir handles properly
|
|
|
func TestHandleStaticDir(t *testing.T) {
|
|
func TestHandleStaticDir(t *testing.T) {
|
|
|
-
|
|
|
|
|
- req, _ := http.NewRequest("GET", "/", nil)
|
|
|
|
|
-
|
|
|
|
|
- w := httptest.NewRecorder()
|
|
|
|
|
-
|
|
|
|
|
- r := Default()
|
|
|
|
|
|
|
+ // SETUP
|
|
|
|
|
+ r := New()
|
|
|
r.Static("/", "./")
|
|
r.Static("/", "./")
|
|
|
|
|
|
|
|
- r.ServeHTTP(w, req)
|
|
|
|
|
|
|
+ // RUN
|
|
|
|
|
+ w := PerformRequest(r, "GET", "/")
|
|
|
|
|
|
|
|
|
|
+ // TEST
|
|
|
|
|
+ bodyAsString := w.Body.String()
|
|
|
if w.Code != 200 {
|
|
if w.Code != 200 {
|
|
|
t.Errorf("Response code should be Ok, was: %s", w.Code)
|
|
t.Errorf("Response code should be Ok, was: %s", w.Code)
|
|
|
}
|
|
}
|
|
|
-
|
|
|
|
|
- bodyAsString := w.Body.String()
|
|
|
|
|
-
|
|
|
|
|
if len(bodyAsString) == 0 {
|
|
if len(bodyAsString) == 0 {
|
|
|
t.Errorf("Got empty body instead of file tree")
|
|
t.Errorf("Got empty body instead of file tree")
|
|
|
}
|
|
}
|
|
|
-
|
|
|
|
|
if !strings.Contains(bodyAsString, "gin.go") {
|
|
if !strings.Contains(bodyAsString, "gin.go") {
|
|
|
t.Errorf("Can't find:`gin.go` in file tree: %s", bodyAsString)
|
|
t.Errorf("Can't find:`gin.go` in file tree: %s", bodyAsString)
|
|
|
}
|
|
}
|
|
|
-
|
|
|
|
|
if w.HeaderMap.Get("Content-Type") != "text/html; charset=utf-8" {
|
|
if w.HeaderMap.Get("Content-Type") != "text/html; charset=utf-8" {
|
|
|
t.Errorf("Content-Type should be text/plain, was %s", w.HeaderMap.Get("Content-Type"))
|
|
t.Errorf("Content-Type should be text/plain, was %s", w.HeaderMap.Get("Content-Type"))
|
|
|
}
|
|
}
|
|
@@ -266,29 +175,24 @@ func TestHandleStaticDir(t *testing.T) {
|
|
|
|
|
|
|
|
// TestHandleHeadToDir - ensure the root/sub dir handles properly
|
|
// TestHandleHeadToDir - ensure the root/sub dir handles properly
|
|
|
func TestHandleHeadToDir(t *testing.T) {
|
|
func TestHandleHeadToDir(t *testing.T) {
|
|
|
-
|
|
|
|
|
- req, _ := http.NewRequest("HEAD", "/", nil)
|
|
|
|
|
-
|
|
|
|
|
- w := httptest.NewRecorder()
|
|
|
|
|
-
|
|
|
|
|
- r := Default()
|
|
|
|
|
|
|
+ // SETUP
|
|
|
|
|
+ r := New()
|
|
|
r.Static("/", "./")
|
|
r.Static("/", "./")
|
|
|
|
|
|
|
|
- r.ServeHTTP(w, req)
|
|
|
|
|
|
|
+ // RUN
|
|
|
|
|
+ w := PerformRequest(r, "HEAD", "/")
|
|
|
|
|
|
|
|
|
|
+ // TEST
|
|
|
|
|
+ bodyAsString := w.Body.String()
|
|
|
if w.Code != 200 {
|
|
if w.Code != 200 {
|
|
|
t.Errorf("Response code should be Ok, was: %s", w.Code)
|
|
t.Errorf("Response code should be Ok, was: %s", w.Code)
|
|
|
}
|
|
}
|
|
|
-
|
|
|
|
|
- bodyAsString := w.Body.String()
|
|
|
|
|
-
|
|
|
|
|
if len(bodyAsString) == 0 {
|
|
if len(bodyAsString) == 0 {
|
|
|
t.Errorf("Got empty body instead of file tree")
|
|
t.Errorf("Got empty body instead of file tree")
|
|
|
}
|
|
}
|
|
|
if !strings.Contains(bodyAsString, "gin.go") {
|
|
if !strings.Contains(bodyAsString, "gin.go") {
|
|
|
t.Errorf("Can't find:`gin.go` in file tree: %s", bodyAsString)
|
|
t.Errorf("Can't find:`gin.go` in file tree: %s", bodyAsString)
|
|
|
}
|
|
}
|
|
|
-
|
|
|
|
|
if w.HeaderMap.Get("Content-Type") != "text/html; charset=utf-8" {
|
|
if w.HeaderMap.Get("Content-Type") != "text/html; charset=utf-8" {
|
|
|
t.Errorf("Content-Type should be text/plain, was %s", w.HeaderMap.Get("Content-Type"))
|
|
t.Errorf("Content-Type should be text/plain, was %s", w.HeaderMap.Get("Content-Type"))
|
|
|
}
|
|
}
|