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,42 @@
+package handler
+
+import (
+	"context"
+	"encoding/json"
+	"testing"
+)
+
+func TestValidateNodeExtractConfigGraphBackend(t *testing.T) {
+	for _, tc := range []struct {
+		name       string
+		neo4j      string
+		legacy     string
+		disabled   bool
+		incomplete bool
+		wantErr    bool
+	}{
+		{name: "no backend", wantErr: true},
+		{name: "neo4j", neo4j: "true"},
+		{name: "native backend", neo4j: "false", legacy: "true"},
+		{name: "case insensitive native backend", legacy: "TRUE"},
+		{name: "disabled extraction needs no backend", disabled: true},
+		{name: "native backend still validates examples", legacy: "true", incomplete: true, wantErr: true},
+	} {
+		t.Run(tc.name, func(t *testing.T) {
+			t.Setenv("NEO4J_ENABLE", tc.neo4j)
+			t.Setenv("LEGACY_GRAPH_ENABLE", tc.legacy)
+			req := &InitializationRequest{}
+			if err := json.Unmarshal([]byte(`{"nodeExtract":{"text":"Alice knows Bob","tags":["knows"],"nodes":[{"name":"Alice"},{"name":"Bob"}],"relations":[{"node1":"Alice","node2":"Bob","type":"knows"}]}}`), req); err != nil {
+				t.Fatal(err)
+			}
+			req.NodeExtract.Enabled = !tc.disabled
+			if tc.incomplete {
+				req.NodeExtract.Relations = nil
+			}
+			err := validateNodeExtractConfig(context.Background(), req)
+			if (err != nil) != tc.wantErr {
+				t.Fatalf("validateNodeExtractConfig() error = %v, wantErr %v", err, tc.wantErr)
+			}
+		})
+	}
+}