mirror of
https://github.com/go-vikunja/vikunja.git
synced 2026-04-28 02:18:08 -05:00
9a61453e86
Closes https://github.com/go-vikunja/vikunja/pull/2133 Co-authored-by: renovate[bot] <29139614+renovate[bot]@users.noreply.github.com> Co-authored-by: kolaente <k@knt.li>
119 lines
3.5 KiB
Go
119 lines
3.5 KiB
Go
// Vikunja is a to-do list application to facilitate your life.
|
|
// Copyright 2018-present Vikunja and contributors. All rights reserved.
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
package webtests
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"mime/multipart"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"code.vikunja.io/api/pkg/config"
|
|
"code.vikunja.io/api/pkg/modules/auth"
|
|
"code.vikunja.io/api/pkg/routes"
|
|
|
|
"github.com/labstack/echo/v5"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestTaskAttachmentUploadSize(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
fileSize int64
|
|
expectedStatus int
|
|
configMaxSize string
|
|
}{
|
|
{
|
|
name: "Upload file within 32MB boundary",
|
|
fileSize: 30 * 1024 * 1024, // 30MB
|
|
expectedStatus: http.StatusOK,
|
|
configMaxSize: "50MB",
|
|
},
|
|
{
|
|
name: "Upload file above old 32MB limit",
|
|
fileSize: 35 * 1024 * 1024, // 35MB
|
|
expectedStatus: http.StatusOK,
|
|
configMaxSize: "50MB",
|
|
},
|
|
{
|
|
name: "Upload file exceeding configured limit",
|
|
fileSize: 55 * 1024 * 1024, // 55MB
|
|
expectedStatus: http.StatusRequestEntityTooLarge,
|
|
configMaxSize: "50MB",
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
// Setup test environment first (this calls InitDefaultConfig)
|
|
_, err := setupTestEnv()
|
|
require.NoError(t, err)
|
|
|
|
// Now set the config AFTER setupTestEnv
|
|
oldMaxSize := config.FilesMaxSize.GetString()
|
|
config.FilesMaxSize.Set(tt.configMaxSize)
|
|
defer config.FilesMaxSize.Set(oldMaxSize)
|
|
|
|
// Re-initialize config to update maxFileSizeInBytes
|
|
config.InitConfig()
|
|
|
|
// Create Echo instance with the updated config
|
|
e := routes.NewEcho()
|
|
routes.RegisterRoutes(e)
|
|
|
|
// Create multipart form data
|
|
body := &bytes.Buffer{}
|
|
writer := multipart.NewWriter(body)
|
|
part, err := writer.CreateFormFile("files", "test.pdf")
|
|
require.NoError(t, err)
|
|
|
|
// Write dummy data of specified size
|
|
_, err = io.CopyN(part, bytes.NewReader(make([]byte, tt.fileSize)), tt.fileSize)
|
|
require.NoError(t, err)
|
|
|
|
err = writer.Close()
|
|
require.NoError(t, err)
|
|
|
|
// Create request
|
|
req := httptest.NewRequest(http.MethodPut, "/api/v1/tasks/1/attachments", body)
|
|
req.Header.Set(echo.HeaderContentType, writer.FormDataContentType())
|
|
|
|
// Add JWT token to request header for authentication
|
|
token, err := auth.NewUserJWTAuthtoken(&testuser1, false)
|
|
require.NoError(t, err)
|
|
req.Header.Set("Authorization", "Bearer "+token)
|
|
|
|
rec := httptest.NewRecorder()
|
|
|
|
// Execute request
|
|
e.ServeHTTP(rec, req)
|
|
|
|
// Verify status code
|
|
assert.Equal(t, tt.expectedStatus, rec.Code)
|
|
|
|
// If we expect an error, verify the error response includes code and message
|
|
if tt.expectedStatus == http.StatusRequestEntityTooLarge {
|
|
assert.Contains(t, rec.Body.String(), "4013") // Error code
|
|
assert.Contains(t, rec.Body.String(), "uploaded file exceeds")
|
|
}
|
|
})
|
|
}
|
|
}
|