WeKnora deployment source with the reviewed wiki identity, summary and recovery fixes

2026-09-22 16:19:06 +08:00
commit f7fc992f57
3929 changed files with 1006607 additions and 0 deletions
@@ -0,0 +1,45 @@
+package session
+
+import (
+	"context"
+	"testing"
+
+	"github.com/Tencent/WeKnora/internal/types"
+	"github.com/Tencent/WeKnora/internal/types/interfaces"
+	"github.com/stretchr/testify/require"
+)
+
+type uploadedResourceCatalog struct {
+	interfaces.ResourceCatalog
+	tenant uint64
+	bound  []types.ResourceBinding
+}
+
+func (c *uploadedResourceCatalog) Resolve(context.Context, string) (*types.StoredResource, error) {
+	return &types.StoredResource{TenantID: c.tenant}, nil
+}
+
+func (c *uploadedResourceCatalog) Bind(_ context.Context, ref, ownerType, ownerID, relation string) error {
+	c.bound = append(c.bound, types.ResourceBinding{ResourceID: ref, OwnerType: ownerType, OwnerID: ownerID, Relation: relation})
+	return nil
+}
+
+func TestBindUploadedMessageResources(t *testing.T) {
+	ctx := context.WithValue(context.Background(), types.TenantIDContextKey, uint64(7))
+	ref := types.BuildResourcePath("AbCdEfGhIjKlMnOpQrStUv")
+	catalog := &uploadedResourceCatalog{tenant: 7}
+	h := &Handler{resourceCatalog: catalog}
+	message := &types.Message{
+		ID: "message-1", Content: types.BuildResourcePath("XbCdEfGhIjKlMnOpQrStUv"),
+		Images: types.MessageImages{{URL: ref}}, Attachments: types.MessageAttachments{{URL: ref}},
+	}
+	require.NoError(t, h.bindUploadedMessageResources(ctx, message))
+	require.Len(t, catalog.bound, 1)
+	require.Equal(t, types.ResourceOwnerMessage, catalog.bound[0].OwnerType)
+	require.Equal(t, "message-1", catalog.bound[0].OwnerID)
+
+	catalog.bound = nil
+	catalog.tenant = 8
+	require.Error(t, h.bindUploadedMessageResources(ctx, message))
+	require.Empty(t, catalog.bound)
+}