73 lines
1.6 KiB
Go
73 lines
1.6 KiB
Go
package s3
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/charmbracelet/log"
|
|
"github.com/johannesboyne/gofakes3"
|
|
"github.com/johannesboyne/gofakes3/backend/s3mem"
|
|
storconfig "github.com/krau/SaveAny-Bot/config/storage"
|
|
)
|
|
|
|
func newTestContext(t *testing.T) context.Context {
|
|
t.Helper()
|
|
logger := log.NewWithOptions(nil, log.Options{ReportTimestamp: false})
|
|
ctx := context.Background()
|
|
return log.WithContext(ctx, logger)
|
|
}
|
|
|
|
func newFakeS3(t *testing.T) (*S3, *storconfig.S3StorageConfig) {
|
|
t.Helper()
|
|
|
|
backend := s3mem.New()
|
|
fakeSrv := gofakes3.New(backend)
|
|
ts := httptest.NewServer(fakeSrv.Server())
|
|
t.Cleanup(ts.Close)
|
|
|
|
cfg := &storconfig.S3StorageConfig{
|
|
BaseConfig: storconfig.BaseConfig{
|
|
Name: "test-s3",
|
|
Type: "s3",
|
|
Enable: true,
|
|
},
|
|
Endpoint: ts.URL,
|
|
AccessKeyID: "test-access-key",
|
|
SecretAccessKey: "test-secret",
|
|
BucketName: "test-bucket",
|
|
BasePath: "base",
|
|
Region: "us-east-1",
|
|
}
|
|
|
|
if err := backend.CreateBucket("test-bucket"); err != nil {
|
|
t.Fatalf("failed to create fake bucket: %v", err)
|
|
}
|
|
|
|
s := &S3{}
|
|
ctx := newTestContext(t)
|
|
if err := s.Init(ctx, cfg); err != nil {
|
|
t.Fatalf("init s3 failed: %v", err)
|
|
}
|
|
|
|
return s, cfg
|
|
}
|
|
|
|
func TestS3_SaveAndExists(t *testing.T) {
|
|
s, _ := newFakeS3(t)
|
|
ctx := context.Background()
|
|
|
|
content := []byte("hello world")
|
|
reader := bytes.NewReader(content)
|
|
key := "foo/bar.txt"
|
|
|
|
if err := s.Save(ctx, reader, key); err != nil {
|
|
t.Fatalf("Save failed: %v", err)
|
|
}
|
|
|
|
if !s.Exists(ctx, key) {
|
|
t.Fatalf("Exists should return true for saved key")
|
|
}
|
|
}
|