From 53263e4ba24231811a3c907218318e799afb089d Mon Sep 17 00:00:00 2001
From: Dmitry Kutakov <vkd.castle@gmail.com>
Date: Thu, 17 Jan 2019 18:11:22 +0300
Subject: [PATCH] add tests when Context.reset() can be called inside of
 handler

TestEngineHandleContext
TestContextResetInHandler
TestRouterStaticFSFileNotFound
---
 context_test.go | 12 ++++++++++++
 gin_test.go     | 17 +++++++++++++++++
 routes_test.go  | 10 ++++++++++
 3 files changed, 39 insertions(+)

diff --git a/context_test.go b/context_test.go
index 836b3ecb..01053750 100644
--- a/context_test.go
+++ b/context_test.go
@@ -1737,3 +1737,15 @@ func TestContextStreamWithClientGone(t *testing.T) {
 
 	assert.Equal(t, "test", w.Body.String())
 }
+
+func TestContextResetInHandler(t *testing.T) {
+	w := CreateTestResponseRecorder()
+	c, _ := CreateTestContext(w)
+
+	c.handlers = []HandlerFunc{
+		func(c *Context) { c.reset() },
+	}
+	assert.NotPanics(t, func() {
+		c.Next()
+	})
+}
diff --git a/gin_test.go b/gin_test.go
index 353c9be1..e013df09 100644
--- a/gin_test.go
+++ b/gin_test.go
@@ -471,6 +471,23 @@ func TestListOfRoutes(t *testing.T) {
 	})
 }
 
+func TestEngineHandleContext(t *testing.T) {
+	r := New()
+	r.GET("/", func(c *Context) {
+		c.Request.URL.Path = "/v2"
+		r.HandleContext(c)
+	})
+	v2 := r.Group("/v2")
+	{
+		v2.GET("/", func(c *Context) {})
+	}
+
+	assert.NotPanics(t, func() {
+		w := performRequest(r, "GET", "/")
+		assert.Equal(t, 301, w.Code)
+	})
+}
+
 func assertRoutePresent(t *testing.T, gotRoutes RoutesInfo, wantRoute RouteInfo) {
 	for _, gotRoute := range gotRoutes {
 		if gotRoute.Path == wantRoute.Path && gotRoute.Method == wantRoute.Method {
diff --git a/routes_test.go b/routes_test.go
index c4d59725..10e8fe40 100644
--- a/routes_test.go
+++ b/routes_test.go
@@ -426,6 +426,16 @@ func TestRouterStaticFSNotFound(t *testing.T) {
 	assert.Equal(t, "non existent", w.Body.String())
 }
 
+func TestRouterStaticFSFileNotFound(t *testing.T) {
+	router := New()
+
+	router.StaticFS("/", http.FileSystem(http.Dir(".")))
+
+	assert.NotPanics(t, func() {
+		performRequest(router, "GET", "/nonexistent")
+	})
+}
+
 func TestRouteRawPath(t *testing.T) {
 	route := New()
 	route.UseRawPath = true