jbergner
440423c5b6
All checks were successful
release-tag / release-image (push) Successful in 2m43s
71 lines
2.4 KiB
Go
71 lines
2.4 KiB
Go
package sourceagent
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/local/glpi-neural-brain/internal/articlequality"
|
|
)
|
|
|
|
func TestExecuteArticleQualityJobIsDeterministicAndModelFree(t *testing.T) {
|
|
req := ArticleQualityComputeRequest{JobID: "q1", Payload: articlequality.Request{
|
|
ArticleType: "reference",
|
|
Title: "Kurze Referenz",
|
|
Problem: "Ein kurzer technischer Hinweis.",
|
|
Answer: "## Kernaussagen\n- T1018 beschreibt System Discovery.",
|
|
Sources: []articlequality.Document{{ID: "s1", Text: "T1018 Remote System Discovery ATT&CK"}},
|
|
}}
|
|
a := ExecuteArticleQualityJob(req)
|
|
b := ExecuteArticleQualityJob(req)
|
|
if a.Quality.Algorithm != articlequality.Algorithm || b.Quality.Algorithm != articlequality.Algorithm {
|
|
t.Fatalf("unexpected algorithm: %+v %+v", a.Quality, b.Quality)
|
|
}
|
|
if a.Quality.Passed || b.Quality.Passed || a.Quality.Score != b.Quality.Score || a.Quality.WordCount != b.Quality.WordCount {
|
|
t.Fatalf("article quality CPU result must be deterministic and reject the short reference: %+v %+v", a.Quality, b.Quality)
|
|
}
|
|
}
|
|
|
|
func TestArticleQualityComputeClaimCompleteLifecycle(t *testing.T) {
|
|
s := &Store{articleQualityJobs: map[string]*articleQualityJobState{}}
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Second)
|
|
defer cancel()
|
|
done := make(chan ArticleQualityComputeResult, 1)
|
|
errCh := make(chan error, 1)
|
|
go func() {
|
|
res, err := s.SubmitArticleQualityJob(ctx, ArticleQualityComputeRequest{JobID: "job-1", Payload: articlequality.Request{ArticleType: "reference", Title: "T", Answer: "kurz"}})
|
|
if err != nil {
|
|
errCh <- err
|
|
return
|
|
}
|
|
done <- res
|
|
}()
|
|
deadline := time.Now().Add(500 * time.Millisecond)
|
|
var job ArticleQualityComputeRequest
|
|
var ok bool
|
|
for time.Now().Before(deadline) {
|
|
job, ok = s.ClaimArticleQualityJob("agent-1", time.Minute)
|
|
if ok {
|
|
break
|
|
}
|
|
time.Sleep(time.Millisecond)
|
|
}
|
|
if !ok || job.JobID != "job-1" {
|
|
t.Fatalf("job was not claimable: %+v ok=%v", job, ok)
|
|
}
|
|
result := ExecuteArticleQualityJob(job)
|
|
if err := s.CompleteArticleQualityJob("agent-1", result); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
select {
|
|
case err := <-errCh:
|
|
t.Fatal(err)
|
|
case res := <-done:
|
|
if res.JobID != "job-1" || res.Quality.Algorithm != articlequality.Algorithm {
|
|
t.Fatalf("unexpected completed result: %+v", res)
|
|
}
|
|
case <-ctx.Done():
|
|
t.Fatal("timed out waiting for completed quality job")
|
|
}
|
|
}
|