mirror of
https://github.com/gin-gonic/gin.git
synced 2026-07-15 07:42:19 +08:00
Merge branch 'master' into master
This commit is contained in:
commit
78c0df724b
10
context.go
10
context.go
@ -141,6 +141,16 @@ func (c *Context) Copy() *Context {
|
|||||||
cp.Params = make([]Param, len(cParams))
|
cp.Params = make([]Param, len(cParams))
|
||||||
copy(cp.Params, cParams)
|
copy(cp.Params, cParams)
|
||||||
|
|
||||||
|
if c.Errors != nil {
|
||||||
|
cp.Errors = make(errorMsgs, len(c.Errors))
|
||||||
|
copy(cp.Errors, c.Errors)
|
||||||
|
}
|
||||||
|
|
||||||
|
if c.Accepted != nil {
|
||||||
|
cp.Accepted = make([]string, len(c.Accepted))
|
||||||
|
copy(cp.Accepted, c.Accepted)
|
||||||
|
}
|
||||||
|
|
||||||
return &cp
|
return &cp
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@ -248,13 +248,11 @@ func TestSaveUploadedFileWithPermission(t *testing.T) {
|
|||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
assert.Equal(t, "permission_test", f.Filename)
|
assert.Equal(t, "permission_test", f.Filename)
|
||||||
var mode fs.FileMode = 0o755
|
var mode fs.FileMode = 0o755
|
||||||
require.NoError(t, c.SaveUploadedFile(f, "permission_test", mode))
|
dst := filepath.Join(t.TempDir(), "subdir", "permission_test")
|
||||||
t.Cleanup(func() {
|
require.NoError(t, c.SaveUploadedFile(f, dst, mode))
|
||||||
assert.NoError(t, os.Remove("permission_test"))
|
info, err := os.Stat(filepath.Dir(dst))
|
||||||
})
|
|
||||||
info, err := os.Stat(filepath.Dir("permission_test"))
|
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
assert.Equal(t, info.Mode().Perm(), mode)
|
assert.Equal(t, mode, info.Mode().Perm())
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestSaveUploadedFileWithPermissionFailed(t *testing.T) {
|
func TestSaveUploadedFileWithPermissionFailed(t *testing.T) {
|
||||||
@ -272,7 +270,8 @@ func TestSaveUploadedFileWithPermissionFailed(t *testing.T) {
|
|||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
assert.Equal(t, "permission_test", f.Filename)
|
assert.Equal(t, "permission_test", f.Filename)
|
||||||
var mode fs.FileMode = 0o644
|
var mode fs.FileMode = 0o644
|
||||||
require.Error(t, c.SaveUploadedFile(f, "test/permission_test", mode))
|
dst := filepath.Join(t.TempDir(), "test", "permission_test")
|
||||||
|
require.Error(t, c.SaveUploadedFile(f, dst, mode))
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestContextReset(t *testing.T) {
|
func TestContextReset(t *testing.T) {
|
||||||
@ -689,6 +688,50 @@ func TestContextCopy(t *testing.T) {
|
|||||||
assert.Equal(t, cp.fullPath, c.fullPath)
|
assert.Equal(t, cp.fullPath, c.fullPath)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestContextCopyCopiesErrors(t *testing.T) {
|
||||||
|
c, _ := CreateTestContext(httptest.NewRecorder())
|
||||||
|
c.Request, _ = http.NewRequest(http.MethodGet, "/", nil)
|
||||||
|
_ = c.Error(errors.New("first error"))
|
||||||
|
_ = c.Error(errors.New("second error"))
|
||||||
|
|
||||||
|
cp := c.Copy()
|
||||||
|
|
||||||
|
// copied context has the same errors
|
||||||
|
assert.Len(t, cp.Errors, 2)
|
||||||
|
assert.Equal(t, c.Errors[0].Error(), cp.Errors[0].Error())
|
||||||
|
assert.Equal(t, c.Errors[1].Error(), cp.Errors[1].Error())
|
||||||
|
|
||||||
|
// mutations on the copy do not affect the original
|
||||||
|
_ = cp.Error(errors.New("third error"))
|
||||||
|
assert.Len(t, c.Errors, 2)
|
||||||
|
assert.Len(t, cp.Errors, 3)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestContextCopyCopiesAccepted(t *testing.T) {
|
||||||
|
c, _ := CreateTestContext(httptest.NewRecorder())
|
||||||
|
c.Request, _ = http.NewRequest(http.MethodGet, "/", nil)
|
||||||
|
c.SetAccepted("application/json", "text/html")
|
||||||
|
|
||||||
|
cp := c.Copy()
|
||||||
|
|
||||||
|
assert.Equal(t, c.Accepted, cp.Accepted)
|
||||||
|
|
||||||
|
// mutations on the copy do not affect the original
|
||||||
|
cp.SetAccepted("text/plain")
|
||||||
|
assert.Equal(t, []string{"application/json", "text/html"}, c.Accepted)
|
||||||
|
assert.Equal(t, []string{"text/plain"}, cp.Accepted)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestContextCopyNilErrorsAndAccepted(t *testing.T) {
|
||||||
|
c, _ := CreateTestContext(httptest.NewRecorder())
|
||||||
|
c.Request, _ = http.NewRequest(http.MethodGet, "/", nil)
|
||||||
|
|
||||||
|
cp := c.Copy()
|
||||||
|
|
||||||
|
assert.Nil(t, cp.Errors)
|
||||||
|
assert.Nil(t, cp.Accepted)
|
||||||
|
}
|
||||||
|
|
||||||
func TestContextHandlerName(t *testing.T) {
|
func TestContextHandlerName(t *testing.T) {
|
||||||
c, _ := CreateTestContext(httptest.NewRecorder())
|
c, _ := CreateTestContext(httptest.NewRecorder())
|
||||||
c.handlers = HandlersChain{func(c *Context) {}, handlerNameTest}
|
c.handlers = HandlersChain{func(c *Context) {}, handlerNameTest}
|
||||||
|
|||||||
2
go.mod
2
go.mod
@ -12,7 +12,7 @@ require (
|
|||||||
github.com/mattn/go-isatty v0.0.20
|
github.com/mattn/go-isatty v0.0.20
|
||||||
github.com/modern-go/reflect2 v1.0.2
|
github.com/modern-go/reflect2 v1.0.2
|
||||||
github.com/pelletier/go-toml/v2 v2.2.4
|
github.com/pelletier/go-toml/v2 v2.2.4
|
||||||
github.com/quic-go/quic-go v0.59.0
|
github.com/quic-go/quic-go v0.60.0
|
||||||
github.com/stretchr/testify v1.11.1
|
github.com/stretchr/testify v1.11.1
|
||||||
github.com/ugorji/go/codec v1.3.1
|
github.com/ugorji/go/codec v1.3.1
|
||||||
go.mongodb.org/mongo-driver/v2 v2.5.0
|
go.mongodb.org/mongo-driver/v2 v2.5.0
|
||||||
|
|||||||
6
go.sum
6
go.sum
@ -50,10 +50,12 @@ github.com/pelletier/go-toml/v2 v2.2.4 h1:mye9XuhQ6gvn5h28+VilKrrPoQVanw5PMw/TB0
|
|||||||
github.com/pelletier/go-toml/v2 v2.2.4/go.mod h1:2gIqNv+qfxSVS7cM2xJQKtLSTLUE9V8t9Stt+h56mCY=
|
github.com/pelletier/go-toml/v2 v2.2.4/go.mod h1:2gIqNv+qfxSVS7cM2xJQKtLSTLUE9V8t9Stt+h56mCY=
|
||||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||||
|
github.com/quic-go/go-ossfuzz-seeds v0.1.0 h1:APacT+iIaNF6fd8AGEiN3bT/Jtkd2jz4v4TzM7MFjy0=
|
||||||
|
github.com/quic-go/go-ossfuzz-seeds v0.1.0/go.mod h1:3IOHRbJIc+L6YKMwfDtJAM9Vj9k0YY4muhuyUYk5tbk=
|
||||||
github.com/quic-go/qpack v0.6.0 h1:g7W+BMYynC1LbYLSqRt8PBg5Tgwxn214ZZR34VIOjz8=
|
github.com/quic-go/qpack v0.6.0 h1:g7W+BMYynC1LbYLSqRt8PBg5Tgwxn214ZZR34VIOjz8=
|
||||||
github.com/quic-go/qpack v0.6.0/go.mod h1:lUpLKChi8njB4ty2bFLX2x4gzDqXwUpaO1DP9qMDZII=
|
github.com/quic-go/qpack v0.6.0/go.mod h1:lUpLKChi8njB4ty2bFLX2x4gzDqXwUpaO1DP9qMDZII=
|
||||||
github.com/quic-go/quic-go v0.59.0 h1:OLJkp1Mlm/aS7dpKgTc6cnpynnD2Xg7C1pwL6vy/SAw=
|
github.com/quic-go/quic-go v0.60.0 h1:xcQioE8OM66UQLeUMHltK1CCcOu3JbVB4JAQdDQSB+0=
|
||||||
github.com/quic-go/quic-go v0.59.0/go.mod h1:upnsH4Ju1YkqpLXC305eW3yDZ4NfnNbmQRCMWS58IKU=
|
github.com/quic-go/quic-go v0.60.0/go.mod h1:wpKpjmPpftl30sL6pFh7REVpjbcCVy4zt2vDyK1TuJk=
|
||||||
github.com/rogpeppe/go-internal v1.10.0 h1:TMyTOH3F/DB16zRVcYyreMH6GnZZrwQVAoYjRBZyWFQ=
|
github.com/rogpeppe/go-internal v1.10.0 h1:TMyTOH3F/DB16zRVcYyreMH6GnZZrwQVAoYjRBZyWFQ=
|
||||||
github.com/rogpeppe/go-internal v1.10.0/go.mod h1:UQnix2H7Ngw/k4C5ijL5+65zddjncjaFoBhdsK/akog=
|
github.com/rogpeppe/go-internal v1.10.0/go.mod h1:UQnix2H7Ngw/k4C5ijL5+65zddjncjaFoBhdsK/akog=
|
||||||
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||||
|
|||||||
@ -106,7 +106,12 @@ func secureRequestDump(r *http.Request) string {
|
|||||||
return strings.Join(lines, "\r\n")
|
return strings.Join(lines, "\r\n")
|
||||||
}
|
}
|
||||||
|
|
||||||
func defaultHandleRecovery(c *Context, _ any) {
|
func defaultHandleRecovery(c *Context, err any) {
|
||||||
|
e, ok := err.(error)
|
||||||
|
if !ok {
|
||||||
|
e = fmt.Errorf("%v", err)
|
||||||
|
}
|
||||||
|
c.Error(e) //nolint: errcheck
|
||||||
c.AbortWithStatus(http.StatusInternalServerError)
|
c.AbortWithStatus(http.StatusInternalServerError)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@ -5,6 +5,7 @@
|
|||||||
package gin
|
package gin
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"errors"
|
||||||
"net"
|
"net"
|
||||||
"net/http"
|
"net/http"
|
||||||
"os"
|
"os"
|
||||||
@ -152,6 +153,49 @@ func TestPanicWithAbortHandler(t *testing.T) {
|
|||||||
assert.NotContains(t, out, "panic recovered")
|
assert.NotContains(t, out, "panic recovered")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestPanicInHandlerRecordsError(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
recoveredErr any
|
||||||
|
expectedErr string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "string panic",
|
||||||
|
recoveredErr: "Oops, Houston, we have a problem",
|
||||||
|
expectedErr: "Oops, Houston, we have a problem",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "error panic",
|
||||||
|
recoveredErr: errors.New("recovered error"),
|
||||||
|
expectedErr: "recovered error",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
router := New()
|
||||||
|
|
||||||
|
var recoveredErrors errorMsgs
|
||||||
|
router.Use(func(c *Context) {
|
||||||
|
c.Next()
|
||||||
|
recoveredErrors = c.Errors
|
||||||
|
})
|
||||||
|
router.Use(RecoveryWithWriter(nil))
|
||||||
|
router.GET("/recovery", func(_ *Context) {
|
||||||
|
panic(tt.recoveredErr)
|
||||||
|
})
|
||||||
|
|
||||||
|
w := PerformRequest(router, http.MethodGet, "/recovery")
|
||||||
|
|
||||||
|
assert.Equal(t, http.StatusInternalServerError, w.Code)
|
||||||
|
if assert.Len(t, recoveredErrors, 1) {
|
||||||
|
assert.EqualError(t, recoveredErrors[0], tt.expectedErr)
|
||||||
|
assert.Equal(t, ErrorTypePrivate, recoveredErrors[0].Type)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestCustomRecoveryWithWriter(t *testing.T) {
|
func TestCustomRecoveryWithWriter(t *testing.T) {
|
||||||
errBuffer := new(strings.Builder)
|
errBuffer := new(strings.Builder)
|
||||||
buffer := new(strings.Builder)
|
buffer := new(strings.Builder)
|
||||||
|
|||||||
@ -15,10 +15,33 @@ import (
|
|||||||
"github.com/stretchr/testify/require"
|
"github.com/stretchr/testify/require"
|
||||||
)
|
)
|
||||||
|
|
||||||
// TODO
|
// TestResponseWriterFlushWithFlusher verifies Flush() calls the underlying Flusher.
|
||||||
// func (w *responseWriter) Hijack() (net.Conn, *bufio.ReadWriter, error) {
|
func TestResponseWriterFlushWithFlusher(t *testing.T) {
|
||||||
// func (w *responseWriter) CloseNotify() <-chan bool {
|
testWriter := httptest.NewRecorder()
|
||||||
// func (w *responseWriter) Flush() {
|
writer := &responseWriter{ResponseWriter: testWriter}
|
||||||
|
writer.Flush()
|
||||||
|
assert.True(t, testWriter.Flushed)
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestResponseWriterFlushWithNonFlusher verifies Flush() is a no-op
|
||||||
|
// when the underlying ResponseWriter does not implement http.Flusher.
|
||||||
|
// Guards against the panic reported in https://github.com/gin-gonic/gin/issues/4460
|
||||||
|
func TestResponseWriterFlushWithNonFlusher(t *testing.T) {
|
||||||
|
nonFlusher := &nonFlusherWriter{header: http.Header{}}
|
||||||
|
writer := &responseWriter{ResponseWriter: nonFlusher}
|
||||||
|
require.NotPanics(t, func() {
|
||||||
|
writer.Flush()
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// nonFlusherWriter is a minimal http.ResponseWriter that does NOT implement http.Flusher.
|
||||||
|
type nonFlusherWriter struct {
|
||||||
|
header http.Header
|
||||||
|
}
|
||||||
|
|
||||||
|
func (w *nonFlusherWriter) Header() http.Header { return w.header }
|
||||||
|
func (w *nonFlusherWriter) Write(b []byte) (int, error) { return len(b), nil }
|
||||||
|
func (w *nonFlusherWriter) WriteHeader(code int) {}
|
||||||
|
|
||||||
var (
|
var (
|
||||||
_ ResponseWriter = &responseWriter{}
|
_ ResponseWriter = &responseWriter{}
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user