big update: added reactions, groups, media share

server/internal/media/storage_test.go

@@ -0,0 +1,73 @@
+package media
+
+import (
+	"bytes"
+	"context"
+	"errors"
+	"io"
+	"strings"
+	"testing"
+)
+
+func TestLocalSaveOpenDelete(t *testing.T) {
+	ctx := context.Background()
+	store, err := NewLocal(t.TempDir())
+	if err != nil {
+		t.Fatalf("NewLocal: %v", err)
+	}
+
+	payload := []byte("hello attachment")
+	if err := store.Save(ctx, "a/b/file.txt", bytes.NewReader(payload)); err != nil {
+		t.Fatalf("Save: %v", err)
+	}
+
+	rc, err := store.Open(ctx, "a/b/file.txt")
+	if err != nil {
+		t.Fatalf("Open: %v", err)
+	}
+	got, err := io.ReadAll(rc)
+	rc.Close()
+	if err != nil {
+		t.Fatalf("ReadAll: %v", err)
+	}
+	if !bytes.Equal(got, payload) {
+		t.Fatalf("content = %q, want %q", got, payload)
+	}
+
+	if err := store.Delete(ctx, "a/b/file.txt"); err != nil {
+		t.Fatalf("Delete: %v", err)
+	}
+	if err := store.Delete(ctx, "a/b/file.txt"); err != nil {
+		t.Fatalf("Delete idempotent: %v", err)
+	}
+	if _, err := store.Open(ctx, "a/b/file.txt"); !errors.Is(err, ErrNotFound) {
+		t.Fatalf("Open after delete = %v, want ErrNotFound", err)
+	}
+}
+
+func TestLocalPathTraversalRejected(t *testing.T) {
+	store, err := NewLocal(t.TempDir())
+	if err != nil {
+		t.Fatalf("NewLocal: %v", err)
+	}
+	err = store.Save(context.Background(), "../../etc/evil", strings.NewReader("x"))
+	if err == nil {
+		t.Fatal("Save with traversal key = nil, want error")
+	}
+}
+
+func TestIsImage(t *testing.T) {
+	cases := map[string]bool{
+		"image/png":       true,
+		"image/jpeg":      true,
+		"image/gif":       true,
+		"image/webp":      true,
+		"application/pdf": false,
+		"text/plain":      false,
+	}
+	for ct, want := range cases {
+		if got := IsImage(ct); got != want {
+			t.Errorf("IsImage(%q) = %v, want %v", ct, got, want)
+		}
+	}
+}