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

internal/application/service/knowledge_process_status_test.go

@@ -0,0 +1,106 @@
+package service
+
+import (
+	"testing"
+	"time"
+
+	"github.com/Tencent/WeKnora/internal/types"
+)
+
+func TestFinalizeIndexedKnowledgeState(t *testing.T) {
+	now := time.Date(2026, 5, 20, 12, 0, 0, 0, time.UTC)
+
+	tests := []struct {
+		name                 string
+		hasPendingMultimodal bool
+		textChunkCount       int
+		wantParseStatus      string
+		wantSummaryStatus    string
+		wantErrorMessage     string
+	}{
+		{
+			name:              "text document stays processing so post-process can fan out enrichment",
+			textChunkCount:    2,
+			wantParseStatus:   types.ParseStatusProcessing,
+			wantSummaryStatus: types.SummaryStatusNone,
+			// Still processing: FinalizeSubtask clears the column when this
+			// row is eventually promoted to completed.
+			wantErrorMessage: "previous attempt failed",
+		},
+		{
+			name:              "empty indexed document is completed without summary work",
+			textChunkCount:    0,
+			wantParseStatus:   types.ParseStatusCompleted,
+			wantSummaryStatus: types.SummaryStatusNone,
+			wantErrorMessage:  "",
+		},
+		{
+			name:                 "document waits while multimodal image work is pending",
+			hasPendingMultimodal: true,
+			textChunkCount:       2,
+			wantParseStatus:      types.ParseStatusProcessing,
+			wantSummaryStatus:    types.SummaryStatusNone,
+			wantErrorMessage:     "previous attempt failed",
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			knowledge := &types.Knowledge{
+				ParseStatus:   types.ParseStatusProcessing,
+				SummaryStatus: types.SummaryStatusCompleted,
+				ErrorMessage:  "previous attempt failed",
+			}
+
+			finalizeIndexedKnowledgeState(knowledge, 4096, tt.textChunkCount, tt.hasPendingMultimodal, now)
+
+			if knowledge.ErrorMessage != tt.wantErrorMessage {
+				t.Fatalf("ErrorMessage = %q, want %q", knowledge.ErrorMessage, tt.wantErrorMessage)
+			}
+
+			if knowledge.ParseStatus != tt.wantParseStatus {
+				t.Fatalf("ParseStatus = %q, want %q", knowledge.ParseStatus, tt.wantParseStatus)
+			}
+			if knowledge.SummaryStatus != tt.wantSummaryStatus {
+				t.Fatalf("SummaryStatus = %q, want %q", knowledge.SummaryStatus, tt.wantSummaryStatus)
+			}
+			if knowledge.EnableStatus != "enabled" {
+				t.Fatalf("EnableStatus = %q, want enabled", knowledge.EnableStatus)
+			}
+			if knowledge.StorageSize != 4096 {
+				t.Fatalf("StorageSize = %d, want 4096", knowledge.StorageSize)
+			}
+			if knowledge.ProcessedAt == nil || !knowledge.ProcessedAt.Equal(now) {
+				t.Fatalf("ProcessedAt = %v, want %v", knowledge.ProcessedAt, now)
+			}
+			if !knowledge.UpdatedAt.Equal(now) {
+				t.Fatalf("UpdatedAt = %v, want %v", knowledge.UpdatedAt, now)
+			}
+		})
+	}
+}
+
+// TestMarkKnowledgeProcessingClearsPreviousAttemptError covers the transition
+// every worker performs before it starts a new attempt. A row that failed
+// earlier still carries that attempt's error_message, and leaving it in place
+// makes the UI report a failure on a document it is simultaneously showing as
+// processing.
+func TestMarkKnowledgeProcessingClearsPreviousAttemptError(t *testing.T) {
+	now := time.Date(2026, 5, 20, 12, 0, 0, 0, time.UTC)
+	knowledge := &types.Knowledge{
+		ParseStatus:  types.ParseStatusFailed,
+		ErrorMessage: "Task interrupted due to application restart",
+	}
+
+	markKnowledgeProcessing(knowledge, now)
+
+	if knowledge.ParseStatus != types.ParseStatusProcessing {
+		t.Fatalf("ParseStatus = %q, want %q", knowledge.ParseStatus, types.ParseStatusProcessing)
+	}
+	if knowledge.ErrorMessage != "" {
+		t.Fatalf("ErrorMessage = %q, want empty", knowledge.ErrorMessage)
+	}
+	if !knowledge.UpdatedAt.Equal(now) {
+		t.Fatalf("UpdatedAt = %v, want %v", knowledge.UpdatedAt, now)
+	}
+}