0.29.1原版

2026-07-02 19:14:14 +08:00
commit d94008f0fb
947 changed files with 174905 additions and 0 deletions
@@ -0,0 +1,305 @@
+package test
+
+import (
+	"context"
+	"fmt"
+	"testing"
+
+	"github.com/lithammer/shortuuid/v4"
+	"github.com/stretchr/testify/require"
+
+	storepb "github.com/usememos/memos/proto/gen/store"
+	"github.com/usememos/memos/store"
+)
+
+func TestAttachmentNeedsInstanceStorageSetting(t *testing.T) {
+	tests := []struct {
+		name       string
+		attachment *store.Attachment
+		want       bool
+	}{
+		{
+			name: "nil attachment",
+		},
+		{
+			name: "local attachment",
+			attachment: &store.Attachment{
+				StorageType: storepb.AttachmentStorageType_LOCAL,
+			},
+		},
+		{
+			name: "s3 attachment without payload",
+			attachment: &store.Attachment{
+				StorageType: storepb.AttachmentStorageType_S3,
+			},
+		},
+		{
+			name: "s3 attachment with embedded config",
+			attachment: &store.Attachment{
+				StorageType: storepb.AttachmentStorageType_S3,
+				Payload: &storepb.AttachmentPayload{
+					Payload: &storepb.AttachmentPayload_S3Object_{
+						S3Object: &storepb.AttachmentPayload_S3Object{
+							S3Config: &storepb.StorageS3Config{},
+						},
+					},
+				},
+			},
+		},
+		{
+			name: "s3 attachment without embedded config",
+			attachment: &store.Attachment{
+				StorageType: storepb.AttachmentStorageType_S3,
+				Payload: &storepb.AttachmentPayload{
+					Payload: &storepb.AttachmentPayload_S3Object_{
+						S3Object: &storepb.AttachmentPayload_S3Object{},
+					},
+				},
+			},
+			want: true,
+		},
+	}
+
+	for _, test := range tests {
+		t.Run(test.name, func(t *testing.T) {
+			if got := store.AttachmentNeedsInstanceStorageSetting(test.attachment); got != test.want {
+				t.Fatalf("AttachmentNeedsInstanceStorageSetting() = %v, want %v", got, test.want)
+			}
+		})
+	}
+}
+
+func TestAttachmentStore(t *testing.T) {
+	t.Parallel()
+	ctx := context.Background()
+	ts := NewTestingStore(ctx, t)
+	_, err := ts.CreateAttachment(ctx, &store.Attachment{
+		UID:       shortuuid.New(),
+		CreatorID: 101,
+		Filename:  "test.epub",
+		Blob:      []byte("test"),
+		Type:      "application/epub+zip",
+		Size:      637607,
+	})
+	require.NoError(t, err)
+
+	correctFilename := "test.epub"
+	incorrectFilename := "test.png"
+	attachment, err := ts.GetAttachment(ctx, &store.FindAttachment{
+		Filename: &correctFilename,
+	})
+	require.NoError(t, err)
+	require.Equal(t, correctFilename, attachment.Filename)
+	require.Equal(t, int32(1), attachment.ID)
+
+	notFoundAttachment, err := ts.GetAttachment(ctx, &store.FindAttachment{
+		Filename: &incorrectFilename,
+	})
+	require.NoError(t, err)
+	require.Nil(t, notFoundAttachment)
+
+	var correctCreatorID int32 = 101
+	var incorrectCreatorID int32 = 102
+	_, err = ts.GetAttachment(ctx, &store.FindAttachment{
+		CreatorID: &correctCreatorID,
+	})
+	require.NoError(t, err)
+
+	notFoundAttachment, err = ts.GetAttachment(ctx, &store.FindAttachment{
+		CreatorID: &incorrectCreatorID,
+	})
+	require.NoError(t, err)
+	require.Nil(t, notFoundAttachment)
+
+	err = ts.DeleteAttachment(ctx, &store.DeleteAttachment{
+		ID: 1,
+	})
+	require.NoError(t, err)
+	err = ts.DeleteAttachment(ctx, &store.DeleteAttachment{
+		ID: 2,
+	})
+	require.ErrorContains(t, err, "attachment not found")
+	ts.Close()
+}
+
+func TestAttachmentStoreWithFilter(t *testing.T) {
+	t.Parallel()
+	ctx := context.Background()
+	ts := NewTestingStore(ctx, t)
+
+	_, err := ts.CreateAttachment(ctx, &store.Attachment{
+		UID:       shortuuid.New(),
+		CreatorID: 101,
+		Filename:  "test.png",
+		Blob:      []byte("test"),
+		Type:      "image/png",
+		Size:      1000,
+	})
+	require.NoError(t, err)
+
+	_, err = ts.CreateAttachment(ctx, &store.Attachment{
+		UID:       shortuuid.New(),
+		CreatorID: 101,
+		Filename:  "test.jpg",
+		Blob:      []byte("test"),
+		Type:      "image/jpeg",
+		Size:      2000,
+	})
+	require.NoError(t, err)
+
+	_, err = ts.CreateAttachment(ctx, &store.Attachment{
+		UID:       shortuuid.New(),
+		CreatorID: 101,
+		Filename:  "test.pdf",
+		Blob:      []byte("test"),
+		Type:      "application/pdf",
+		Size:      3000,
+	})
+	require.NoError(t, err)
+
+	attachments, err := ts.ListAttachments(ctx, &store.FindAttachment{
+		CreatorID: &[]int32{101}[0],
+		Filters:   []string{`mime_type == "image/png"`},
+	})
+	require.NoError(t, err)
+	require.Len(t, attachments, 1)
+	require.Equal(t, "image/png", attachments[0].Type)
+
+	attachments, err = ts.ListAttachments(ctx, &store.FindAttachment{
+		CreatorID: &[]int32{101}[0],
+		Filters:   []string{`mime_type in ["image/png", "image/jpeg"]`},
+	})
+	require.NoError(t, err)
+	require.Len(t, attachments, 2)
+
+	attachments, err = ts.ListAttachments(ctx, &store.FindAttachment{
+		CreatorID: &[]int32{101}[0],
+		Filters:   []string{`filename.contains("test")`},
+	})
+	require.NoError(t, err)
+	require.Len(t, attachments, 3)
+
+	ts.Close()
+}
+
+func TestAttachmentUpdate(t *testing.T) {
+	t.Parallel()
+	ctx := context.Background()
+	ts := NewTestingStore(ctx, t)
+
+	attachment, err := ts.CreateAttachment(ctx, &store.Attachment{
+		UID:       shortuuid.New(),
+		CreatorID: 101,
+		Filename:  "original.png",
+		Blob:      []byte("test"),
+		Type:      "image/png",
+		Size:      1000,
+	})
+	require.NoError(t, err)
+
+	// Update filename
+	newFilename := "updated.png"
+	err = ts.UpdateAttachment(ctx, &store.UpdateAttachment{
+		ID:       attachment.ID,
+		Filename: &newFilename,
+	})
+	require.NoError(t, err)
+
+	// Verify update
+	found, err := ts.GetAttachment(ctx, &store.FindAttachment{ID: &attachment.ID})
+	require.NoError(t, err)
+	require.Equal(t, newFilename, found.Filename)
+
+	ts.Close()
+}
+
+func TestAttachmentGetByUID(t *testing.T) {
+	t.Parallel()
+	ctx := context.Background()
+	ts := NewTestingStore(ctx, t)
+
+	uid := shortuuid.New()
+	_, err := ts.CreateAttachment(ctx, &store.Attachment{
+		UID:       uid,
+		CreatorID: 101,
+		Filename:  "test.png",
+		Blob:      []byte("test"),
+		Type:      "image/png",
+		Size:      1000,
+	})
+	require.NoError(t, err)
+
+	// Get by UID
+	found, err := ts.GetAttachment(ctx, &store.FindAttachment{UID: &uid})
+	require.NoError(t, err)
+	require.NotNil(t, found)
+	require.Equal(t, uid, found.UID)
+
+	// Get non-existent UID
+	nonExistentUID := "non-existent-uid"
+	notFound, err := ts.GetAttachment(ctx, &store.FindAttachment{UID: &nonExistentUID})
+	require.NoError(t, err)
+	require.Nil(t, notFound)
+
+	ts.Close()
+}
+
+func TestAttachmentListWithPagination(t *testing.T) {
+	t.Parallel()
+	ctx := context.Background()
+	ts := NewTestingStore(ctx, t)
+
+	// Create 5 attachments
+	for i := 0; i < 5; i++ {
+		_, err := ts.CreateAttachment(ctx, &store.Attachment{
+			UID:       shortuuid.New(),
+			CreatorID: 101,
+			Filename:  fmt.Sprintf("test%d.png", i),
+			Blob:      []byte("test"),
+			Type:      "image/png",
+			Size:      int64(1000 + i),
+		})
+		require.NoError(t, err)
+	}
+
+	// Test limit
+	limit := 3
+	attachments, err := ts.ListAttachments(ctx, &store.FindAttachment{
+		CreatorID: &[]int32{101}[0],
+		Limit:     &limit,
+	})
+	require.NoError(t, err)
+	require.Equal(t, 3, len(attachments))
+
+	// Test offset
+	offset := 2
+	offsetAttachments, err := ts.ListAttachments(ctx, &store.FindAttachment{
+		CreatorID: &[]int32{101}[0],
+		Limit:     &limit,
+		Offset:    &offset,
+	})
+	require.NoError(t, err)
+	require.Equal(t, 3, len(offsetAttachments))
+
+	ts.Close()
+}
+
+func TestAttachmentInvalidUID(t *testing.T) {
+	t.Parallel()
+	ctx := context.Background()
+	ts := NewTestingStore(ctx, t)
+
+	// Create with invalid UID (contains spaces)
+	_, err := ts.CreateAttachment(ctx, &store.Attachment{
+		UID:       "invalid uid with spaces",
+		CreatorID: 101,
+		Filename:  "test.png",
+		Blob:      []byte("test"),
+		Type:      "image/png",
+		Size:      1000,
+	})
+	require.Error(t, err)
+	require.Contains(t, err.Error(), "invalid uid")
+
+	ts.Close()
+}