Context#Redirect

Signed-off-by: Vishal Rana <[email protected]>

context.go

@@ -285,13 +285,13 @@ func (c *context) NoContent(code int) error {
 	return nil
 }
 
-// Redirect redirects the request using http.Redirect with status code.
+// Redirect redirects the request with status code.
 func (c *context) Redirect(code int, url string) error {
 	if code < http.StatusMultipleChoices || code > http.StatusTemporaryRedirect {
 		return ErrInvalidRedirectCode
 	}
-	// TODO: v2
-	// http.Redirect(c.response, c.request, url, code)
+	c.response.Header().Set(Location, url)
+	c.response.WriteHeader(code)
 	return nil
 }
 

context_test.go

@@ -171,18 +171,20 @@ func TestContext(t *testing.T) {
 	rec = test.NewResponseRecorder()
 	c = NewContext(req, rec, e)
 	c.NoContent(http.StatusOK)
-	assert.Equal(t, http.StatusOK, c.Response().Status())
+	assert.Equal(t, http.StatusOK, rec.Status())
 
 	// Redirect
 	rec = test.NewResponseRecorder()
 	c = NewContext(req, rec, e)
 	assert.Equal(t, nil, c.Redirect(http.StatusMovedPermanently, "http://labstack.github.io/echo"))
+	assert.Equal(t, "http://labstack.github.io/echo", rec.Header().Get(Location))
+	assert.Equal(t, http.StatusMovedPermanently, rec.Status())
 
 	// Error
 	rec = test.NewResponseRecorder()
-	c = NewContext(req, rec, e).Object()
+	c = NewContext(req, rec, e)
 	c.Error(errors.New("error"))
-	assert.Equal(t, http.StatusInternalServerError, c.Response().Status())
+	assert.Equal(t, http.StatusInternalServerError, rec.Status())
 
 	// reset
 	c.Object().reset(req, test.NewResponseRecorder())