feat(media): add MinIO storage switch and HTTPS OSS endpoint
This commit is contained in:
@@ -0,0 +1,51 @@
|
||||
package oss
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"fmt"
|
||||
"io"
|
||||
"os"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/minio/minio-go/v7"
|
||||
)
|
||||
|
||||
func TestLiveMinIOUploadDownload(t *testing.T) {
|
||||
endpoint := os.Getenv("LIVE_MINIO_ENDPOINT")
|
||||
accessKey := os.Getenv("LIVE_MINIO_ACCESS_KEY")
|
||||
secretKey := os.Getenv("LIVE_MINIO_SECRET_KEY")
|
||||
bucket := os.Getenv("LIVE_MINIO_BUCKET")
|
||||
if endpoint == "" || accessKey == "" || secretKey == "" || bucket == "" {
|
||||
t.Skip("live MinIO configuration is not set")
|
||||
}
|
||||
storage, err := newMinIOStorage(minioConfig{
|
||||
Endpoint: endpoint, Bucket: bucket, Region: "us-east-1",
|
||||
AccessKeyID: accessKey, SecretAccessKey: secretKey,
|
||||
})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
||||
defer cancel()
|
||||
key := fmt.Sprintf("im-smoke/%d.txt", time.Now().UnixNano())
|
||||
defer func() {
|
||||
if err := storage.client.RemoveObject(context.Background(), bucket, key, minio.RemoveObjectOptions{}); err != nil {
|
||||
t.Errorf("cleanup object: %v", err)
|
||||
}
|
||||
}()
|
||||
content := []byte("im-minio-smoke-test")
|
||||
if _, err := storage.client.PutObject(ctx, bucket, key, bytes.NewReader(content), int64(len(content)), minio.PutObjectOptions{ContentType: "text/plain"}); err != nil {
|
||||
t.Fatalf("upload: %v", err)
|
||||
}
|
||||
object, err := storage.client.GetObject(ctx, bucket, key, minio.GetObjectOptions{})
|
||||
if err != nil {
|
||||
t.Fatalf("download: %v", err)
|
||||
}
|
||||
defer object.Close()
|
||||
got, err := io.ReadAll(object)
|
||||
if err != nil || !bytes.Equal(got, content) {
|
||||
t.Fatalf("downloaded content differs: %v", err)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user