fix: add VirtualHost option to S3StorageConfig and implement endpoint validation, close #150

storage/s3/s3_test.go

@@ -0,0 +1,72 @@
+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")
+	}
+}