RC-4

internal/engine/autonomous_queue_v10_test.go

@@ -0,0 +1,61 @@
+package engine
+
+import (
+	"context"
+	"testing"
+	"time"
+
+	"github.com/local/glpi-neural-brain/internal/graph"
+	"github.com/local/glpi-neural-brain/internal/model"
+)
+
+func TestAutonomousResearchSemanticDuplicateSurvivesChangingSeedMembership(t *testing.T) {
+	a := model.ResearchTask{Topic: "T1133 / T1078 / T1588.001", SeedNodeIDs: []string{"a", "b", "c"}, Questions: []string{"Wie hängen T1133 T1078 und T1588.001 zusammen?"}}
+	b := model.ResearchTask{Topic: "T1133 / T1078 / T1588.001", SeedNodeIDs: []string{"a", "d"}, Questions: []string{"Welche Abwehrmaßnahmen betreffen T1133, T1078 und T1588.001?"}}
+	if !autonomousResearchSemanticDuplicate(a, b) {
+		t.Fatalf("same ATT&CK opportunity with changed seeds should dedupe, similarity=%.3f", autonomousResearchTaskSimilarity(a, b))
+	}
+	c := model.ResearchTask{Topic: "Windows Update CBS 0x80D02002", SeedNodeIDs: []string{"x"}}
+	if autonomousResearchSemanticDuplicate(a, c) {
+		t.Fatalf("unrelated opportunities should not dedupe, similarity=%.3f", autonomousResearchTaskSimilarity(a, c))
+	}
+}
+
+func TestMergeQueuedResearchTaskAndPersistentTotals(t *testing.T) {
+	store, err := graph.Open(t.TempDir())
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer store.Close()
+	ctx := context.Background()
+	queued, created, err := store.EnqueueResearchTask(ctx, model.ResearchTask{DedupeKey: "a", Topic: "ATT&CK cluster", RequestedBy: "autonomous-scanner", Priority: .7, SeedNodeIDs: []string{"a"}, Questions: []string{"q1"}}, time.Hour)
+	if err != nil || !created {
+		t.Fatalf("enqueue failed created=%v err=%v", created, err)
+	}
+	merged, changed, err := store.MergeQueuedResearchTask(ctx, queued.ID, model.ResearchTask{Topic: "ATT&CK cluster", Priority: .9, SeedNodeIDs: []string{"b"}, Questions: []string{"q2"}})
+	if err != nil || !changed {
+		t.Fatalf("merge failed changed=%v err=%v", changed, err)
+	}
+	if merged.Priority != .9 || len(merged.SeedNodeIDs) != 2 || len(merged.Questions) != 2 {
+		t.Fatalf("unexpected merged task: %+v", merged)
+	}
+	leased, ok, err := store.LeaseNextResearchTask(ctx, 0, time.Minute)
+	if err != nil || !ok {
+		t.Fatalf("lease failed ok=%v err=%v", ok, err)
+	}
+	if err := store.MarkResearchTaskRunning(ctx, leased.ID); err != nil {
+		t.Fatal(err)
+	}
+	leased.EvidenceCount = 3
+	leased.ArticleCreated = true
+	if err := store.CompleteResearchTask(ctx, leased); err != nil {
+		t.Fatal(err)
+	}
+	totals, err := store.ResearchTaskPersistentTotals(ctx)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if totals["completed"] != 1 || totals["evidence"] != 3 || totals["articles"] != 1 {
+		t.Fatalf("unexpected persistent totals: %+v", totals)
+	}
+}