pphua
9793b522d9
Add read-only task context inheritance, ordered task-level LLM failover, live Worker controls, cascading task cleanup, FIFO task concurrency, cancellation diagnostics, UI updates, tests, build script, and changelog.
101 lines
2.8 KiB
Go
101 lines
2.8 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/Autumn-27/artex/db"
|
|
)
|
|
|
|
func TestInheritedActivityDetailAndRelationDeletion(t *testing.T) {
|
|
m, err := NewManager(t.TempDir(), "")
|
|
if err != nil {
|
|
t.Skipf("postgres unavailable (%v) - skipping", err)
|
|
}
|
|
defer m.Close()
|
|
|
|
source, err := m.CreateTaskWithOptions("detail source", "source goal", db.TaskCreateOptions{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
sourceID := mustTaskID(t, source.ID)
|
|
current, err := m.CreateTaskWithOptions("detail current", "current goal", db.TaskCreateOptions{SourceTaskIDs: []int64{sourceID}})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
t.Cleanup(func() {
|
|
_, _ = m.DeleteTask(current.ID, DeleteTaskOptions{})
|
|
_, _ = m.DeleteTask(source.ID, DeleteTaskOptions{})
|
|
})
|
|
|
|
plannerStep, err := source.Store.AppendActivity(db.Activity{
|
|
Worker: "planner", Kind: "text", Summary: "private planner", Detail: "private planner detail",
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
intentID, err := source.Store.AddIntent(map[string]any{"summary": "completed source work"}, 1, nil, "planner")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := source.Store.SetIntentState(intentID, "done"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
workerStep, err := source.Store.AppendActivity(db.Activity{
|
|
NodeID: &intentID, Worker: "worker", Kind: "result", Summary: "shared result", Detail: "shared worker detail",
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
s := New(context.Background(), m, t.TempDir(), t.TempDir(), t.TempDir())
|
|
token, err := signJWT(s.jwtKey)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
fetchDetail := func(stepID int64) string {
|
|
req := httptest.NewRequest(http.MethodGet,
|
|
fmt.Sprintf("/api/exploration/activity/%d?task=%s", stepID, current.ID), nil)
|
|
req.Header.Set("Authorization", "Bearer "+token)
|
|
rec := httptest.NewRecorder()
|
|
s.Handler().ServeHTTP(rec, req)
|
|
if rec.Code != http.StatusOK {
|
|
t.Fatalf("activity detail %d: status=%d body=%s", stepID, rec.Code, rec.Body.String())
|
|
}
|
|
var body struct {
|
|
Detail string `json:"detail"`
|
|
}
|
|
if err := json.Unmarshal(rec.Body.Bytes(), &body); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return body.Detail
|
|
}
|
|
if got := fetchDetail(plannerStep); got != "" {
|
|
t.Fatalf("source planner transcript leaked through API: %q", got)
|
|
}
|
|
if got := fetchDetail(workerStep); got != "shared worker detail" {
|
|
t.Fatalf("terminal inherited worker detail missing: %q", got)
|
|
}
|
|
|
|
if _, err := m.DeleteTask(source.ID, DeleteTaskOptions{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
currentAfter, ok := m.Task(current.ID)
|
|
if !ok || len(currentAfter.SourceTaskIDs) != 0 {
|
|
t.Fatalf("deleted source remained in live task DTO state: %+v", currentAfter)
|
|
}
|
|
}
|
|
|
|
func mustTaskID(t *testing.T, raw string) int64 {
|
|
t.Helper()
|
|
var id int64
|
|
if _, err := fmt.Sscan(raw, &id); err != nil || id <= 0 {
|
|
t.Fatalf("bad task id %q: %v", raw, err)
|
|
}
|
|
return id
|
|
}
|