| 1 | package control |
| 2 | |
| 3 | import ( |
| 4 | "context" |
| 5 | "strings" |
| 6 | "testing" |
| 7 | |
| 8 | "reasonix/internal/agent" |
| 9 | "reasonix/internal/event" |
| 10 | "reasonix/internal/provider" |
| 11 | "reasonix/internal/skill" |
| 12 | "reasonix/internal/tool" |
| 13 | ) |
| 14 | |
| 15 | func TestStructuredResearchReceivesExpandedPastedText(t *testing.T) { |
| 16 | const label = "[Pasted text #1 · 536 lines]" |
| 17 | const bodyMarker = "line-536-body-marker" |
| 18 | expanded := label + "\n\n--- Begin " + label + " ---\nfirst line\n" + bodyMarker + "\n--- End " + label + " ---" |
| 19 | display := "/research " + label |
| 20 | |
| 21 | sess := agent.NewSession("parent system") |
| 22 | exec := agent.New(nil, tool.NewRegistry(), sess, agent.Options{}, event.Discard) |
| 23 | events := make(chan event.Event, 12) |
| 24 | var gotTask string |
| 25 | var gotDisplay string |
| 26 | c := newOwnedTestController(t, Options{ |
| 27 | Executor: exec, |
| 28 | Sink: event.FuncSink(func(e event.Event) { events <- e }), |
| 29 | Skills: []skill.Skill{{ |
| 30 | Name: "research", Body: "research system", RunAs: skill.RunSubagent, |
| 31 | Invocation: "manual", Scope: skill.ScopeBuiltin, |
| 32 | }}, |
| 33 | SkillRunner: func(_ context.Context, _ skill.Skill, task string, _ skill.SubagentRunOptions) (string, error) { |
| 34 | gotTask = task |
| 35 | return "research answer", nil |
| 36 | }, |
| 37 | }) |
| 38 | defer c.Close() |
| 39 | c.SetDisplayRecorder(func(_, shown string) { gotDisplay = shown }) |
| 40 | |
| 41 | c.SubmitInvocationDisplay(display, expanded, []InvocationRequest{{Name: "research", Kind: "subagent", Offset: 0}}) |
| 42 | waitForTurnEvents(t, events) |
| 43 | waitIdle(t, c) |
| 44 | |
| 45 | if !strings.Contains(gotTask, bodyMarker) || !strings.Contains(gotTask, "--- Begin "+label+" ---") { |
| 46 | t.Fatalf("research task lost expanded pasted text: %q", gotTask) |
| 47 | } |
| 48 | if gotDisplay != display { |
| 49 | t.Fatalf("display = %q, want folded invocation %q", gotDisplay, display) |
| 50 | } |
| 51 | msgs := c.History() |
| 52 | if len(msgs) != 3 || msgs[1].Role != provider.RoleUser || !strings.Contains(msgs[1].Content, bodyMarker) { |
| 53 | t.Fatalf("parent history lost expanded pasted text: %#v", msgs) |
| 54 | } |
| 55 | } |
| 56 |