Carlos Florêncio
99 lines
2.5 KiB
Go
99 lines
2.5 KiB
Go
package runtime
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/kandev/kandev/internal/office/models"
|
|
)
|
|
|
|
func TestContextBuilderBuildsAndPersistsRuntimeSnapshot(t *testing.T) {
|
|
agents := &recordingAgentReader{
|
|
agent: &models.AgentInstance{
|
|
ID: "agent-1",
|
|
WorkspaceID: "ws-1",
|
|
Role: models.AgentRoleCEO,
|
|
},
|
|
}
|
|
store := &recordingRunSnapshotStore{}
|
|
builder := ContextBuilder{Agents: agents, Runs: store}
|
|
run := &models.Run{
|
|
ID: "run-1",
|
|
AgentProfileID: "agent-1",
|
|
Reason: "task_assigned",
|
|
Payload: `{"task_id":"task-1","session_id":"session-1"}`,
|
|
}
|
|
|
|
runCtx, err := builder.BuildAndPersist(context.Background(), run)
|
|
if err != nil {
|
|
t.Fatalf("BuildAndPersist: %v", err)
|
|
}
|
|
|
|
if runCtx.WorkspaceID != "ws-1" || runCtx.AgentID != "agent-1" {
|
|
t.Fatalf("unexpected identity: %+v", runCtx)
|
|
}
|
|
if runCtx.TaskID != "task-1" || runCtx.SessionID != "session-1" {
|
|
t.Fatalf("unexpected task/session: %+v", runCtx)
|
|
}
|
|
if !runCtx.Capabilities.Allows(CapabilityCreateAgent) {
|
|
t.Fatal("expected CEO runtime context to allow create_agent")
|
|
}
|
|
if len(store.calls) != 1 {
|
|
t.Fatalf("expected 1 snapshot write, got %d", len(store.calls))
|
|
}
|
|
var caps Capabilities
|
|
if err := json.Unmarshal([]byte(store.calls[0].Capabilities), &caps); err != nil {
|
|
t.Fatalf("decode capabilities: %v", err)
|
|
}
|
|
if !caps.Allows(CapabilityCreateAgent) {
|
|
t.Fatal("persisted capabilities should include create_agent")
|
|
}
|
|
}
|
|
|
|
type recordingAgentReader struct {
|
|
agent *models.AgentInstance
|
|
}
|
|
|
|
func (r *recordingAgentReader) GetAgentInstance(_ context.Context, _ string) (*models.AgentInstance, error) {
|
|
return r.agent, nil
|
|
}
|
|
|
|
func (r *recordingAgentReader) ListAgentInstances(_ context.Context, _ string) ([]*models.AgentInstance, error) {
|
|
return []*models.AgentInstance{r.agent}, nil
|
|
}
|
|
|
|
func (r *recordingAgentReader) ListAgentInstancesByIDs(
|
|
_ context.Context,
|
|
_ []string,
|
|
) ([]*models.AgentInstance, error) {
|
|
return []*models.AgentInstance{r.agent}, nil
|
|
}
|
|
|
|
type recordingRunSnapshotStore struct {
|
|
calls []snapshotCall
|
|
}
|
|
|
|
type snapshotCall struct {
|
|
RunID string
|
|
Capabilities string
|
|
InputSnapshot string
|
|
SessionID string
|
|
}
|
|
|
|
func (s *recordingRunSnapshotStore) UpdateRunRuntimeSnapshot(
|
|
_ context.Context,
|
|
id string,
|
|
capabilities string,
|
|
inputSnapshot string,
|
|
sessionID string,
|
|
) error {
|
|
s.calls = append(s.calls, snapshotCall{
|
|
RunID: id,
|
|
Capabilities: capabilities,
|
|
InputSnapshot: inputSnapshot,
|
|
SessionID: sessionID,
|
|
})
|
|
return nil
|
|
}
|