1ea1852677
- Deleted the obsolete prompt.md file, streamlining the codebase. - Added tests for image upload and download functionality using local and S3 storage drivers, ensuring robust handling of image files. - Implemented image compression during upload for both local and S3 storage, maintaining aspect ratio and limiting dimensions to a maximum size of 1920x1080. - Enhanced the local and S3 storage drivers to support image compression, improving storage efficiency. - Updated tests to verify image dimensions and content type after upload and download operations, ensuring data integrity.
150 lines
4 KiB
Go
150 lines
4 KiB
Go
package local
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"image"
|
|
"image/png"
|
|
"io"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/yaoapp/yao/config"
|
|
"github.com/yaoapp/yao/test"
|
|
)
|
|
|
|
func TestLocalStorage(t *testing.T) {
|
|
test.Prepare(t, config.Conf)
|
|
defer test.Clean()
|
|
|
|
t.Run("Create Storage", func(t *testing.T) {
|
|
storage, err := New(map[string]interface{}{
|
|
"path": "/__vision_test",
|
|
"compression": true,
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, storage)
|
|
assert.Equal(t, "/__vision_test", storage.Path)
|
|
assert.True(t, storage.Compression)
|
|
})
|
|
|
|
t.Run("Upload and Download", func(t *testing.T) {
|
|
storage, err := New(map[string]interface{}{
|
|
"path": "/__vision_test",
|
|
"compression": true,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
content := []byte("test content")
|
|
reader := bytes.NewReader(content)
|
|
fileID, err := storage.Upload(context.Background(), "test.txt", reader, "text/plain")
|
|
assert.NoError(t, err)
|
|
assert.NotEmpty(t, fileID)
|
|
|
|
// Download
|
|
reader2, contentType, err := storage.Download(context.Background(), fileID)
|
|
assert.NoError(t, err)
|
|
assert.Contains(t, contentType, "text/plain")
|
|
|
|
downloaded, err := io.ReadAll(reader2)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, content, downloaded)
|
|
})
|
|
|
|
t.Run("Upload and Download Image with Compression", func(t *testing.T) {
|
|
storage, err := New(map[string]interface{}{
|
|
"path": "/__vision_test",
|
|
"compression": true,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
// Create a test image (2000x2000 pixels)
|
|
img := image.NewRGBA(image.Rect(0, 0, 2000, 2000))
|
|
var buf bytes.Buffer
|
|
err = png.Encode(&buf, img)
|
|
assert.NoError(t, err)
|
|
|
|
// Upload
|
|
reader := bytes.NewReader(buf.Bytes())
|
|
fileID, err := storage.Upload(context.Background(), "test.png", reader, "image/png")
|
|
assert.NoError(t, err)
|
|
assert.NotEmpty(t, fileID)
|
|
|
|
// Download and verify size
|
|
reader2, contentType, err := storage.Download(context.Background(), fileID)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "image/png", contentType)
|
|
|
|
downloaded, err := io.ReadAll(reader2)
|
|
assert.NoError(t, err)
|
|
|
|
// Decode the downloaded image
|
|
downloadedImg, _, err := image.Decode(bytes.NewReader(downloaded))
|
|
assert.NoError(t, err)
|
|
|
|
// Verify dimensions
|
|
bounds := downloadedImg.Bounds()
|
|
assert.LessOrEqual(t, bounds.Dx(), MaxImageSize)
|
|
assert.LessOrEqual(t, bounds.Dy(), MaxImageSize)
|
|
})
|
|
|
|
t.Run("Upload Image without Compression", func(t *testing.T) {
|
|
storage, err := New(map[string]interface{}{
|
|
"path": "/__vision_test",
|
|
"compression": false,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
// Create a test image (2000x2000 pixels)
|
|
img := image.NewRGBA(image.Rect(0, 0, 2000, 2000))
|
|
var buf bytes.Buffer
|
|
err = png.Encode(&buf, img)
|
|
assert.NoError(t, err)
|
|
|
|
// Upload
|
|
reader := bytes.NewReader(buf.Bytes())
|
|
fileID, err := storage.Upload(context.Background(), "test.png", reader, "image/png")
|
|
assert.NoError(t, err)
|
|
assert.NotEmpty(t, fileID)
|
|
|
|
// Download and verify size
|
|
reader2, contentType, err := storage.Download(context.Background(), fileID)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, "image/png", contentType)
|
|
|
|
downloaded, err := io.ReadAll(reader2)
|
|
assert.NoError(t, err)
|
|
|
|
// Decode the downloaded image
|
|
downloadedImg, _, err := image.Decode(bytes.NewReader(downloaded))
|
|
assert.NoError(t, err)
|
|
|
|
// Verify dimensions are unchanged
|
|
bounds := downloadedImg.Bounds()
|
|
assert.Equal(t, 2000, bounds.Dx())
|
|
assert.Equal(t, 2000, bounds.Dy())
|
|
})
|
|
|
|
t.Run("URL Generation", func(t *testing.T) {
|
|
storage, err := New(map[string]interface{}{
|
|
"path": "/__vision_test",
|
|
"compression": true,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
fileID := "20240101/test-12345678.txt"
|
|
url := storage.URL(context.Background(), fileID)
|
|
assert.Equal(t, "/__vision_test/20240101/test-12345678.txt", url)
|
|
})
|
|
|
|
t.Run("Download Non-existent File", func(t *testing.T) {
|
|
storage, err := New(map[string]interface{}{
|
|
"path": "/__vision_test",
|
|
"compression": true,
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
_, _, err = storage.Download(context.Background(), "non-existent.txt")
|
|
assert.Error(t, err)
|
|
})
|
|
}
|