jbergner
94dbd4ccab
All checks were successful
release-tag / release-image (push) Successful in 2m32s
62 lines
2.5 KiB
Go
62 lines
2.5 KiB
Go
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)
|
|
}
|
|
}
|