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

internal/application/repository/knowledge_metadata_prefix_test.go

@@ -0,0 +1,112 @@
+package repository
+
+import (
+	"context"
+	"fmt"
+	"testing"
+
+	"github.com/google/uuid"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+// TestFindByMetadataKeyPrefix verifies that FindByMetadataKeyPrefix returns only
+// knowledge items whose metadata["external_id"] starts with the given prefix.
+//
+// Setup: three rows in the same KB:
+//   - "nodeA"          — the parent document; must NOT be matched by prefix "nodeA#"
+//   - "nodeA#file#x"   — an attachment child; MUST be matched
+//   - "nodeB"          — a sibling document; must NOT be matched
+//
+// A fourth row in a different KB with a matching external_id is also inserted to
+// verify tenant/KB isolation.
+func TestFindByMetadataKeyPrefix(t *testing.T) {
+	db := setupKnowledgeTestDB(t)
+	repo := NewKnowledgeRepository(db).(*knowledgeRepository)
+	ctx := context.Background()
+
+	const tenantID uint64 = 42
+	kbID := uuid.New().String()
+	otherKBID := uuid.New().String()
+
+	insertRow := func(tid uint64, kb, extID string) string {
+		id := uuid.New().String()
+		metadata := fmt.Sprintf(`{"external_id":%q}`, extID)
+		require.NoError(t, db.Exec(`
+			INSERT INTO knowledges
+			  (id, tenant_id, knowledge_base_id, type, title, source, parse_status, metadata)
+			VALUES (?, ?, ?, 'document', ?, 'feishu', 'completed', ?)
+		`, id, tid, kb, extID, metadata).Error)
+		return id
+	}
+
+	_ = insertRow(tenantID, kbID, "nodeA")               // parent — must NOT match
+	childID := insertRow(tenantID, kbID, "nodeA#file#x") // attachment child — MUST match
+	_ = insertRow(tenantID, kbID, "nodeB")               // sibling — must NOT match
+	_ = insertRow(tenantID, otherKBID, "nodeA#file#y")   // different KB — must be excluded
+
+	results, err := repo.FindByMetadataKeyPrefix(ctx, tenantID, kbID, "external_id", "nodeA#")
+	require.NoError(t, err)
+
+	require.Len(t, results, 1, "only the attachment child should match the prefix 'nodeA#'")
+	assert.Equal(t, childID, results[0].ID)
+}
+
+// TestFindByMetadataKeyPrefix_DeletedRowsExcluded verifies that soft-deleted rows
+// (deleted_at IS NOT NULL) are excluded from the prefix query.
+func TestFindByMetadataKeyPrefix_DeletedRowsExcluded(t *testing.T) {
+	db := setupKnowledgeTestDB(t)
+	repo := NewKnowledgeRepository(db).(*knowledgeRepository)
+	ctx := context.Background()
+
+	const tenantID uint64 = 43
+	kbID := uuid.New().String()
+
+	id := uuid.New().String()
+	metadata := fmt.Sprintf(`{"external_id":%q}`, "nodeC#file#z")
+	require.NoError(t, db.Exec(`
+		INSERT INTO knowledges
+		  (id, tenant_id, knowledge_base_id, type, title, source, parse_status, metadata, deleted_at)
+		VALUES (?, ?, ?, 'document', 'deleted-child', 'feishu', 'completed', ?, '2026-01-01 00:00:00')
+	`, id, tenantID, kbID, metadata).Error)
+
+	results, err := repo.FindByMetadataKeyPrefix(ctx, tenantID, kbID, "external_id", "nodeC#")
+	require.NoError(t, err)
+	assert.Empty(t, results, "soft-deleted rows must not appear in prefix results")
+}
+
+// TestFindByMetadataKeyPrefix_UnderscoreIsLiteral verifies that an underscore in
+// the prefix is treated as a literal character and does NOT act as a SQL LIKE
+// wildcard. Without escapeLikeKeyword applied to the prefix, a query for "ab_c#"
+// would also match "abXc#file#y" because '_' is a wildcard.
+func TestFindByMetadataKeyPrefix_UnderscoreIsLiteral(t *testing.T) {
+	db := setupKnowledgeTestDB(t)
+	repo := NewKnowledgeRepository(db).(*knowledgeRepository)
+	ctx := context.Background()
+
+	const tenantID uint64 = 44
+	kbID := uuid.New().String()
+
+	insertRow := func(extID string) string {
+		id := uuid.New().String()
+		metadata := fmt.Sprintf(`{"external_id":%q}`, extID)
+		require.NoError(t, db.Exec(`
+			INSERT INTO knowledges
+			  (id, tenant_id, knowledge_base_id, type, title, source, parse_status, metadata)
+			VALUES (?, ?, ?, 'document', ?, 'feishu', 'completed', ?)
+		`, id, tenantID, kbID, extID, metadata).Error)
+		return id
+	}
+
+	// The prefix we will search for is "ab_c#". Without escaping, '_' is a
+	// wildcard that matches any single character, so "abXc#file#y" would also
+	// be returned. With escaping it must NOT be returned.
+	targetID := insertRow("ab_c#file#x") // literal underscore — MUST match
+	_ = insertRow("abXc#file#y")         // 'X' instead of '_' — must NOT match
+
+	results, err := repo.FindByMetadataKeyPrefix(ctx, tenantID, kbID, "external_id", "ab_c#")
+	require.NoError(t, err)
+
+	require.Len(t, results, 1, "underscore in prefix must be treated as literal: only 'ab_c#file#x' should match")
+	assert.Equal(t, targetID, results[0].ID)
+}