blackteay/gin
3
0
Fork 0
mirror of https://github.com/gin-gonic/gin.git synced 2026-06-06 12:08:20 +08:00
Code Issues Projects Releases Wiki Activity

test: cover uploaded file directory permissions

Browse Source
This commit is contained in:
Herrtian 2026-05-22 13:02:47 +02:00
parent 6cfac86ae7
commit a3e91a0053
1 changed files with 45 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
context_test.go
View File

@ -259,6 +259,51 @@ func TestSaveUploadedFileWithPermission(t *testing.T) {
} }
} }
func TestDirExists(t *testing.T) {
tmpDir := t.TempDir()
assert.True(t, dirExists(tmpDir))
assert.False(t, dirExists(filepath.Join(tmpDir, "missing")))
filePath := filepath.Join(tmpDir, "file")
require.NoError(t, os.WriteFile(filePath, []byte("test"), 0o600))
assert.False(t, dirExists(filePath))
}
func TestSaveUploadedFileKeepsExistingDirPermission(t *testing.T) {
buf := new(bytes.Buffer)
mw := multipart.NewWriter(buf)
w, err := mw.CreateFormFile("file", "permission_test")
require.NoError(t, err)
_, err = w.Write([]byte("permission_test"))
require.NoError(t, err)
mw.Close()
c, _ := CreateTestContext(httptest.NewRecorder())
c.Request, _ = http.NewRequest(http.MethodPost, "/", buf)
c.Request.Header.Set("Content-Type", mw.FormDataContentType())
f, err := c.FormFile("file")
require.NoError(t, err)
tmpDir := t.TempDir()
existingDir := filepath.Join(tmpDir, "existing")
require.NoError(t, os.Mkdir(existingDir, 0o700))
if runtime.GOOS != "windows" {
require.NoError(t, os.Chmod(existingDir, 0o700))
}
var mode fs.FileMode = 0o755
dst := filepath.Join(existingDir, "permission_test")
require.NoError(t, c.SaveUploadedFile(f, dst, mode))
saved, err := os.ReadFile(dst)
require.NoError(t, err)
assert.Equal(t, "permission_test", string(saved))
if runtime.GOOS != "windows" {
info, err := os.Stat(existingDir)
require.NoError(t, err)
assert.Equal(t, fs.FileMode(0o700), info.Mode().Perm())
}
}
func TestSaveUploadedFileWithPermissionFailed(t *testing.T) { func TestSaveUploadedFileWithPermissionFailed(t *testing.T) {
buf := new(bytes.Buffer) buf := new(bytes.Buffer)
mw := multipart.NewWriter(buf) mw := multipart.NewWriter(buf)