142 lines
3.0 KiB
Go
142 lines
3.0 KiB
Go
package file_manager
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestS3UploadDownload(t *testing.T) {
|
|
fm := New(
|
|
WithS3(
|
|
"http://localhost:9000",
|
|
"INsEaUOyLx5PXiqLHv0v",
|
|
"xsie9sil2jFtyi0UlnXv7zFafHYk0rWAeROSEYg7",
|
|
"test",
|
|
"us-east-1",
|
|
),
|
|
WithS3PathStyle(true),
|
|
WithTimeout(10*time.Minute),
|
|
WithExpire(24*time.Hour),
|
|
)
|
|
defer fm.CloseManager()
|
|
|
|
ctx := context.Background()
|
|
|
|
content := []byte("hello world from s3")
|
|
size := int64(len(content))
|
|
reader := bytes.NewReader(content)
|
|
|
|
result, err := fm.Create(ctx, "test.txt", size, "")
|
|
if err != nil {
|
|
t.Fatalf("Create failed: %v", err)
|
|
}
|
|
|
|
total, written, err := fm.Upload(ctx, result.Code, 0, size, reader)
|
|
if err != nil {
|
|
t.Fatalf("Upload failed: %v", err)
|
|
}
|
|
if total != size {
|
|
t.Errorf("expected total %d, got %d", size, total)
|
|
}
|
|
if written != size {
|
|
t.Errorf("expected written %d, got %d", size, written)
|
|
}
|
|
|
|
data, err := fm.Get(ctx, result.Code)
|
|
if err != nil {
|
|
t.Fatalf("Get failed: %v", err)
|
|
}
|
|
if !bytes.Equal(data, content) {
|
|
t.Errorf("data mismatch, expected %s, got %s", content, data)
|
|
}
|
|
|
|
info, err := fm.GetInfo(ctx, result.Code)
|
|
if err != nil {
|
|
t.Fatalf("GetInfo failed: %v", err)
|
|
}
|
|
if info.Filename != "test.txt" {
|
|
t.Errorf("expected filename test.txt, got %s", info.Filename)
|
|
}
|
|
if info.Size != size {
|
|
t.Errorf("expected size %d, got %d", size, info.Size)
|
|
}
|
|
if !info.Complete {
|
|
t.Error("expected complete to be true")
|
|
}
|
|
if info.SHA256 == "" {
|
|
t.Error("expected sha256 to be set")
|
|
}
|
|
|
|
err = fm.Delete(ctx, result.Code)
|
|
if err != nil {
|
|
t.Fatalf("Delete failed: %v", err)
|
|
}
|
|
|
|
_, err = fm.Get(ctx, result.Code)
|
|
if err == nil {
|
|
t.Error("expected error after delete")
|
|
}
|
|
}
|
|
|
|
func TestS3UploadLargeFile(t *testing.T) {
|
|
fm := New(
|
|
WithS3(
|
|
"http://localhost:9000",
|
|
"INsEaUOyLx5PXiqLHv0v",
|
|
"xsie9sil2jFtyi0UlnXv7zFafHYk0rWAeROSEYg7",
|
|
"test",
|
|
"us-east-1",
|
|
),
|
|
WithS3PathStyle(true),
|
|
WithTimeout(30*time.Minute),
|
|
WithExpire(24*time.Hour),
|
|
)
|
|
defer fm.CloseManager()
|
|
|
|
ctx := context.Background()
|
|
|
|
filePath := "../../x-file/VID_20240215_195046.mp4"
|
|
file, err := os.Open(filePath)
|
|
if err != nil {
|
|
t.Fatalf("open file failed: %v", err)
|
|
}
|
|
defer file.Close()
|
|
|
|
stat, err := file.Stat()
|
|
if err != nil {
|
|
t.Fatalf("stat file failed: %v", err)
|
|
}
|
|
|
|
result, err := fm.Create(ctx, "video.mp4", stat.Size(), "")
|
|
if err != nil {
|
|
t.Fatalf("Create failed: %v", err)
|
|
}
|
|
|
|
total, written, err := fm.Upload(ctx, result.Code, 0, stat.Size(), file)
|
|
if err != nil {
|
|
t.Fatalf("Upload failed: %v", err)
|
|
}
|
|
if total != stat.Size() {
|
|
t.Errorf("expected total %d, got %d", stat.Size(), total)
|
|
}
|
|
if written != stat.Size() {
|
|
t.Errorf("expected written %d, got %d", stat.Size(), written)
|
|
}
|
|
|
|
data, err := fm.Get(ctx, result.Code)
|
|
if err != nil {
|
|
t.Fatalf("Get failed: %v", err)
|
|
}
|
|
if int64(len(data)) != stat.Size() {
|
|
t.Errorf("expected size %d, got %d", stat.Size(), len(data))
|
|
}
|
|
|
|
err = fm.Delete(ctx, result.Code)
|
|
if err != nil {
|
|
t.Fatalf("Delete failed: %v", err)
|
|
}
|
|
}
|