feat: impl collection store

test/store/collection_test.go

@@ -0,0 +1,49 @@
+package teststore
+
+import (
+	"context"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+
+	storepb "github.com/boojack/slash/proto/gen/store"
+	"github.com/boojack/slash/store"
+)
+
+func TestCollectionStore(t *testing.T) {
+	ctx := context.Background()
+	ts := NewTestingStore(ctx, t)
+	user, err := createTestingAdminUser(ctx, ts)
+	require.NoError(t, err)
+	collection, err := ts.CreateCollection(ctx, &storepb.Collection{
+		CreatorId:   user.ID,
+		Name:        "test",
+		Title:       "My collection",
+		Description: "A test collection",
+		ShortcutIds: []int32{101, 102},
+		Visibility:  storepb.Visibility_PRIVATE,
+	})
+	require.NoError(t, err)
+	collections, err := ts.ListCollections(ctx, &store.FindCollection{
+		CreatorID: &user.ID,
+	})
+	require.NoError(t, err)
+	require.Equal(t, 1, len(collections))
+	require.Equal(t, collection, collections[0])
+	newShortcutIds := []int32{101, 103}
+	updatedCollection, err := ts.UpdateCollection(ctx, &store.UpdateCollection{
+		ID:          collection.Id,
+		ShortcutIDs: newShortcutIds,
+	})
+	require.NoError(t, err)
+	require.Equal(t, newShortcutIds, updatedCollection.ShortcutIds)
+	err = ts.DeleteCollection(ctx, &store.DeleteCollection{
+		ID: collection.Id,
+	})
+	require.NoError(t, err)
+	collections, err = ts.ListCollections(ctx, &store.FindCollection{
+		CreatorID: &user.ID,
+	})
+	require.NoError(t, err)
+	require.Equal(t, 0, len(collections))
+}